python 学习分享-实战篇增删改查作业

一大波函数来袭

作业要求：

1本次作业通过空格及逗号，将文件拆分成列表，在通过判断add、del、update、select等关键字，来判断用户执行的是哪种命令，根据不同的命令调用不同的函数去处理。

2增加操作及删除操作较为简单，根据规范来执行即可，查询及修改操作较为复杂。

3查询函数，首先要进行判断=及like是否在其中，根据这一点来确定select方式，然后判断关键字的位置，即其在列表中的位置。通过这些就可以查询出内容。

4修改函数类似于查询函数，因为修改操作比较单一，所以只需要判断关键字即可。

5最后就是主程序函数的编写，根据关键字判断是哪种操作，然后执行即可。

import time
staff_head = ['staff_id','name','age','phone','dept','enroll_date']
with open('user.txt','r') as user:
    staff_table = []
    for us_line in user:
        sta_line = us_line.strip('\n').split(',')
        staff_table.append(sta_line)  #staff_table 列表
def staff_table_get():     #打印staff_table函数
    print('STAFF_TABLE'.center(60,'='))
    print('\033[34;0m【staff_id】【name】【age】【phone】【dept】【enroll_date】\033[0m')
    for i in staff_table:
            print(i)
def input_get():           #input order函数
    print('ORDER'.center(60,'='))
    add_order = 'add name=\033[1;31;0mLaay\033[0m age=\033[1;31;0m18\033[0m phone=\033[1;31;0m13563630129\033[0m dept=\033[1;31;0mIT\033[0m'
    del_order = 'del \033[1;31;0m1\033[0m'
    upd_order = 'update staff_table set \033[1;31;0mdept\033[0m = Market where dept = \033[1;31;0mIT\033[0m'
    sel_order_1 = 'select \033[1;31;0mname,age\033[0m from staff_table where \033[1;31;0mage > 22\033[0m'
    sel_order_2 = 'select  \033[1;31;0m*\033[0m from staff_table where \033[1;31;0mdept = IT\033[0m'
    sel_order_3 = 'select  \033[1;31;0m*\033[0m from staff_table where \033[1;31;0menroll_date like 2013\033[0m'
    print('增加=>  ',add_order)
    print('删除=>  ',del_order)
    print('修改=>  ',upd_order)
    print('查询=>  ',sel_order_1)
    print('查询=>  ',sel_order_2)
    print('查询=>  ',sel_order_3)
    print('退出=>  ','\033[1;32;0mq\033[0m')
    print('INPUT ORDER'.center(60, '='))
    input_order = input('\033[31;0m Input your order:\033[0m' )
    return input_order
def func_sel(input_ord):  #查询函数
    input_list = input_ord.split()
    judge = ' '.join(input_list[5:])
    out_list = []
    for i in staff_table:
        staff_id = i[0]
        name = i[1]
        age = int(i[2])
        phone = int(i[3])
        dept = i[4]
        enroll_date = i[5]
        if 'like' in judge:
            if input_list[7] in i[staff_head.index(input_list[5])] and input_list[1] == '*':
                out_list.append(i)
            elif input_list[7] in i[staff_head.index(input_list[5])]  and input_list[1] != '*':
                input_list_mix = input_list[1].split(',')
                out_mix = []
                for j in input_list_mix:
                    out_mix.append(i[staff_head.index(j)])
                out_list.append(out_mix)
        elif '=' in judge:
            if input_list[7]  ==  i[staff_head.index(input_list[5])] and input_list[1] == '*':
                out_list.append(i)
            elif input_list[7]  ==  i[staff_head.index(input_list[5])]and input_list[1] != '*':
                input_list_mix = input_list[1].split(',')
                out_mix = []
                for j in input_list_mix:
                    out_mix.append(i[staff_head.index(j)])
                out_list.append(out_mix)
        else:
            if eval(judge) and input_list[1] == '*':
                out_list.append(i)
            elif eval(judge) and input_list[1] != '*':
                input_list_mix = input_list[1].split(',')
                out_mix = []
                for j in input_list_mix:
                    out_mix.append(i[staff_head.index(j)])
                out_list.append(out_mix)
    if len(out_list)>0:
        print('查询结果'.center(60,'='))
        for z in out_list:
            print('\033[36;3m%s\033[3m'%(z))
        print('共计有\033[35;3m{}\033[3m条数据'.format(len(out_list)))
    else:
        print('wrong ,please try again!')
def func_upd(input_ord): #更改函数
    input_list = input_ord.split()
    if input_list[3] in staff_head:
        j = staff_head.index(input_list[3])
        i_j = []
        for i in staff_table:
            i_j.append(i[j])
        if input_list[9] in i_j:
            for z in staff_table:
                if z[j] == input_list[9]:
                    z[j] = input_list[5]
                    print('修改成功')
                else:
                    continue
        else:
            print('wrong input ,please try again!')
    else:
        print('wrong input,please try again!')
#add name=bb age=22 phone=13563636541 dept=Sales
def func_add(input_ord): #增加函数
    input_list = input_ord.split()
    i_j = []
    for i in staff_table:
        i_j.append(i[3])
    if input_list[3].split('=')[1] not in i_j:
        staff_add = []
        # staff_add[0] = str(len(staff_table)+2)
        # staff_add[1] = input_list[1].split('=')[1]
        # staff_add[2] = str(input_list[2].split('=')[1])
        # staff_add[3] = str(input_list[3].split('=')[1])
        # staff_add[4] = input_list[4].split('=')[1]
        # staff_add[5] = str(time.strftime('%Y-%m-%d',time.localtime(time.time())))
        staff_add.append(str(int(staff_table[-1][0])+1))
        staff_add.append(input_list[1].split('=')[1])
        staff_add.append(str(input_list[2].split('=')[1]))
        staff_add.append(str(input_list[3].split('=')[1]))
        staff_add.append(input_list[4].split('=')[1])
        staff_add.append(str(time.strftime('%Y-%m-%d',time.localtime(time.time()))))
        staff_table.append(staff_add)
        print('增加成功')
    else:
        print('这个号码已经在表中了')
def func_del(input_ord):
    input_list = input_ord.split()
    i_j = []
    for i in staff_table:
        i_j.append(i[0])
    if input_list[1] in i_j:
        for j in staff_table:
            if input_list[1] == j[0]:
                staff_table.remove(j)
                print('\033[34;0m删除成功\33[0m')
            else:
                continue
    else:
        print('wrong staff_id')
def judge_get():         #输入判断函数
    while 1:
        staff_table_get()
        input_order = input_get()
        if 'select' in input_order and len(input_order.split()) == 8:
            func_sel(input_order)
        elif 'update' in input_order and len(input_order.split()) == 10:
            func_upd(input_order)
        elif 'add' in input_order and len(input_order.split()) == 5:
            func_add(input_order)
        elif 'del' in input_order and len(input_order.split()) == 2:
            func_del(input_order)
        elif input_order.strip() == 'q':
            break
        else:
            print('Wrong input，please try again')
            continue
def wrt_op():
    with open('user.txt','r') as old_us , open('user_back.txt','w') as back_user:
        for l_b in old_us:
            back_user.write(l_b)
    with open('user.txt', 'w') as new_us :
        for n_b in staff_table:
            wrt = n_b[0]+','+n_b[1]+','+n_b[2]+','+n_b[3]+','+n_b[4]+','+n_b[5]+'\n'
            new_us.write(wrt)
judge_get()
wrt_op()