1
'''
2
功能列表
3
1.添加用户
4
2.查看指定用户
5
3.查看所有用户
6
4.删除指定用户
7
5.删除所有用户
8
6.修改指定用户信息
9
7.退出系统
10

11
用户信息总览
12
编号 名字 年龄 爱好 身份信息
13

14
指定编号(自定义)：可以根据指定编号，修改查看指定员工信息
15
指定名字(自定义)
16
指定年龄(自定义)
17
指定爱好(自定义)：可以有多个爱好，可以修改时在原有基础上增加新的爱好
18
指定身份信息(管理员/员工)：管理员具有删除和修改信息功能，员工只能登录注册添加和查看
19
'''

1
from datetime import datetime
2

3
# 用户名和密码信息存储位置 user_pwd.txt
4
# 用户信息存储位置 user_info.txt
5
# 用户日志存储位置 user_log.txt
6

7
# 登录信息存储字典
8
login_user_dict = {'username': '', "is_admin": False}
9

10

11
def _read_data(path):
12
    try:
13
        with open(path, 'r', encoding='utf8') as fp:
14
            data = fp.read()
15
        return True, data
16
    except Exception as error:
17
        return False, f'文件不存在'
18

19

20
def _create_data_dict(**kwargs):
21
    user_data_dict = {}
22
    if kwargs:
23
        user_data_dict.update(kwargs)
24
    return user_data_dict
25

26

27
# 读取用户名和密码信息
28
def _read_pwd_data():
29
    user_pwd_dict = {}
30
    # 读取文件数据
31
    flag, data = _read_data(path='user_pwd.txt')
32
    if not flag:
33
        return data
34
    # 对数据进行切分
35
    for data in data.split('\n'):
36
        if len(data) == 0:
37
            continue
38
        else:
39
            username, password, role = data.split('|')
40
            # 'dream': {'password': '521', 'role': 'admin'}
41
            user_pwd_dict[username] = _create_data_dict(password=password, role=role)
42
    # {'dream': {'password': '521', 'role': 'admin'}, 'hope': {'password': '369', 'role': 'normal'}}
43
    return user_pwd_dict
44

45

46
def _read_info_data():
47
    user_info_dict = {}
48
    flag, data = _read_data(path='user_info.txt')
49
    if not flag:
50
        return data
51
    for data in data.split('\n'):
52
        if len(data) == 0:
53
            continue
54
        else:
55
            user_id, username, age, hobby = data.split('|')
56
            hobby_list = []
57
            for hobby_every in hobby.split('-'):
58
                if 'None' == hobby_every:
59
                    continue
60
                else:
61
                    hobby_list.append(hobby_every)
62
            user_info_dict[username] = _create_data_dict(user_id=user_id, age=age, hobby=hobby_list)
63
    return user_info_dict
64

65

66
# 【1】读取用户信息
67
def read_data(tag_pwd=None, tag_info=None):
68
    # 【2】读取用户名和密码信息
69
    user_pwd_dict = _read_pwd_data()
70
    if 'None' in user_pwd_dict.keys():
71
        user_pwd_dict.pop('None')
72
    # 【3】读取用户详细信息
73
    user_info_dict = _read_info_data()
74
    if 'None' in user_info_dict.keys():
75
        user_info_dict.pop('None')
76
    # 【4】如果标志位(密码)为真并且密码信息为真
77
    if tag_pwd and user_pwd_dict:
78
        return user_pwd_dict
79
    # 【5】如果标志位(用户详细信息)为真并且用户详细信息为真
80
    elif tag_info and user_info_dict:
81
        return user_info_dict
82
    # 【6】默认为全部信息
83
    else:
84
        return user_pwd_dict, user_info_dict
85

86

87
def _create_data(tag, data):
88
    if tag == 'pwd':
89
        pwd_data = f'{data.get("username_input")}|{data.get("password_input")}|{data.get("role")}'
90
        return pwd_data
91
    elif tag == 'info':
92
        # music-listen
93
        hobby_all = '-'.join(data.get('hobby_list'))
94
        # 001|dream|88|music-listen
95
        info_data = f'{data.get("user_id")}|{data.get("username_input")}|{data.get("age_input")}|{hobby_all}'
96
        return info_data
97

98

99
def save_data(path, data, mode='a'):
100
    try:
101
        with open(path, mode, encoding='utf8') as fp:
102
            fp.write(data + '\n')
103
        return True, None
104
    except Exception as error:
105
        return False, error
106

107

108
def get_user_pwd():
109
    username_input = input("请输入用户名 :>>>> ").strip()
110
    password_input = input("请输入密码 :>>>> ").strip()
111
    return username_input, password_input
112

113

114
def check_user_exists(username):
115
    user_pwd_data = read_data(tag_pwd=True)
116
    if username in user_pwd_data:
117
        return True, user_pwd_data.get(username)
118
    else:
119
        return False, None
120

121

122
# 注册
123
def register():
124
    username_input, password_input = get_user_pwd()
125
    if username_input == 'dream' and password_input == '521':
126
        pwd_data = _create_data(tag='pwd', data={"username_input": username_input,
127
                                                 "password_input": password_input,
128
                                                 "role": 'admin'})
129
    else:
130
        pwd_data = _create_data(tag='pwd', data={"username_input": username_input,
131
                                                 "password_input": password_input,
132
                                                 "role": 'normal'})
133
    flag, msg = save_data(path='user_pwd.txt', data=pwd_data)
134
    if flag:
135
        # 001|dream|88|music-listen
136
        info_data = f'None|{username_input}|None|None'
137

138
        save_data('user_info.txt', data=info_data)
139
        save_data('user_log.txt',
140
                  data=f'当前用户 *{username_input}* :>>>> 注册成功! 时间为 :>>>> {datetime.now().strftime("%Y年%m月%d日 %H时%M分%S秒")}')
141
        return True, f'当前用户 *{username_input}* :>>>> 注册成功!'
142
    else:
143
        return False, msg
144

145

146
def _check_login():
147
    if not login_user_dict.get('username'):
148
        return False, f'请先登录!谢谢!'
149
    else:
150
        return True, None
151

152

153
# 登陆
154
def login():
155
    username_input, password_input = get_user_pwd()
156
    flag, user_pwd_data = check_user_exists(username_input)
157
    if not flag:
158
        return False, f'当前用户 *{username_input}* :>>>> 不存在!请先注册!'
159
    if password_input != user_pwd_data.get('password'):
160
        return False, f'当前用户 *{username_input}* :>>>> 用户名或密码失败!'
161
    else:
162
        login_user_dict['username'] = username_input
163
        if user_pwd_data.get('role') == 'admin':
164
            login_user_dict['is_admin'] = True
165
        save_data('user_log.txt',
166
                  data=f'当前用户 *{username_input}* :>>>> 登陆成功!  时间为 :>>>> {datetime.now().strftime("%Y年%m月%d日 %H时%M分%S秒")}')
167
        return True, f'当前用户 *{username_input}* :>>>> 登陆成功!'
168

169

170
def _create_info_data():
171
    try:
172
        # 编号
173
        user_id = input("请输入用户编号 :>>>> ").strip()
174
        # 用户昵称
175
        # username_input = input("请输入用户昵称(默认为当前登录用户) :>>>> ").strip()
176
        # if not username_input:
177
        username_input = login_user_dict.get('username')
178
        # 年龄
179
        age_input = input("请输入年龄 :>>>> ").strip()
180
        # 爱好
181
        hobby_list = []
182
        while True:
183
            hobby_input = input("请输入爱好(q结束) :>>>> ").strip()
184
            if hobby_input == 'q':
185
                break
186
            else:
187
                hobby_list.append(hobby_input)
188
        return user_id, username_input, age_input, hobby_list
189
    except Exception as error:
190
        return None
191

192

193
# 添加用户
194
def add_user():
195
    if not _check_login()[0]:
196
        return False, _check_login()[1]
197
    # 编号 名字 年龄 爱好 身份信息
198
    user_id, username_input, age_input, hobby_list = _create_info_data()
199
    info_data = _create_data(tag='info',
200
                             data={'user_id': user_id, 'username_input': username_input, 'age_input': age_input,
201
                                   'hobby_list': hobby_list})
202
    flag, msg = save_data(path='user_info.txt', data=info_data)
203
    if flag:
204
        return True, f'当前用户 *{username_input}* :>>>> 信息修改成功!'
205
    else:
206
        return False, msg
207

208

209
def _check_user_id(tag_del=False):
210
    user_info_data = read_data(tag_info=True)
211
    user_id_dict = {item.get('user_id'): username for username, item in user_info_data.items()}
212
    if 'None' in user_id_dict:
213
        return False, f'请先修改用户 {user_id_dict.get("None")} 编号信息!'
214
    print(f'当前用户信息如下 :>>>> {user_id_dict}')
215
    user_id_input = input(f'请输入需要查找的编号 :>>>> ').strip()
216
    if user_id_input not in user_id_dict:
217
        return False, f'当前 *{user_id_input}* :>>>> 不存在'
218
    username_out_put = ''
219
    for username, user_info in user_info_data.items():
220
        if user_id_input == user_info.get('user_id'):
221
            username_out_put = username
222
    if not username_out_put:
223
        return False, f'当前 *{user_id_input}* :>>>> 编号查找失败!'
224
    if not tag_del:
225
        return username_out_put, user_info_data[username_out_put]
226
    else:
227
        return username_out_put, user_info_data
228

229

230
# 查看指定用户
231
def search_user_one():
232
    if not _check_login()[0]:
233
        return False, _check_login()[1]
234
    flag, data = _check_user_id()
235
    if flag:
236
        msg = f'''
237
    ----------------- 当前用户 {flag} 信息如下 -----------------
238
                姓名 :>>>> {flag}
239
                编号 :>>>> {data.get('user_id')}
240
                年龄 :>>>> {data.get('age')}
241
                爱好 :>>>> {data.get('hobby')}
242
    --------------------** END **----------------------
243
        '''
244
        return True, msg
245
    else:
246
        return flag, data
247

248

249
# 查看所有用户
250
def search_user_all():
251
    if not _check_login()[0]:
252
        return False, _check_login()[1]
253
    user_info_data = read_data(tag_info=True)
254
    for username, user_info in user_info_data.items():
255
        msg = f'''
256
    ----------------- 当前用户信息 {username} 如下 -----------------
257
                姓名 :>>>> {username}
258
                编号 :>>>> {user_info.get('user_id')}
259
                年龄 :>>>> {user_info.get('age')}
260
                爱好 :>>>> {user_info.get('hobby')}
261
    --------------------** END **----------------------
262
        '''
263
        print(msg)
264
    return True, ''
265

266

267
# 删除指定用户
268
def del_user_one():
269
    if not _check_login()[0]:
270
        return False, _check_login()[1]
271
    flag, data = _check_user_id(tag_del=True)
272
    print(data)
273
    if flag:
274
        data.pop(flag)
275
        print(f'删除后 :>>> {data}')
276
        return True, data
277
    else:
278
        return False, data
279

280

281
# 删除所有用户
282
def del_user_all():
283
    if not _check_login()[0]:
284
        return False, _check_login()[1]
285
    save_data(path='user_pwd.txt', mode='w', data='None|None|None')
286
    save_data(path='user_info.txt', mode='w', data='None|None|None|None')
287
    return True, f'用户信息已清除'
288

289

290
def _change_user_info(username):
291
    user_info_data = read_data(tag_info=True).get(username)
292
    try:
293
        # 编号
294
        user_id = user_info_data.get('user_id')
295
        # 用户昵称
296
        username_input = username
297
        # 年龄
298
        age_input = input("请输入年龄 :>>>> ").strip()
299
        # 爱好
300
        hobby_list = user_info_data.get('hobby')
301
        while True:
302
            hobby_input = input("请输入爱好(q结束) :>>>> ").strip()
303
            if hobby_input == 'q':
304
                break
305
            elif hobby_input in hobby_list:
306
                print(f"当前爱好 {hobby_input} :>>>> 已存在!")
307
                continue
308
            else:
309
                hobby_list.append(hobby_input)
310
        return user_id, username_input, age_input, hobby_list
311
    except Exception as error:
312
        return None
313

314

315
# 修改指定用户信息
316
def change_user_one():
317
    if not _check_login()[0]:
318
        return False, _check_login()[1]
319
    flag, data = _check_user_id()
320
    print(data)
321
    if flag:
322
        user_id, username_input, age_input, hobby_list = _change_user_info(username=flag)
323
        info_data = _create_data(tag='info', data={'user_id': user_id,
324
                                                   'username_input': username_input,
325
                                                   'age_input': age_input,
326
                                                   'hobby_list': hobby_list})
327
        flag, msg = save_data(path='user_info.txt', data=info_data)
328
        if flag:
329
            return True, f'当前用户 *{username_input}* :>>>> 信息修改成功!'
330
        else:
331
            return False, msg
332
    else:
333
        return False, data
334

335

336
# 退出系统
337
def back():
338
    return False, 'break'
339

340

341
func_menu = '''
342
--------------- 欢迎来到梦梦员工系统 ---------------
343
                1:注册
344
                2:登陆
345
                3:添加员工信息
346
                4:查看指定员工信息
347
                5:查看所有员工信息
348
                6:删除指定员工信息
349
                7:删除所有员工信息
350
                8:修改指定员工信息
351
                9:退出系统
352
'''
353

354
func_dict = {
355
    1: register,
356
    2: login,
357
    3: add_user,
358
    4: search_user_one,
359
    5: search_user_all,
360
    6: del_user_one,
361
    7: del_user_all,
362
    8: change_user_one,
363
    9: back
364
}
365

366

367
# 系统入口
368
def main_system():
369
    # 【1】打印功能菜单
370
    while True:
371
        print(func_menu)
372
        func_id = input("请输入功能ID :>>>> ").strip()
373
        if not func_id.isdigit():
374
            print(f'{func_id} :>>>> 非法字符')
375
            continue
376
        func_id = int(func_id)
377
        if func_id not in func_dict.keys():
378
            print(f"{func_id} :>>>> 不存在该功能!")
379
            continue
380
        func = func_dict.get(func_id)
381
        flag, msg = func()
382
        if flag:
383
            print(msg)
384
        elif not flag and msg == 'break':
385
            print(f"欢迎下次使用!再见!")
386
            break
387
        else:
388
            print(msg)
389
            continue
390

391

392
if __name__ == '__main__':
393
    main_system()