第四十七天Python学习记录

作业

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

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

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

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

import os


def table_staff():  # 将txt文本信息转换成列表，并向外部函数返回
    data_file = open('data.txt', 'r+')
    data = data_file.readlines()
    data_list = []
    for i in data:
        data_list.append(str(i).split(','))
    return data_list


def del_update_share(data_value, data_file1):  # 删除更改共享函数，实现去掉最末行换行符、转换、写文件和更名的功能

    a = (','.join(data_value[len(data_value) - 1]).replace("
", "")).split(",")  # 去掉最末行换行符
    data_value[len(data_value) - 1] = a

    for i in data_value:
        str_i = ','.join(i)
        data_file1.write(str_i)

    data_file1.flush()
    data_file1.close()

    os.remove("data.txt")
    os.rename("data1.txt", "data.txt")


def query_staff():  # 员工信息查询，通过切片方式获取用户输入关键字，根据关键字获取用户查询信息
    data_value = table_staff()
    # print(data_value)
    input_query = input("请输入查询语句：")
    num1 = input_query[0:input_query.rfind('from', 0) + -1][5:]  # 截取 name,age
    num2 = input_query[input_query.rfind('>', 1) + 2:]  # 截取22
    num3 = input_query[input_query.rfind('where', 1) + 6:]  # 截取 dept = "IT"

    if num1 == '*':
        value = 'dept = "IT"'
        if value == num3:
            count = 0
            for k in data_value:
                if "IT" in k[4]:
                    str_i = ','.join(k).replace("
", "").replace(",", " ")
                    print(str_i)
                    count += 1
            print("查询到%s条结果！" % count)

        else:
            count = 0
            for k in data_value:
                if "2013" in k[5]:
                    str_i = ','.join(k).replace("
", "").replace(",", " ")
                    print(str_i)
                    count += 1
            print("查询到%s条结果！" % count)

    else:
        count = 0
        for k in data_value:
            if int(k[2]) > int(num2):
                print(k[1], k[2])
                count += 1
        print("查询到%s条结果！" % count)


def add_staff():  # 增加员工信息，phone不能重复，通过a+在data.txt后追加，
    data_value = table_staff()

    input_add = input("请输入新增语句：")
    input_list = list(input_add[16:].split(','))

    phone_list = []
    for i in data_value:
        phone_list.append(i[3])

    if input_list[2] in phone_list:
        print("手机号已存在！")
        exit()

    else:
        staff_id = len(data_value) + 1
        input_list.insert(0, str(staff_id))
        input_list_str = ','.join(input_list)

        data_file = open('data.txt', 'a+')
        data_file.write("
")
        data_file.write(input_list_str)
        data_file.close()

        print("新增1条员工记录！")


def remove_staff():  # 删除员工信息，利用输入的切片信息作为判断条件，以占硬盘的方式的删除
    data_value = table_staff()
    input_remove = input('请输入删除语句：')
    id_value = input_remove[input_remove.rfind('=', 1) + 2:]

    if input_remove[:32] == "del from staff_table where id = ":
        data_file1 = open('data1.txt', 'a+', encoding='utf-8')

        del data_value[int(id_value) - 1]
        del_update_share(data_value, data_file1)

        print("删除1条员工记录！")

    else:
        print("您输入的格式不正确！")


def update_staff():  # 修改员工信息，利用输入的切片信息作为判断条件，以占硬盘的方式的修改
    data_value = table_staff()

    input_update = input('请输入修改语句：')
    if input_update[:22] == "UPDATE staff_table SET":
        num1 = input_update[:35][29:]  # 截取Market
        num2 = input_update[:29][27:]  # 截取age

        data_file1 = open('data1.txt', 'a+', encoding='utf-8')
        if input_update[:28] == "UPDATE staff_table SET dept=":
            count = 0
            for i in data_value:
                if i[4] == 'IT':
                    i[4] = num1
                    count += 1

            del_update_share(data_value, data_file1)
            print("您修改了%s条员工信息！" % count)

        elif input_update[:27] == "UPDATE staff_table SET age=":
            count = 0
            for i in data_value:
                if i[1] == 'Alex Li':
                    i[2] = num2
                    count += 1

            del_update_share(data_value, data_file1)
            print("您修改了%s条员工信息！" % count)

        else:
            print("您输入的格式不正确！")

    else:
        print("您输入的格式不正确！")


while True:  # 一直循环主程序，直到用户选择功能后退出
    print(
        '欢迎进入员工信息增删改查程序'.center(50, '*'), """
    1. 模糊查询  (语法： 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. 新增信息  (语法： add staff_table Alex,25,134435344,IT,2015-10-29 )
    
    3. 删除信息  (语法： del from staff_table where id = 3)
    
    4. 修改信息  (语法： UPDATE staff_table SET dept="Market" WHERE dept = "IT"
                       UPDATE staff_table SET age=25 WHERE name = "Alex Li" )
    5. 退出                  
    """,
        '欢迎进入员工信息增删改查程序'.center(50, '*'))

    choice = int(input("请选择要进入的功能序号："))
    if choice == 1:
        print("成功进入查询功能".center(50, '*'))
        query_staff()
        break

    elif choice == 2:
        print("成功进入新增功能".center(50, '*'))
        add_staff()
        break

    elif choice == 3:
        print("成功进入删除功能".center(50, '*'))
        remove_staff()
        break

    elif choice == 4:
        print("成功进入新增功能".center(50, '*'))
        update_staff()
        break

    elif choice == 5:
        print("退出成功！")
        break
    else:
        print("您输入的信息有误！")

注意：此代码有个bug，只能顺序增加和删除，不能从中间删除和添加，从中间删除后，会导致staff_id 和 索引的混乱，