员工信息表-装逼版

import os
file_name = '员工信息表'
noun_list = ['staff_id', 'name', 'age', 'phone', 'dept', 'enroll_date']


def file_operation(file):
    '''
    先进行文件操作，把文件内容加载到内存，以方便调用的形式储存文件信息
    :param file_name:
    :return:
    '''

    file_info = open(file, 'r', encoding='utf-8')
    data_dic = {'staff_id': [], 'name': [], 'age': [], 'phone': [], 'dept': [], 'enroll_date': []}
    for line in file_info:
        info_id, info_name, info_age, info_tel, info_dept,info_time=line.split(',')
        # 按照上述的顺序依次把信息存入value的空列表中
        data_dic['staff_id'].append(info_id)
        data_dic['name'].append(info_name)
        data_dic['age'].append(info_age)
        data_dic['phone'].append(info_tel)
        data_dic['dept'].append(info_dept)
        data_dic['enroll_date'].append(info_time)
    file_info.close()
    return data_dic


def file_save():
    f = open('临时存放','w',encoding='UTF-8')
    for id_index, id in enumerate(data['staff_id']):
        info_list = []
        for noun in noun_list:
            info_list.append(data[noun][id_index])
        f.write(','.join(info_list))
    f.close()
    os.replace('临时存放',file_name)


def catch_order():   #获取用户指令  # step:1
    while True:
        user_order = input('请输入指令:\n--->\t').strip()  # 交互第一步，让用户输入指令，首尾去空格
        if not user_order: continue
        input_order(user_order)  # 把用户指令传入（指令分析）的函数  # 2


def print_login(msg,login_type='info'):
    if login_type == 'info':
        print('\033[31;1m%s\033[0m' %msg)
    elif login_type == 'error':
        print('\033[41;1m%s\033[0m' %msg)


def input_order(order):   # step:2
    '''
    获取并分析用户输入的指令
    :return:
    '''
    judge = {'find': find_info, 'add': add_info, 'del': del_info, 'update': update_info}
    if order.split()[0] in ('find', 'add', 'del', 'update'):  # 查看指令的开头写的动作，是否在支持的范围内
        if 'where'in order:
            left_order, right_order = order.split('where')   # 以where为界，把信息分为两半，左半是动作，右半是条件
            catch_info = judge_right(right_order)  # 把右半边的筛选条件传入（判断where）的函数里面  # 3 -> 5
           # 把抓取的信息和具体的操作指令传入操作函数中
            info_action = order.split()[0]
            if info_action in judge:
                judge[info_action](catch_info, left_order)
        else:
            catch_info = []
            for id_index, id in enumerate(data['staff_id']):
                info_list = []
                for noun in noun_list:
                    info_list.append(data[noun][id_index])
                catch_info.append(info_list)
            info_action = order.split()[0]
            if info_action in judge:
                judge[info_action](catch_info, order)

    else:
        print_login('输入有误，无法获取具体指令！')


def judge_greater(n, c):   # 判断语句中有大于号的情况 # step:4
    '''
    判断大于的情况
    :param n: 这是要筛选的对象名称（年龄，职位...）
    :param c: 这是要筛选的具体条件（IT，22, 2017...）
    :return:  把筛选出的内容
    '''
    staff_info_1 = []
    for index,condition in enumerate(data[n]):
        if float(condition) > float(c):  # 如果原有信息大于传进来的具体条件
            staff_info_2 = []
            for noun in noun_list:
                staff_info_2.append(data[noun][index])
            staff_info_1.append(staff_info_2)
    return staff_info_1


def judge_smaller(n, c):   # 判断语句中有小于号的情况 # step:4
    '''
    判断小于的情况
    :param n: 这是要筛选的对象名称（年龄，职位...）
    :param c: 这是要筛选的具体条件（IT，22, 2017...）
    :return:
    '''
    pass
    staff_info_1 = []
    for index,condition in enumerate(data[n]):
        if float(condition) < float(c):
            staff_info_2 = []
            for noun in noun_list:
                staff_info_2.append(data[noun][index])
            staff_info_1.append(staff_info_2)
    return staff_info_1


def judge_equal(n, c):   # 判断语句中有等于号的情况 # step:4
    '''
    判断等于的情况
    :param n: 这是要筛选的对象名称（年龄，职位...）
    :param c: 这是要筛选的具体条件（IT，22, 2017...）
    :return:
    '''
    staff_info_1 = []
    for index,condition in enumerate(data[n]):
        if condition == c:  # 如果原有信息大于传进来的具体条件
            staff_info_2 = []
            for noun in noun_list:
                staff_info_2.append(data[noun][index])
            staff_info_1.append(staff_info_2)
    return staff_info_1


def judge_like(n, c):   # 判断语句中有like的情况 # step:4
    '''
    判断相似的情况
    :param n: 这是要筛选的对象名称（年龄，职位...）
    :param c: 这是要筛选的具体条件（IT，22, 2017...）
    :return:
    '''
    staff_info_1 = []
    for index, condition in enumerate(data[n]):
        if c in condition:  # 如果原有信息大于传进来的具体条件
            staff_info_2 = []
            for noun in noun_list:
                staff_info_2.append(data[noun][index])
            staff_info_1.append(staff_info_2)
    return staff_info_1


def judge_right(right_order):  # step: 3
    '''
    分析右半边的筛选条件，并过滤数据输出符合条件的信息
    :param right_order:
    :return:
    '''
    judge = {'>': judge_greater, '<': judge_smaller, '=': judge_equal, 'like': judge_like}
    # 把符号和对应的函数名存成字典，循环字典来判断语句中的筛选符号是什么
    for judge_k,judge_func in judge.items():
        if judge_k in right_order:
            noun, condition = (right_order.split(judge_k))
            noun = str(noun).strip()
            condition = str(condition).strip()
            # 把名词，条件按照相应输入的判断符号来区分开
            catch_info = judge_func(noun,condition)  # 把获取来的名词和筛选条件传进匹配函数，来进行筛选  # 4
            return catch_info
    else:
        print_login('输入有误，缺少明确的判断符号(>,<,=,like...)！')
        #走到这就说明指令里面没有明确的判断符，无法进行下一步判断


def find_info(catch_info, left_order):
    '''
    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
    :param catch_info:
    :param left_order:
    :return:
    '''
    count = 0
    if '*' not in left_order:
        noun_line = left_order.split('from')[0][4:].split(',')
        noun_point = [i.strip() for i in noun_line]
        for row_info in catch_info:
            for noun in noun_point:
                    noun_index = noun_list.index(noun)
                    print(noun, row_info[noun_index])
                    count += 1
    else:
        print(catch_info)
    count = len(catch_info)
    print_login('本次共涉及%s条信息'%(count), 'info')


def add_info(catch_info,left_order):
    '''
    add staff_table Alex Li,25,134435344,IT,2015‐10‐29
    :param catch_info: [['1', 'Alex Li', '22', '13651054608', 'IT', '2013‐04‐01\n'], ['3', 'Rain Wang', '21', '13451054608', 'IT', '2017‐04‐01\n'], ['4', 'Mack Qiao', '44', '15653354208', 'Sales', '2016‐02‐01\n'], ['5', 'Rachel Chen', '23', '13351024606', 'IT', '2013‐03‐16\n'], ['6', 'Eric Liu', '19', '18531054602', 'Marketing', '2012‐12‐01\n'], ['7', 'Chao Zhang', '21', '13235324334', 'Administration', '2011‐08‐08\n'], ['8', 'Kevin Chen', '22', '13151054603', 'Sales', '2013‐04‐01\n'], ['9', 'Shit Wen', '20', '13351024602', 'IT', '2017‐07‐03\n'], ['10', 'Shanshan Du', '26', '13698424612', 'Operation', '2017‐07‐02']]
    :param left_order:  add staff_table Alex Li,25,134435344,IT,2015‐10‐29
    :return:
    '''
    info_list = [i.strip() for i in ''.join([i.strip() for i in left_order.split('staff_table')[1:]]).split(',')]
    if info_list[2] not in data['phone']:
        data['staff_id'].append(str(int(data['staff_id'][-1]) + 1))
        data['name'].append(info_list[0])
        data['age'].append(info_list[1])
        data['phone'].append(info_list[2])
        data['dept'].append(info_list[3])
        #info_list[4] = '%s\n'%(info_list[4])
        data['enroll_date'].append('%s\n'%(info_list[4]))
        file_save()
    else:
        print('电话号码有重复，不可注册！', 'error')


def del_info(catch_info,left_order):
    '''
    del from staff where staff_id=3
    删除的语法要写staff_id,写id是找不到对应函数的
    :param catch_info:
    :param left_order: del from staff where id=3
    :return:
    '''
    count = 0
    print('你将要删除:', catch_info)
    for info in catch_info:
        index = data['staff_id'].index(info[0])
        for noun in noun_list:
            del data[noun][index]
        count += 1
    print_login('本次共涉及%s条信息'%(count), 'info')
    file_save()


def update_info(catch_info,left_order):
    '''
    update staff_table set dept=Market where dept = IT
    update staff_table set age=25 where name = Alex Li
    :param catch_info:
    :param left_order:
    :return:
    '''
    count = 0
    noun = [i.strip() for i in left_order.split('set')[1].strip().split('=')]
    print('你要修改 %s 为 %s'%(noun[0], noun[1]))
    for info in catch_info:
        index_id = data['staff_id'].index(info[0])
        for n in noun_list:
            data[noun[0]][index_id] = noun[1]
        count += 1
    print_login('本次共涉及%s条信息'%(count), 'info')
    file_save()


data = file_operation(file_name)  # 获得文件内容
catch_order()   # 执行操作