Python简易用户系统逻辑_def()函数

Python简易用户系统逻辑

目标：用def函数，包装简易的用户系统，实现添加，删除，修改，查询获取所有信息

主要是看逻辑，代码不重要，如有缺漏或者逻辑不对，代码不对的地方请提醒

# 定义用户列表
name_list = []


def display_menu():
    print('_' * 30)
    print("   名片管理系统  v1.0")
    print("1.添加名片")
    print("2.删除名片")
    print("3.修改名片")
    print("4.查询名片")
    print("5.获取所有名片信息")
    print("6.退出系统")
    print('_' * 30)

    # 1添加用户信息
    def add_info():
        new_name = input("请输入用户名")
        if not new_name:
            print('输入为空，请重新输入')
            return add_info()
        elif new_name in name_list:
            print('已有用户，无需添加，是否为用户名相同的不同用户')
            choose = input('请输入yes 或 no')
            if choose == 'yes':
                print('因为技术原因，请自行添加序号以区分')
                # 这里可以判断，为yes在上次输入的值加序号并储存到name_list，再判断是否继续输入
                return add_info()
            elif choose == "no":
                print('请检查输入')
                return add_info()
            else:
                print('输入有误')
            return display_menu()
        else:
            name_list.append(new_name)

    # 2删除用户信息
    def print_chu_info():
        if not name_list:
            print('没有任何用户的储存信息')
            return display_menu()
        else:
            chu_name = input("请输入需要删除的用户名")
            if chu_name in name_list:
                name_list.remove(chu_name)
            else:
                print('没有此用户，无需删除')

    # 3修改用户信息
    def print_xiu_info():
        if not name_list:
            print('没有任何用户的储存信息')
            return display_menu()
        else:
            a = input("请输入需要修改的用户名:")
            b = input("请输入修改后的用户名:")
            if a in name_list:
                name_list.remove(a)
                name_list.append(b)
            else:
                print('未查询到此用户，无需修改')

    # 4查询名片
    def print_cha_info():
        if not name_list:
            print('没有任何用户的储存信息')
            return display_menu()
        else:
            cha_mane = input("请输入需要查询的名片")
            if cha_mane in name_list:
                print("您输入的用户目名查询到了")
            else:
                print("您输入的用户名没有找到")
                return display_menu()

    # 5查看所有名片的信息
    def print_all_info(namelist: list) -> name_list:
        if not name_list:
            print('没有任何用户的储存信息')
        else:
            print("=" * 30)
            for info in namelist:
                print(info)
            print("=" * 30)

    # 获取用户输入的信息
    def get_choice():
        selected_key = input("请输入对应功能的序号")
        return selected_key

    # 用户的选择判定
    key = get_choice()
    if key == "1":
        add_info()
    elif key == "2":
        print_chu_info()
    elif key == "3":
        print_xiu_info()
    elif key == "4":
        print_cha_info()
    elif key == "5":
        print_all_info(name_list)
    elif key == "6":
        exit()
    else:
        print("输入有误，请重新输入！")


i = 0
while i < 1:
    # 打印菜单
    display_menu()