1 import pymysql, hashlib
2 class loginRegistration:
3 def __init__(self):
4 # 连接数据库
5 self.bankData = pymysql.connect("127.0.0.1", "root", "123123", "user")
6 # 设置字符编码
7 self.bankData.set_charset('utf8')
8 # 创建游标对象
9 self.cursor = self.bankData.cursor()
10
11 def registration(self):
12 # 查找login表所有数据
13 sql = 'select * from login'
14 # 运行sql句子
15 self.cursor.execute(sql)
16 # 如果数据为0
17 if self.cursor.rowcount == 0:
18 print('注册用户为0')
19 else:
20 print('注册人数为', self.cursor.rowcount, "个")
21 userInputName = input('请输入用户名')
22 userInputPassword = input('请输入密码')
23 sql = 'select username from login where username="{}"'.format(userInputName)
24 self.cursor.execute(sql)
25 # 如果数据为0
26 if self.cursor.rowcount == 0:
27 # 调用md5加密
28 passwordMd5 = hashlib.md5()
29 passwordMd5.update(userInputPassword.encode('utf8'))
30 password = passwordMd5.hexdigest()
31 # 添加数据，用try来检测提交是否成功
32 try:
33 sql = "insert into login(username,password) values('{}','{}' )".format(userInputName, password)
34 self.cursor.execute(sql)
35 # 提交服务器中
36 self.bankData.commit()
37 print("注册成功")
38 except:
39 print("注册失败")
40 else:
41 print("用户已存在")
42
43 def login(self):
44 sql = 'select * from login'
45 # 运行sql句子
46 self.cursor.execute(sql)
47 userInputName = input('请输入用户名')
48 userInputPassword = input('请输入密码')
49 sql = 'select username,password from login where username="{}"'.format(userInputName)
50 self.cursor.execute(sql)
51 passwordMd5 = hashlib.md5()
52 passwordMd5.update(userInputPassword.encode('utf8'))
53 password = passwordMd5.hexdigest()
54 # 如果数据为0
55 if self.cursor.rowcount == 0:
56 print("用户不已存在")
57 elif self.cursor.fetchone()[1] != password:
58 print("密码错误")
59 else:
60 print("登录成功")
61
62
63 lr = loginRegistration()
64 while True:
65 print("1、注册 2、登录")
66 userinput = input("")
67 if userinput is "1":
68 lr.registration()
69 elif userinput is "2":
70 lr.login()
