员工管理系统

# 【一】用户可以注册，并将注册信息临时保存起来，登陆时可根据指定用户名或密码进行登陆  --->注册 和 登录
# 【二】设定好用户名和密码，用户通过输入指定的用户名和密码进行登陆 ---> 登录功能
# 【1】最大尝试次数：用户最多尝试猜测3次
# 【2】最大尝试次数后：如3次后，问用户是否继续登陆
# （1）如果回答Y或y，就再给3次机会，提示【还剩最后三次机会】
# （2）3次都猜错的话登录结束
# （3）如果回答N或n，登陆结束！
# 【三】如果格式输入错误，提示【输入格式错误，请重新输入：】 ---> 对用户名有要求 默认都是校验数字
# 【四】如果猜对了，登陆结束！


# 【1】添加员工信息
# 用户登录选择添加员工信息功能，可以添加指定员工信息
# 信息格式如下
# {'员工编号':{员工数据}, '员工编号':{员工数据}, '员工编号':{员工数据}}
# 【2】修改员工薪资
# 用户可以根据员工编号选择指定员工
# 然后可以修改指定员工薪资(也可以修改其他选项)
# 【3】查看指定员工
# 用户根据指定员工编号查看指定员工信息
# 【4】查看所有员工
# 用户可以查看所有员工信息
# 【5】删除员工数据
# 用户可以根据指定员工编号删除指定员工信息

function_menu = '''
--------------- 员工管理系统 ---------------
                1.管理员注册功能
                2.登录功能
                3.添加员工信息
                4.修改员工信息
                5.查找所有员工信息
                6.查看指定员工信息
                7.删除指定员工信息
                8.删除所有员工信息
                9.查看当前登录信息
                退出系统（q）
'''
user_data_dict = {'max':{'username':'max', 'password':'999', 'role': True},
                  'yn':{'userID':'0001' ,'username':'yn', 'password':'123', 'age':18, 'salary':850000.0,
                        'hobby':['music','basketball'], 'role':False}}
login_dict = {
    'username': '',
    'role': None,
}
while True:
    print(function_menu)
    func_id = input("请输入指令>>>>>").strip()
    if func_id =='q':
        break

    if not func_id.isdigit():
        print("当前指令格式错误！")
        continue
    func_id = int(func_id)
    if func_id not in [1, 2, 3, 4, 5, 6, 7, 8, 9]:
        print("当前指令不合法！")
        continue

    #注册功能
    if func_id == 1:
        print("注册中.....")
        username_input = input("username>>>>>").strip()
        password_input = input("password>>>>>").strip()
    #校验当前用户名是否存在
        if username_input in user_data_dict:
            print(f"当前用户 {username_input} 已存在，请登录")
            continue

        #添加管理员的信息
        if username_input == 'max' and password_input == '999':
            role = True
        else:
            role = False

        #增加数据字典
        user_data_dict[username_input] ={
            'username':username_input,
            'password':password_input,
            'role':role,
        }
        print(f"用户 {username_input} 注册成功")

#员工登录
    elif func_id == 2:
        print("正在登录中.....")
        username_input = input("username>>>>>").strip()
        password_input = input("password>>>>>").strip()
#先校验该用户是否注册过
        if username_input not in user_data_dict.keys():
            print(f"当前用户 {username_input} 不存在请先注册！")
            continue
        elif username_input in user_data_dict.keys() and password_input == user_data_dict[username_input].get('password'):
            print(f"欢迎用户 {username_input} 上线")
            #验证其是否是管理员登录
            login_dict['username'] = username_input
            login_dict['role'] = user_data_dict[username_input].get('role')
        else:
            print("当前用户密码错误！")
            continue

#添加员工信息
    elif func_id == 3:
        print("添加中.....")
        #验证是否是管理员进入
        if not login_dict.get('username'):
            print("请先登录再操作！")
            continue
        elif not login_dict.get('role'):
            print("用户权限不足！")
            continue
        #获取员工数据
        username_input = input("username>>>>").strip()
        #判断用户是否存在
        user_data = user_data_dict.get(username_input)
        #能从数据库获取该人的用户信息，不能注册已经存在的人
        if user_data:
            print(f"该用户 {username_input} 已存在")
        else:
            #添加新员工信息
            password_input = input("password>>>>>").strip()
            #添加年龄
            while True:
                age_input = input("age>>>>>").strip()
                if not 16< int(age_input) <80:
                    print("当前年龄不合法！")
                    continue
                elif not age_input.isdigit():
                    print("当前年龄格式不正确！")
                break
            #添加薪资
            salary_input = float(int(input("salary>>>>>").strip()))
            #添加爱好
            hobby_all = []
            while True:
                hobby_input = input("hobby>>>>>").strip()
                if hobby_input =='q':
                    break
                elif hobby_input not in hobby_all:
                    hobby_all.append(hobby_input)
                else:
                    print("这个爱好已经有了，换一个")
                    continue
            #添加员工ID
            user_ID = input("ID>>>>>").strip()
            user_ID = user_ID.rjust(4,'0')
            user_data_dict[username_input] = {
                'userID' : user_ID,
                'username' : username_input,
                'password' : password_input,
                'age' : age_input,
                'salary' : salary_input,
                'hobby' : hobby_all,
                'role' : False
            }
            print(f"当前用户 {username_input} 信息添加成功")

#修改员工信息
    elif func_id == 4:
        print("修改中.....")
        if not login_dict.get('username'):
            print("请先登录再操作！")
            continue
        elif not login_dict.get('role'):
            print("用户权限不足！")
            continue
        print(f"已有员工信息如下.....")
        user_id_list = []
        for username, user_data in user_data_dict.items():
            if username == 'max':
                pass
            else:
                user_id_list.append(user_data.get('userID'))
                print(f"当前用户ID : {user_data.get('userID')} | 用户名: {username}")
        #输入想要修改员工信息的ID
        userid_input = input("请输入员工ID>>>>>").strip()
        if userid_input not in user_id_list:
            print(f"当前ID不存在")
            continue

        #若存在要先创建空字典存储用户数据
        user_data = {}
        for username,user_data_ in user_data_dict.items():
            #判断输入的ID和已存在的ID一致
            if userid_input == user_data_.get('userid_input'):
                user_data_dict.update(user_data_)
        #修改薪资
        salary_input = float(int(input("salary>>>>>").strip()))
        user_data_dict[username]['salary'] = salary_input
        print(f"当前用户薪资修改成功！")



#查找所有员工信息
    elif func_id == 5:
        print("查找所有用户中.....")
        if not login_dict.get('username'):
            print("请先登录再操作！")
            continue
        print(f'''
        ---------------------用户信息---------------------
        ''')
        for username, user_data in user_data_dict.items():
            if username == 'max':
                pass
            else:
                print(f'''
                编  号：{user_data.get('userID')}
                用户名：{username}
                年  龄：{user_data.get('age')}
                薪  资：{user_data.get('salary')}
                爱  好：{user_data.get('hobby')}
                ''')
        print(f'''
        -------------------------------------------------
                ''')

    #查看指定员工信息
    elif func_id == 6:
        print(f"查看指定用户.....")
        #判断是否有管理员用户登录
        if not login_dict.get('username'):
            print("请先登录再操作！")
            continue
        print(f"当前已有用户信息如下...")
        user_id_list = []
        for username,user_data in user_data_dict.items():
            if username == 'max':
                pass
            else:
                user_id_list.append(user_data.get('userID'))
                print(f"当前员工ID：{user_data.get('userID')} | 用户名：{username}")

        user_ID = input("输入员工ID>>>>>").strip()
        if user_ID not in user_id_list:
            print(f"当前员工ID不存在！")
            continue
            #若存在要先创建空字典来存储用户数据
        user_data = {}
        for username,user_data_ in user_data_dict.items():
            if user_ID == user_data_.get('userID'):
                #将原字典更新成新字典
                user_data.update(user_data_)
        print(f'''
        ---------------------当前ID：{user_ID} 员工信息---------------------
                          用户名：{user_data.get('username')}
                          年  龄：{user_data.get('age')}
                          薪  资：{user_data.get('salary')}
                          爱  好：{user_data.get('hobby')}
        -----------------------------------------------------------------                  
        ''')

    #删除指定员工
    elif func_id == 7:
        print("删除员工...")
        #判断当前是否有管理员用户登录
        if not login_dict.get('username'):
            print("请先登录再操作！")
            continue
        elif not login_dict.get('role'):
            print("用户权限不足！")
            continue
        print(f"当前已存在用户名如下")
        for username, user_data in user_data_dict.items():
            if user_data.get('role'):
                pass
            else:
                print(username)
        username_input = input("需要删除的用户名：").strip()
        #判断当前用户是否存在
        if username_input not in user_data_dict.keys() and username_input != 'max':
            print(f"当前用户 {username_input} 不存在！")
        del user_data_dict[username_input]
        print(f"当前用户信息 {username_input} 已删除")

    #删除所有用户
    elif func_id == 8:
        print(f"清空所有员工信息....")
        # 判断当前是否有管理员用户登录
        if not login_dict.get('username'):
            print("请先登录再操作！")
            continue
        elif not login_dict.get('role'):
            print("用户权限不足！")
            continue

        #构架一个假的原来数据格式的字典
        #{'username’:{}}
        user_data_dict_ = {}
        #遍历原来的字典
        for username, user_data in user_data_dict.items():
            if user_data.get('role'):
                #判断如果是管理员，按照原来的格式更新上面的空字典
                #更新字典的时候是动的，根据每一个用户名动，不能是死值（username）
                user_data_dict_.update({user_data.get('username'): user_data})
            else:
                pass
        #清空原来的字典
        user_data_dict.clear()
        #将管理员字典信息更新回去
        user_data_dict.update(user_data_dict_)
        print(f"信息清空！")

    #查看当前登录用户信息
    elif func_id == 9:
        print(f'''
        ---------------------当前登录用户信息---------------------
                        用户名：{login_dict.get('username') if login_dict.get('username') else "未登录"}
                        身  份：{'管理员' if login_dict.get('role') else '未知'}
        ''')