1 # -*- coding:utf-8 -*-
2 def login(username, password):
3 """
4 用于用户登陆
5 :param username: 用户名
6 :param password: 密码
7 :return:
8 """
9 f = open('text', 'r')
10 lines = f.readlines() # 调用文件的readlines方法，读取文件全部内容
11 for line in lines:
12 line_list = line.strip().split(' ')
13 if line_list[0] == username and line_list[1] == password:
14 return True
15 return False
16
17
18 def register():
19 """
20 用于用户注册
21 :return:
22 """
23 user = input("请输入您的用户名：")
24 pwd = input("请输入您的密码：")
25 pwd_2 = input("请再次输入您的密码：")
26 if pwd != pwd_2:
27 print("第二次输入的密码和第一次不同，请重新注册！\n")
28 register()
29 f = open('text', 'a')
30 temp = '\n' + user + ' ' + pwd
31 f.write(temp)
32 f.close()
33 print("注册成功！")
34
35
36 def main():
37 choice = input("1、登陆 2、注册\n")
38 if choice == '1':
39 user = input("请输入您的用户名：")
40 pwd = input("请输入您的密码：")
41 result = login(user, pwd)
42 if result:
43 print("登陆成功！")
44 else:
45 print("登陆失败，用户名或密码错误！")
46 if choice == '2':
47 register()
48
49 main()
