python - 一个简单的员工信息增删改查程序

最新推荐文章于 2024-06-28 08:30:00 发布

不才一首歌

最新推荐文章于 2024-06-28 08:30:00 发布

阅读量6.3k

点赞数 2

分类专栏： python学习笔记文章标签：练习

本文链接：https://blog.csdn.net/albert_ycl/article/details/80144796

版权

本文介绍了如何使用Python编写一个简单的员工信息增删改查程序，涵盖了模糊查询、新员工记录创建（以phone为唯一键）、员工信息删除及修改等功能，并在每次操作后显示受影响的记录数。

摘要由CSDN通过智能技术生成

作业

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

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

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

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

代码如下：

from tabulate import tabulate
import os

DB_FILE = "staff.db"
COLUMNS = ['id','name','age','phone','dept','enrolled_date']


def print_log(msg,log_type="info"):
    if log_type == 'info':
        print("\033[32;1m%s\033[0m"%msg)
    elif log_type == 'error':
        print("\033[31;1m%s\033[0m"%msg)


def load_db(db_file):
    """
    第三步：加载员工信息表并转成指定的格式
    :param db_file:
    :return:
    """
    date = {}
    for i in COLUMNS:
        date[i] = []

    f = open(db_file,"r")
    for line in f:
        staff_id,name,age,phone,dept,enrolled_date = line.split(