1 #!/usr/bin/env python3
 2 # -*- coding:utf-8 -*-
 3 while True:
 4     #初始系统
 5     operation = input('''
 6     ------------------------------------
 7     欢迎进入管理系统
 8     请选择您需要的操作
 9     1.注册
10     2.登陆
11     ------------------------------------
12     ''')
13
14     #用户注册
15     if operation == '1':
16         #读取账号文件
17         account_file = open('register_file.txt', 'a+')
18         account_file.close()
19         account_file = open('register_file.txt','r+')
20         account_list = account_file.readlines()
21         #提示用户输入注册的账号
22         while True:
23             name = input('请输入需要注册的用户名')
24             for account_line in account_list:
25                 (user,passwd) = account_line.strip('\n').split()
26                 if name == user:
27                     print('该用户名已存在')
28                     break
29             else:
30                 password = input('请输入需要注册的密码')
31                 account_file.write(name + ' ' +password + '\n')
32                 account_file.close()
33                 break
34     #用户登录
35     elif operation == '2':
36         name = input('请输入用户名')
37         lock_account = open('lock_file.txt', 'a+')
38         lock_account.close()
39         lock_account = open('lock_file.txt','r+')
40         lock_list = lock_account.readlines()
41         lock = 0
42         for lock_line in lock_list:
43             if name == lock_line.strip('\n'):
44                 lock =1
45                 print('该用户已被锁定')
46         #读取账户文件
47         if lock == 0:
48             account_file = open('register_file.txt','r+')
49             account_list = account_file.readlines()
50             #循环将账号密码赋给user，passwd两个变量
51             for account_line in account_list:
52                 (user,passwd) = account_line.strip('\n').split()
53                 #账号和用户输入的账户名相等时
54                 if name == user:
55                     i = 0
56                     while i <3:
57                         #提示用户输入密码
58                         password = input('请输入密码')
59                         #密码也相同，则进入系统
60                         if password == passwd:
61                             print('登陆成功，欢迎进入管理系统')
62                             account_file.close()
63                             break
64                         #密码不相同，提示用户重新输入
65                         else:
66                             i += 1
67                             print('密码错误，请重新输入')
68                     #输入次数过多，锁定用户
69                     else:
70                         print('错误次数过多，用户已锁定')
71                         lock_account.write('\n'+user)
72                         account_file.close()
73                         break
74                     break
75             else:
76                 print('用户不存在')
77                 account_file.close()
78         break
79
80     else:
81         print('选项无效，请重新输入')