员工管理系统

# 完成以下功能
 '''
--------------- 员工系统 --------------- 
                1:注册
                2:登陆
                3:添加员工信息
                4:查看指定员工信息
                5:查看所有员工信息
                6:删除指定员工信息
                7:删除所有员工信息
                8:修改指定员工信息
                9:退出系统
'''

from datetime import datetime

# 部分文件内容声明
# 共存在三份文本文件，负责存储用户登录信息，用户个人信息，操作流水
# user_info : 编号|用户名|年龄|性别|爱好
# user_login : 用户名|密码|权限
# user_flow : 用户>>>>{操作}>>>>{流水时间}
# files_list = ['user_info.txt','user_login.txt','user_flow.txt']

time = f'时间为 :>>>> {datetime.now().strftime("%Y年%m月%d日 %H时%M分%S秒")}'

login_data = {'username': '', 'pwd': '', 'is_admin': False}


def check_login(tag):  # 验证登录信息
    if login_data.get('username') != '':
        if tag == 'admin':
            if login_data.get('username') == 'lea4ning':
                login_data['is_admin'] = True
                return True, None
            else:
                return False, f"当前功能只有管理员可以使用！"
        elif tag == 'user':
            pass
    else:
        return False, f"请先登录！"


def user_input():  # 输入信息   # 返回信息字典
    username = input("请输入用户名:").strip()
    user_pwd = input("请输入密码:").strip()
    return username, user_pwd


def register():  # 注册功能   # user_login
    # 函数块主题需要传回 flag和msg
    # 如果是进行注册就返回一个字典{用户名：{pwd，limit}}
    username, user_pwd = user_input()
    data = read_data('user_login.txt', 'login')
    login_dict = {}
    if username == 'lea4ning' and user_pwd == '12345':
        user_limit = 'admin'
    elif username in data.keys():
        return False, "用户已注册，请登录"
    else:
        user_limit = 'user'
    login_dict[username] = {'pwd': user_pwd, 'limit': user_limit}
    save_data(tag="register", data=login_dict)
    str_data = f"{username}:>>>>>注册成功>>>{time}"
    print(str_data)
    flag, msg = save_file(path='user_log.txt', str_data=str_data)
    return flag, msg


def add_user():
    flag, msg = check_login('admin')
    if flag:
        flag, msg = add_info('add_user')
    return flag, msg


def add_info(tag):  # 添加用户信息 user_info
    data = read_data('user_info.txt', 'info')
    username = ''
    user_pwd = ''
    user_sex = ''
    user_id = ''
    user_age = ''
    hobby_list = []
    if tag == 'renew':
        renew_name = list_id('renew')
        username = renew_name
    else:
        username, user_pwd = user_input()
    while True:
        print("信息提交".center(30, '-'))
        if tag == 'renew':
            user_id = data[username]['id']
        else:
            user_id = input("请输入编号（编号不可重复）:").strip()
            if user_id in [key['id'] for key in data.values()]:
                print("此编号已存在，请重试！")
                continue
        # 编号不可以重复
        user_age = input("请输入年龄（请输入纯数字）:").strip()
        if not user_age.isdigit():
            print("请输入纯数字！")
            continue
        # 年龄为纯数字
        user_sex = input("请输入性别（请在男或女中选择:").strip()
        # 性别限定选项
        if user_sex not in ['男', '女']:
            print(f"输入有误，错误内容{user_sex}，请重新输入")
            continue
        while True:
            hobby_input = input("请输入爱好（q结束添加）:").strip()
            if tag == 'renew':
                hobby_list = data[username]['hobby']
            if hobby_input.upper() == 'Q':
                break
            elif hobby_input in hobby_list:
                print(f"请不要重复添加，{hobby_input}已经存在！")
                continue
            else:
                hobby_list.append(hobby_input)  # 爱好通过上述筛选后，添加到hobby_list中
        break  # 退出信息输入的循环
    info_dict = {username: {'id': user_id, 'name': username, 'age': user_age, 'sex': user_sex, 'hobby': hobby_list}}
    flag, msg = save_data(tag=tag, data=info_dict)
    return flag, msg


def save_data(tag, data):  # 做切分，返回路径和字符串内容
    str_data = ''
    path = ''
    if tag == 'register':
        path = 'user_login.txt'
        for key, value in data.items():
            str_data = f"{key}|{value.get('pwd')}|{value.get('limit')}"
    elif tag == 'add_user':
        path = 'user_info.txt'
        for key, value in data.items():
            str_hobby = '-'.join(value.get('hobby'))
            str_data = f"{value.get('id')}|{key}|{value.get('age')}|{value.get('sex')}|{str_hobby}"
    elif tag == 'del_one':
        path = 'user_info.txt'
        del_name = data
        str_data = f"None|{del_name}|None|None|None"
    elif tag == 'renew':
        path = 'user_info.txt'
        for key, value in data.items():
            str_hobby = '-'.join(value.get('hobby'))
            str_data = f"{value.get('id')}|{key}|{value.get('age')}|{value.get('sex')}|{str_hobby}"
    flag, msg = save_file(path=path, str_data=str_data)
    return flag, msg


def save_file(path, str_data):  # 将数据写入至文件中，保存到本地文本文档
    print(str_data)
    try:
        with open(path, 'a', encoding='utf-8') as fp:
            fp.write(str_data + '\n')
            return True, f"文件更改成功，可在{path}中查看！"
    except Exception as e:
        return False, f"文件写入失败{e}！"


def read_data(path, tag):
    try:
        data = {}
        with open(path, 'r', encoding='utf8') as fp:
            list_data = fp.read().split('\n')
            if tag == 'login':
                for login in list_data:
                    # print(len(login))
                    if len(login) == 0:
                        continue
                    # print(login)
                    username, user_pwd, limit = login.split('|')
                    data[username] = {'pwd': user_pwd, 'limit': limit}
            elif tag == 'info':
                for info in list_data:  # 编号|用户名|年龄|性别|爱好
                    if info == '':
                        continue
                    elif 'None' in info:
                        continue
                    user_id, username, user_age, user_sex, hobby_list = info.split('|')
                    hobby = hobby_list.split('-')
                    data[username] = {'id': user_id, 'age': user_age, 'sex': user_sex, 'hobby': hobby}
        return data
    except Exception as e:
        return f"文件读取失败！错误信息为：{e}"


def login():  # 登录功能
    # 函数块主题需要传回 flag和msg
    username, user_pwd = user_input()
    data = read_data(path='user_login.txt', tag='login')
    if username not in data.keys():
        return False, f"{username}尚未注册，请先注册!"
    true_pwd = data[username].get('pwd')
    if user_pwd == true_pwd:
        str_data = f"{username}:>>>>>登录成功>>>{time}"
        login_data['username'] = username
        login_data['pwd'] = user_pwd
        save_file(path='user_log.txt', str_data=str_data)
        return True, f"登录成功！欢迎{username}！"
    else:
        return False, "登录失败！"


def list_id(tag):
    data = read_data(path='user_info.txt', tag='info')
    list_id = [data[key].get('id') for key in data.keys()]
    # dict_id = {key: keys for key in list_id for keys in data.keys() for value in data.values() if value['id'] == key}
    dict_id = {key: keys for key in list_id for keys, value in data.items() if value['id'] == key}
    print(f"当前所有id如下：\n{dict_id}")
    choice = ''
    if tag == 'check':
        choice = input("请输入需要查看的用户id:")
    elif tag == 'del':
        choice = input("请输入需要删除的用户id:")
    elif tag == 'renew':
        choice = input("请输入需要修改的用户id:")
    for check_id in dict_id:
        if check_id == choice:
            check_name = dict_id[choice]
            return check_name
        else:
            return "输入编号有误，请重新输入！"


def check_one():  # 查看指定用户信息
    # 函数块主题需要传回 flag和msg
    data = read_data(path='user_info.txt', tag='info')
    check_name = list_id('check')
    txt = data[check_name]
    return True, txt


def check_all():  # 查看所有用户信息
    # 函数块主题需要传回 flag和msg
    data = read_data(path='user_info.txt', tag='info')
    return True, data


def del_one():  # 删除指定用户信息
    # 函数块主题需要传回 flag和msg
    check_login('admin')
    del_name = list_id('del')  # 编号：姓名
    print(f"已将信息删除")
    flag, msg = save_data(tag='del_one', data=del_name)
    return flag, msg


def del_all():  # 删除所有用户信息
    check_login('admin')
    # 函数块主题需要传回 flag和msg
    try:
        with open('user_info.txt', 'w', encoding='utf8') as fp:
            fp.write('')
            return True, "信息库已清空！"
    except Exception as e:
        return False, f"删除失败！错误信息:{e}"


def renew_info():  # 修改指定用户信息
    flag, msg = check_login('admin')
    # 函数块主题需要传回 flag和msg
    if flag:
        flag, msg = add_info('renew')
    return flag, msg


def back_system():  # 退出系统
    # 函数块主题需要传回 flag和msg
    return False, 'break'


# 功能菜单
func_menu = '''
--------------- 欢迎来到员工管理系统 --------------- 
                1:注册
                2:登陆
                3:添加员工信息
                4:查看指定员工信息
                5:查看所有员工信息
                6:删除指定员工信息
                7:删除所有员工信息
                8:修改指定员工信息
                9:退出系统
'''
func_dict = {
    1: register,
    2: login,
    3: add_user,
    4: check_one,
    5: check_all,
    6: del_one,
    7: del_all,
    8: renew_info,
    9: back_system
}


def main_system():  # 主系统操作
    while True:
        print(func_menu)  # 打印系统菜单
        func_id = input("请输入功能ID :>>>> ").strip()  # 收集用户输入信息
        if not func_id.isdigit():  # 如果输入除数字外的内容，报错并要求用户重新选择
            print(f'{func_id} :>>>> 非法字符')
            continue
        func_id = int(func_id)  # 如果输入的全是数字，强转成数字类型
        if func_id not in func_dict.keys():  # 如果输入的数字并不在功能列表中，输出功能不存在并返回菜单页面
            print(f"{func_id} :>>>> 不存在该功能!")
            continue
        # 以上用户输入信息经过筛选，只剩下在功能表格中的数字，通过字典取值，取到相应的功能
        # func作为变量名接收功能的变量名
        # func() 作为运行的函数
        # 举例：选择6,对应在功能字典中的值就是del_one，func = del_one
        # flag,msg = del_one()
        func = func_dict.get(func_id)
        flag, msg = func()
        if flag:
            print(msg)
        elif not flag and msg == 'break':
            print(f"欢迎下次使用!再见!")
            break
        else:
            print(msg)
            continue


if __name__ == '__main__':
    main_system()