Python初学第一段代码之【优化】-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_46012855/article/details/140016672


          
          
           
           # 登录功能
           
           
def login():
           
           
    import os
           
           
    while True:
           
           
        login_username = input('登录==> 用户[退出登录程序q]: ').strip()
           
           
        if login_username == 'q':
           
           
            print('''
           
           
            已退出登录程序！！！！
           
           
                    ''')
           
           
            break
           
           
        if not os.path.exists('db.txt'):
           
           
            with open('db.txt', 'w', encoding='utf-8') as _:
           
           
                ...
           
           
        with open('db.txt','r',encoding='utf-8',) as f:
           
           
            for line in f:
           
           
                username,password = line.strip().split(':')
           
           
                if login_username == username:
           
           
                    while True:
           
           
                        login_password = input('登录==>密码: ').strip()
           
           
                        if login_password == password:
           
           
                            print(f'{login_username} 登录成功！\n')
           
           
                            break
           
           
                        else:
           
           
                            print('密码错误，请重新输入!!!\n')
           
           
                    break
           
           
            else:
           
           
                print(f'{login_username} 用户不存在，请重新输入！！！')
           
           

           
           
# 注册功能
           
           
def register():
           
           
    import os
           
           
    while True:
           
           
        register_username = input('注册==> 用户[退出注册程序q]: ').strip()
           
           
        if register_username == 'q':
           
           
            print('''
           
           
    已退出注册程序！！！！
           
           
            ''')
           
           
            break
           
           
        register_password = input('注册==> 密码: ').strip()
           
           
        if len(register_username) == 0 or len(register_password) == 0:
           
           
            print('非法输入，不能为空！')
           
           
            continue
           
           
        if not os.path.exists('db.txt'):
           
           
            with open('db.txt','w',encoding='utf-8') as _:
           
           
                ...
           
           
        with open('db.txt','r+',encoding='utf-8') as f:
           
           
            for line in f:
           
           
                a1,_ = line.split(':')
           
           
                if a1 in register_username:
           
           
                    print('该用户已存在，请重新注册!')
           
           
                    break
           
           
            else:
           
           
                f.seek(0,2)
           
           
                f.write(f'{register_username}:{register_password}\n')
           
           
                print(f'[{register_username}] 注册成功!\n')
           
           

           
           
# 功能字典
           
           
dict_list = {
           
           
    '0':('退出',None),
           
           
    '1':('登录',login),
           
           
    '2':('注册',register),
           
           
}
           
           

           
           
# 功能调用
           
           
while True:
           
           
    for line in dict_list:
           
           
        print('==>',line,dict_list[line][0])
           
           
    number = input('请输入功能编号: ')
           
           
    if number == '0':
           
           
        print('退出程序！')
           
           
        break
           
           
    print('请输入编号，大憨憨！\n') if not number.isdigit() else ...
           
           
    print('对不起, 该编号不存在\n') if not dict_list.get(number) else dict_list[number][1]()