1 """
2 2、注册
3 需要自己创建一个user表,id,username,password,error_count
4 username
5 password
6 cpassword
7 密码要存密文
8 用户是否存在要校验
9 """
10
11 import tools,login
12
13 def check_data(username,password,cpassword):
14 if login.check_data(username,password) and password == cpassword:
15 return True
16
17 def check_db(username, password):
18 query_user_sql = "select * from nhy_user where username='%s';" % username
19 user_info = tools.op_mysql(query_user_sql, all=False, data_type=2)
20 if user_info:
21 print("用户已存在")
22 return
23 md5_password = tools.my_md5(password)
24 create_user_sql = "insert into nhy_user (username,password) values ('%s','%s');" % (username,md5_password)
25 tools.op_mysql(create_user_sql)
26 print("注册成功!")
27 return True
28
29
30 def main():
31 for i in range(5):
32 username = input("username:").strip().lower()
33 password = input("password:").strip()
34 cpassword = input("cpassword:").strip()
35 if not check_data(username,password,cpassword):
36 continue
37 if check_db(username,password):
38 break
39 else:
40 print("错误次数过多")
41
42
43 if __name__ == '__main__':
44 main()