python-员工信息管理系统

　　 README

一、作业

现要求你写一个简单的员工信息增删改查程序，需求如下：

当然此表你在文件存储时可以这样表示

1,Alex Li,22,13651054608,IT,2013-04-01
2,Jack Wang,28,13451024608,HR,2015-01-07
3,Rain Wang,21,13451054608,IT,2017-04-01
4,Mack Qiao,44,15653354208,Sales,2016-02-01
5,Rachel Chen,23,13351024606,IT,2013-03-16
6,Eric Liu,19,18531054602,Marketing,2012-12-01
7,Chao Zhang,21,13235324334,Administration,2011-08-08
8,Kevin Chen,22,13151054603,Sales,2013-04-01
9,Shit Wen,20,13351024602,IT,2017-07-03
10,Shanshan Du,26,13698424612,Operation,2017-07-02

1.可进行模糊查询，语法至少支持下面3种查询语法:

find name,age from staff_table where age > 22

find * from staff_table where dept = "IT"

find * from staff_table where enroll_date like "2013"

2.可创建新员工纪录，以phone做唯一键(即不允许表里有手机号重复的情况)，staff_id需自增

语法: add staff_table Alex Li,25,134435344,IT,2015-10-29

3.可删除指定员工信息纪录，输入员工id，即可删除

语法: del from staff where  id=3

4.可修改员工信息，语法如下:

UPDATE staff_table SET dept="Market" WHERE  dept = "IT" 把所有dept=IT的纪录的dept改成Market
UPDATE staff_table SET age=25 WHERE  name = "Alex Li"  把name=Alex Li的纪录的年龄改成25

5.以上每条语名执行完毕后，要显示这条语句影响了多少条纪录。 比如查询语句 就显示 查询出了多少条、修改语句就显示修改了多少条等。

注意：以上需求，要充分使用函数，请尽你的最大限度来减少重复代码！

 二、程序目录结构

├──员工管信息管理系统
            │
            ├—— README
            │
            ├── staff_table   # 员工信息表数据
            │
            ├—— staff.py      # 主程序

三、程序效果图

四、功能验证

 

1、用户登录验证 （用户名：admin  密码：123456）

 

验证目的：当用户未登录时，调用增删改查功能时，都需要身份验证

　

 

2、增加员工信息

 

增加员工时，staff_id 自增，phone 不可重复

 

3、删除员工信息

 

现在删除刚才增加的员工信息，staff_id 为删除条件，删除后，staff_id 自动排序

 

  

4、修改员工信息

 

修改方式：

 

只有age和dept能被修改，修改语句如下(请任选以下修改语句):

 

  a、UPDATE staff_table SET dept="Market" WHERE  dept = "IT"

　

 

b、UPDATE staff_table SET age=25 WHERE  name = "Alex Li"

 

5、查询员工信息

 

本系统支持以下三种查询方法(查询条件分别为：age、dept、enroll_date):

 

a、find name,age from staff_table where age > 22

 

b、find * from staff_table where dept = "IT"

 

c、find * from staff_table where enroll_date like "2013" 

  

 

 

 六、程序代码：

import sys

status_flag = False  # 标记用户登录状态 False代表未登录，Ture代表已登录

# 打印菜单
def menu():
    info = '''
    ==========员工信息表==========

    1、增加员工信息
    2、删除员工信息
    3、修改员工信息
    4、查询员工信息
    5、退出
    =============================
    '''
    print(info)

# 用户登录验证器
def login(func):
    def inner():
        global status_flag     # 声明全局变量
        if status_flag:
            func()
        else:
            user = 'admin'
            passwd = '123456'
            print("请先登录管理员账户！")
            username = input("用户名: ")
            password = input("密码:")
            if username == user and password == passwd:
                print("Welcome！", username)
                status_flag = True
                func()
            else:
                print("Logon failure!")
                sys.exit(1)
    return inner

#将文件信息转换成[[],[]...]形式的列表
def change():
    staff_list = []
    with open('staff_table','r',encoding='utf-8') as staff_f:
        for i in staff_f:
            i = i.strip('
').split(',')
            staff_list.append(i)
    return staff_list

#更新文件信息
def update_file(staff_list):
    with open('staff_table', 'w',encoding='utf-8') as staff_f:
        for i in staff_list:
            i = ','.join(i)+'
'
            staff_f.write(i)

#增加函数
@login
def add():
    print('=*='*20)
    print("添加的内容格式请按照: Alex Li,25,13651054608,IT,2015-10-29")
    print('=*='*20)
    staff_list = change()
    addInfo = input("add staff_table ").strip()
    addInfo_list = addInfo.split(',')
    if len(addInfo_list) == 5:
        phone_list = []
        for i in staff_list:
            phone_list.append(i[3])
        if addInfo_list[2] in phone_list:
            print("phone 号码已存在，此次添加无效！")
        else:
            max_id = int(staff_list[-1][0])
            add_id = max_id + 1
            addInfo_list.insert(0,add_id)
            addInfo_list[0] = str(addInfo_list[0])
            staff_list.append(addInfo_list)
            update_file(staff_list)
            print("以下员工信息已添加成功！")
            print(','.join(addInfo_list))
    else:
        print("%s为无效的输入！"%addInfo)

#删除函数
@login
def delete():
    print('=*=' * 20)
    print("请输入您想要删除员工的staff_id")
    staff_list = change()
    ID = input("del from staff where  id = ").strip()
    if ID.isdigit():
        ID = int(ID)
        del_list = []
        max_id = int(staff_list[-1][0])  # 获取最后一行数据的序号
        if ID <= max_id:
            for i,k in enumerate(staff_list):
                if int(k[0]) == ID:
                    del_list.append(k)
                    del staff_list[i]
            if len(del_list) == 0:
                print("staff_ID为%d的员工不存在..."%ID)
            else:
                print("以下员工已被删除成功！")
                for i in del_list:
                    i[0] = str(i[0])
                    print(','.join(i))
                update_file(staff_list)
        else:
            print("staff_id:%d 不存在！"%ID)
    else:
        print("%s为无效的输入..."%ID)

#修改函数
@login
def update():
    update_info = '''
    =*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*=
    只有age和dept能被修改，修改语句如下(请任选以下修改语句):
    1、UPDATE staff_table SET dept="Market" WHERE  dept = "IT" 
    2、UPDATE staff_table SET age=25 WHERE  name = "Alex Li"  
    '''
    print(update_info)
    staff_list = change()
    update_choice = input("您想根据什么修改信息？(name or dept)：").strip()
    if update_choice == 'dept':
        old_dept = input("请输入修改条件dept：").strip()
        new_dept = input("请输入修改之后dept的值：").strip()
        updateDept_list = []
        for i in staff_list:
            if i[4] == old_dept:
                i[4] = new_dept
                updateDept_list.append(i)
        if len(updateDept_list) > 0:
            print("以下信息已经修改成功！")
            for i in updateDept_list:
                print(','.join(i))
            update_file(staff_list)
        else:
            print("%s不存在！"%old_dept)
    elif update_choice == 'name':
        Name = input("请输入修改条件name的值：").strip()
        new_age = input("请输入修改后的age：").strip()
        updateAge_list = []
        for i,k in  enumerate(staff_list):
            if k[1] == Name:
                staff_list[i][2] = new_age
                updateAge_list.append(k)
        if len(updateAge_list) > 0:
            print("以下信息已经修改成功！")
            for i in updateAge_list:
                print(','.join(i))
            update_file(staff_list)
        else:
            print("%s不存在！"%Name)
    else:
        print("%s为无效的输入..."%update_choice)

#查询函数
@login
def find():
    find_info = '''
    =*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*==*=
    本系统支持以下三种查询方法(查询条件分别为：age、dept、enroll_date):
    1、find name,age from staff_table where age > 22
    2、find * from staff_table where dept = "IT"
    3、find * from staff_table where enroll_date like "2013"    
    '''
    print(find_info)
    staff_list = change()
    find_choice = input("您想使用哪种方法查询员工信息(请填写查询条件)：").strip()
    if find_choice == 'age':
        find_age = input("find name,age from staff_table where age > ").strip()
        if find_age.isdigit():
            find_age = int(find_age)
            findAge_list = []
            for i in staff_list:
                i[2] = int(i[2])
                if i[2] > find_age:
                    findAge_list.append(i)
            if len(findAge_list) > 0:
                print("符合条件的信息如下：")
                for i in findAge_list:
                    print("%s,%s"%(i[1],i[2]))
            else:
                print("无age大于%s的员工信息！"%find_age)
        else:
            print("%s非数字，输入无效！"%find_age)
    elif find_choice == 'dept':
        find_dept = input("find * from staff_table where dept = ")
        findDept_list = []
        for i in staff_list:
            if i[4] == find_dept:
                findDept_list.append(i)
        if len(findDept_list) > 0:
            print("符合条件的信息如下：")
            for i in findDept_list:
                print(','.join(i))
        else:
            print("无dept为%s的员工信息！" % find_dept)
    elif find_choice == 'enroll_date':
        findEnroll_date = input("find * from staff_table where enroll_date like ")
        findEnroll_date_list = []
        for i in staff_list:
            i_date = i[5]
            i_date = i_date.split('-')[0]
            if i_date == findEnroll_date:
                findEnroll_date_list.append(i)
        if len(findEnroll_date_list) > 0:
            print("符合条件的信息如下：")
            for i in findEnroll_date_list:
                print(','.join(i))
        else:
            print("Enroll_date为%s 的信息不存在！"%findEnroll_date)
    else:
        print("%s为无效的输入！"%find_choice)

def main():
    while True:
        menu()
        choice = input("请选择:").strip()
        if choice == '1':
            add()
        elif choice == '2':
            delete()
        elif choice == '3':
            update()
        elif choice == '4':
            find()
        elif choice == '5':
            sys.exit("logout......")
        else:
            print("无效的输入！")

main()