python员工管理系统

创建员工类：

属性有：员工号、员工姓名、员工工资

1：初始化3个员工对象（可以使用列表、元组、字典存储）持久化存储数据

创建用户类：

属性有：用户名、密码

2：初始化3个用户对象（可以使用列表、元组、字典存储）

功能菜单：

一级菜单：登录（需要验证身份）：（验证用户名和密码）

二级菜单：登录成功进入二级菜单

（1）、查询所有员工信息

（2）、根据工号查询员工信息

（3）、添加员工

（4）、修改员工工资

（5）、删除员工

（6）、退出

class Staff:
    def __init__(self, staff_id, staff_name, staff_salary):
        self.staff_id = staff_id
        self.staff_name = staff_name
        self.staff_salary = staff_salary

    def get_sta_id(self):
        return self.staff_id

    def get_sta_name(self):
        return self.staff_name

    def get_sta_salary(self):
        return self.staff_salary


class User:
    def __init__(self, username, password):
        self.username = username
        self.password = password

    def get_username(self):
        return self.username

    def get_password(self):
        return self.password


s1 = Staff('01', '张三', '6000')
s2 = Staff('02', '李四', '7000')
s3 = Staff('03', '王五', '8000')

staff_list = [s1, s2, s3]

u1 = User('admin', '123456')
u2 = User('root', '123456')
u3 = User('user1', '123456')

user_list = [u1, u2, u3]


str_message = """
******************************************
【员工管理系统】V1.0

1.查询所有员工信息
2.根据工号查询员工信息
3.添加员工
4.修改员工工资
5.删除员工

6.退出
******************************************
"""


def login():
    username = input("请输入用户名：")
    password = input("请输入密码：")

    flag = False    # 标志位
    for i in user_list:
        if username == i.get_username() and password == i.get_password():
            flag = True
            print("登录成功")
            print(str_message)
            choose()
    if not flag:
        print("用户名或密码不正确~")


def choose():
    while True:
        temp = input('请输入想要的操作：')
        if temp == '1':
            query_all()
        elif temp == '2':
            query_id()
        elif temp == '3':
            add_staff()
        elif temp == '4':
            update_staff()
        elif temp == '5':
            delete_staff()
        elif temp == '6':
            print("退出系统！")
            break
        else:
            print("您输入的操作有误！")


def query_all():
    print('员工号\t员工姓名\t\t员工工资')
    for i in staff_list:
        print("{}\t\t{}\t\t\t{}".format(i.staff_id, i.staff_name, i.staff_salary))


def query_id():
    staff_id = input('请输入员工工号：')
    flag = False
    for i in staff_list:
        if staff_id == i.get_sta_id():
            flag = True
            print('员工号\t员工姓名\t\t员工工资')
            print("{}\t\t{}\t\t\t{}".format(i.staff_id, i.staff_name, i.staff_salary))

    if not flag:
        print('此员工不存在！')


def add_staff():
    staff_id = input("请输入工号：")
    staff_name = input("请输入员工姓名：")
    staff_salary = input("员工工资：")
    s = Staff(staff_id, staff_name, staff_salary)
    staff_list.append(s)
    print("员工号为{}添加成功：".format(staff_id))


def update_staff():
    staff_id = input("请输入要查询的员工工号：")
    flag = False
    for i in staff_list:
        if staff_id == i.get_sta_id():
            flag = True
            new_salary = input("请输入员工的新工资：")
            s = Staff(staff_id, i.get_sta_name(), new_salary)
            staff_list.append(s)
            staff_list.remove(i)
            print("员工号为{}添加成功".format(staff_id))
            break
    if not flag:
        print("此员工不存在！")


def delete_staff():
    staff_id = input("请输入要删除的员工工号：")
    flag = False
    for i in staff_list:
        if staff_id == i.get_sta_id():
            flag = True
            temp = i
            break
    staff_list.remove(temp)
    if not flag:
        print("此员工不存在！")


login()