python基础案例练习一

"""
员工管理系统练习
1.显示系统菜单
2.获得用户输入的菜单
3.根据用户的输入判断执行操作
"""

# 存储员工信息
employee = {}

def show_menu():
    print('*'*20 + '员工管理系统菜单' + '*'*20)
    print('1.添加员工信息')
    print('2.删除员工信息')
    print('3.修改员工信息')
    print('4.查看员工信息')
    print('5.退出')

# 添加员工信息
def add_employee():

    # 1.获得手动输入的员工信息
    employee_id = input('请输入员工编号：')
    # 2.判断员工编号是否存在，若存在提示编号重复，若不重复则继续操作
    all_employee_id = list(employee.keys())
    if employee_id in all_employee_id:
        print('员工编号重复，无法添加')
        return
    # 3.若员工编号不重复则进行下面的操作
    employee_name = input('请添加员工姓名：')
    employee_age = input('请添加员工年龄')
    employee_salary = input('请输入员工工资')
    # 3..保存员工信息到字典employee中，编号为key，其他信息为value
    employee_info = {'name':employee_name,'age':employee_age,'salary':employee_salary}
    employee[employee_id] = employee_info #保存员工信息到字典employee中
    print('员工编号为' + employee_id + '的信息添加成功')

# 删除员工信息
def remove_employee():
    # 1.获得要删除的员工编号
    employee_id = input('请输入要删除的员工编号：')
    # 2.如果员工编号不存在，提示员工不存在，结束函数执行
    all_employee_id = list(employee.keys())
    if employee_id not in all_employee_id:
        print('员工编号不存在，无法添加')
        return
    # 3.如果员工编号存在，执行删除操作,使用del关键字删除员工信息
    del employee[employee_id]
    print('员工编号为' + employee_id + '的信息删除成功')

# 显示员工信息
def show_employee():
    # 遍历员工信息
    for per_employee in employee.items():
        print(per_employee[0],per_employee[1]['name'],per_employee[1]['age'],per_employee[1]['salary'])

# 修改员工信息
def edit_employee():
    # 1.获得要修改员工的编号
    employee_id = input('请输入要修改的员工编号：')

    all_employee_id = list(employee.keys())
    # 2.判断编号是否存在
    if employee_id not in all_employee_id:
    # 3.如果不存在提示输入错误
        print('员工编号不存在，无法修改')
        return
    # 4.如果存在，显示源信息，执行修改操作
    new_employee_name = input('您的姓名是：'+employee[employee_id]['name']+'您要修改为：')
    new_employee_age = input('您的年龄是：'+employee[employee_id]['age']+'您要修改为：')
    new_employee_salary = input('您的工资是：' + employee[employee_id]['salary'] + '您要修改为：')
    # 若输入为空则不更新
    if new_employee_name != '':
        employee[employee_id]['name'] = new_employee_name
    if new_employee_age != '':
        employee[employee_id]['age'] = new_employee_age
    if new_employee_salary != '':
        employee[employee_id]['salary'] = new_employee_salary
    print('员工编号为'+ employee_id + '的员工信息修改成功')

while True:
    # 显示系统菜单
    show_menu()
    # 获得用户输入的信息
    my_operate = input('请输入您的操作：')
    # 根据用户输入判断执行操作
    if my_operate == '1':
        add_employee()
        print(employee)
    elif my_operate == '2':
        remove_employee()
        print(employee)
    elif my_operate == '3':
        edit_employee()
    elif my_operate == '4':
        show_employee()
    elif my_operate == '5':
        print('退出系统')
        break
    else:
        print('您的输入有误')