Python-名片管理器

# 需要完成的基本功能：
#     添加名片
#     刪除名片
#     修改名片
#     查詢名片
#     退出系統
#     程序運行后，除非選擇退出系統，否則重復執行功能
list_info = []  # 創建一個空列表


# 打印菜單
def menu():
    print("---------------------------")
    print("      名片管理系統")
    print(" 1:添加名片")
    print(" 2:刪除名片")
    print(" 3:修改名片")
    print(" 4:查詢名片")
    print(" 5:顯示所有名片")
    print(" 6:退出系統")
    print("---------------------------")


# 添加名片
def add():
    global list_info
    name = input('請輸入你的姓名：')
    position = input('請輸入你的職務：')
    number = int(input('請輸入你的手機號：'))
    company_name = input('請輸入你的公司名稱：')
    company_address = input('請輸入你的公司地址：')

    for temp_info in list_info:
        if temp_info['姓名'] == name:
            print("此用戶名已經被占用,請重新輸入")
            return  # 如果一個函數只有return就相當於讓函數結束，沒有返回值
    # 創建字典，存儲名片信息
    information = {
        '姓名': name,
        '職務': position,
        '手機號': number,
        '公司名稱': company_name,
        '公司地址': company_address,
    }
    # 向列表中添加這個字典
    list_info.append(information)


# 刪除名片
def delete():
    global list_info

    del_num = int(input("請輸入要刪除的序號:"))
    if 0 <= del_num < len(list_info):
        del_flag = input("你確定要刪除么?yes or no")
        if del_flag == "yes":
            del list_info[del_num]
    else:
        print("輸入序號有誤,請重新輸入")


# 修改名片
def alter():
    global list_info

    modify_num = int(input("請輸入要修改的序號:"))
    if 0 <= modify_num < len(list_info):
        print("你要修改的信息是:")
        print("姓名:%s, 職務:%s, 手機號:%s, 公司名稱:%s, 公司地址:%s" % (list_info[modify_num]['姓名'],
                                                          list_info[modify_num]['職務'], list_info[modify_num]['手機號'],
                                                          list_info[modify_num]['公司名稱'], list_info[modify_num]['公司地址']))
        list_info[modify_num]['姓名'] = input("請輸入新的姓名:")
        list_info[modify_num]['職務'] = input("請輸入新的職務:")
        list_info[modify_num]['手機號'] = input("請輸入新手機號:")
        list_info[modify_num]['公司名稱'] = input("請輸入新公司名稱:")
        list_info[modify_num]['公司地址'] = input("請輸入新公司地址:")
    else:
        print("輸入序號有誤,請重新輸入")


# 查詢名片
def query():
    query_name = input('請輸入你要查詢的名片名字：')
    for temp_info in list_info:
        if temp_info['姓名'] == query_name:
            print("查詢到的信息如下:")
            print("姓名:%s, 職務:%s, 手機號:%s, 公司名稱:%s, 公司地址:%s" % (
                temp_info['姓名'], temp_info['職務'], temp_info['手機號'], temp_info['公司名稱'], temp_info['公司地址']))
            break
    else:
        print("沒有您要找的信息....")


# 顯示所有名片
def traverse():
    print("序號\t姓名\t\t職務\t\t手機號\t\t公司名稱\t\t公司地址")
    i = 0
    for temp in list_info:
        # temp是一個字典
        print("%d\t%s\t\t%s\t\t%s\t\t%s\t\t%s" % (i, temp['姓名'], temp['職務'], temp['手機號'], temp['公司名稱'], temp['公司地址']))
        i += 1


def main():
    while True:
        menu()
        digital = int(input('請輸入相對應的功能數字：'))
        if digital == 1:
            add()
        elif digital == 2:
            delete()
        elif digital == 3:
            alter()
        elif digital == 4:
            query()
        elif digital == 5:
            traverse()
        elif digital == 6:
            break
        else:
            print('請輸入錯誤，請重新輸入')


main()