import pymysql conn = pymysql.connect( host='127.0.0.1', port=3306, user='root', password='123456', database='day48', charset='utf8' ) cursor = conn.cursor(cursor=pymysql.cursors.DictCursor) while True: msg = ''' ======= 欢迎来到注册登录界面 ======= | 1.注册 | | 2.登录 | | 3.退出 | ============== END ============== ''' print(msg) func_num = input("请输入功能编号:").strip() if func_num == '1': while True: reg_usr = input('请输入用户名:').strip() reg_pwd1 = input("请输入密码:").strip() reg_pwd2 = input("请确认密码:").strip() if not (reg_usr and reg_pwd1 and reg_pwd2): print('输入不能为空') continue if not reg_pwd1 == reg_pwd2: print('两次密码不一致') continue sql = "select * from user where usr=%s;" if cursor.execute(sql, (reg_usr,)): print('用户名已存在') continue sql = 'insert into user(usr,pwd) values(%s, %s)' cursor.execute(sql, (reg_usr, reg_pwd1)) conn.commit() # 提交数据到数据库并保存 print('注册成功') break elif func_num == '2': while True: log_usr = input('请输入账号:').strip() log_pwd = input("请输入密码:").strip() if not (log_usr and log_pwd): print('账号密码不能为空') continue sql = "select * from user where usr=%s and pwd=%s;" if cursor.execute(sql, (log_usr, log_pwd)): print('登录成功!') break print('账号或密码错误!') elif func_num == '3': exit() else: print('输入错误,该命令不存在!')