小程序：登陆和注册

 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()