python函数作业

# _*_ coding:utf-8 _*_

import os
def select(data):
    """
    查询语法一：select name,age from staff_table where age > 22
    查询语法二：select * from staff_table where dept = IT
    查询语法三：select  * from staff_table where enroll_date like "2013"
    """
    data1 = data.split()#传入变量去空格成字符串
    if data == ('select name,age from staff_table where age > %s' %(data1[7])):
        with open("aaa.txt", encoding="utf-8") as f:
            list = []
            for line in f:
                i = line.strip().split(",")
                a = i[1]
                b = i[2]
                c = [a, b]
                if b > data1[7]:
                    list.append(c)
            for i in list:
                print(','.join(i))
            print("查询到%s条符合的信息" %len(list))
    elif data == ('select * from staff_table where dept = %s' %(data1[7])):
        with open("aaa.txt", encoding="utf-8") as f2:
            list2 = []
            for line in f2:
                i = line.strip().split(",")
                if data1[7]==i[4]:
                    list2.append(i)
            for i in list2:
                print(','.join(i))
            print("查询到%s条符合的信息" %len(list2))
    elif data == ('select  * from staff_table where enroll_date like %s' % (data1[8])):
        with open("aaa.txt", encoding="utf-8") as f3:
            list3 = []
            for line in f3:
                i = line.strip().split(",")
                if data1[8].strip('""') == i[5][:4]:
                    list3.append(i)
            for i in list3:
                print(','.join(i))
            print("查询到%s条符合的信息" %len(list3))


def add(data):
    """
    添加语法： name,age,phone,dept,enroll-date
    (例如：张,22,123456789,IT,2017-06-18)
    """
    data1 = data.split(",")
    with open("aaa.txt", encoding="utf-8")as f:
        list = []
        phone_list = []
        for line in f:
            i = line.strip().split(",")
            phone_list.append(i[3])
        if data1[2] in phone_list:
            print("phone已存在")
        else:
            with open("aaa.txt", "r+", encoding="utf-8")as f:
                for line in f:
                    i = line.strip().split(",")
                    # print(i)
                    list.append(i)
                w = str(int(list[-1][0]) + 1)
                data1.insert(0, w)
                data1 = ','.join(data1)
                f.write('\n'+data1)
                print(data1 +'\n' "添加成功!!!")

def remove(data):
    """
    删除语法：delete from staff_table where staff_id = 12
    """
    data1 = data.split(" ")
    if data == ("delete from staff_table where staff_id = %s" %data1[6]):
        with open("aaa.txt", encoding="utf-8") as f:
            for line in f:
                i = line.strip().split(",")
                if data1[6] == i[0]:
                    i2 = ','.join(i)
                    print(i2)
        with open("aaa.txt", encoding="utf-8")as f_read,open("aaa_back.txt", "a+", encoding="utf-8")as f_write:
            for i in f_read:
                if i2 in i:
                    i = i.replace(i2, "").strip()
                f_write.write(i)
                # f_write.flush()
        os.remove("aaa.txt")
        os.rename("aaa_back.txt","aaa.txt")
        print("删除成功！！！")

def change(data):
    """
    修改语法：UPDATE staff_table SET dept="Market" WHERE where dept = "IT"
    """
    data1 = data.split()
    with open("aaa.txt", encoding="utf-8") as f_read,open("aaa_back.txt","w",encoding="utf-8")as f_write:
        #f_read读，f_write用来复用
        for line in f_read:
            if  data1[8][1:-1] in line:#查看表中的职位是不是IT
                line = line.replace(data1[8][1:-1],data1[3][6:-1])
            f_write.write(line)
            # f_write.flush()
    os.remove("aaa.txt")
    os.rename("aaa_back.txt","aaa.txt")
    print("修改成功!!!")


if __name__ == "__main__":
    msg = """
    1:查询
    2:添加
    3:删除
    4:修改
    5:退出
    """
    msg_dict = {
        "1": select,
        "2": add,
        "3": remove,
        "4": change,
        "5": exit,
    }
    while True:
        print(msg)
        choice = input("输入序号>>:")
        if len(choice) == 0 or choice not in msg_dict: continue
        if choice =='5':break
        data = input("请输入数据>>:").strip()
        msg_dict[choice](data)

　　文件

1,zhang,22,11111111111,Market,2013-04-01
2,wang,30,22222222222,HR,2015-05-03
3,li,25,33333333333,SALES,2016-04-22
4,zhao,40,44444444444,IT,2013-04-01
5,张,22,123456789,Market,2017-06-18
9,张,22,223456789,Market,2017-06-18

       staff_id  name  age  phone  dept  enroll_date

要求

员工信息表程序，实现增删改查操作

1.可进行模糊查询，语法至少支持下面3种:
　　select name,age from staff_table where age > 22
　　select  * from staff_table where dept = ""IT""
   select  * from staff_table where enroll_date like ""2013""

2.查到的信息，打印后，最后面还要显示查到的条数 
3.可创建新员工纪录，以phone做唯一键，staff_id需自增
4.可删除指定员工信息纪录，输入员工id，即可删除
5.可修改员工信息，语法如下:
　UPDATE staff_table SET dept="Market" WHERE where dept = "IT"