1 def login(username, password):
2 """
3 用于用户登录
4 :param username: 用户输入的用户名
5 :param password: 用户输入的密码
6 :return: true，表示登录成功；false，登录失败
7 """
8 f = open("db", 'r')
9 for line in f:
10 line_list = line.strip().split("|")
11 if line_list[0] == username and line_list[1] == password:
12 return True
13 return False
14
15
16 def register(username, password):
17 """
18 用于用户注册
19 :param username: 用户名
20 :param password: 密码
21 :return: 默认None
22 """
23 f = open("db", 'a')
24 temp = "\n" + username + "|" + password
25 f.write(temp)
26 f.close()
27
28
29 def main():
30 t = input("1：登录；2：注册")
31 if t == "1":
32 user = input("请输入用户名:")
33 pwd = input("请输入密码:")
34 r = login(user, pwd)
35 if r:
36 print("登录成功")
37 else:
38 print("登录失败")
39 elif t == "2":
40 user = input("请输入用户名:")
41 pwd = input("请输入密码:")
42 register(user, pwd)
43
44 main()
