今天写作业，明天后天要在外旅游
写作业写了7个小时。
1 def read_file_as_dict(where): 2 staff_dict = {} 3 f = open('%s' % where, mode="r+", encoding='utf-8') 4 data = f.read() 5 f.close() 6 row = data.strip().split('\n') 7 for staff in row: 8 staff_info = staff.split(',') # 每一条是这样的: ['1', 'Alex Li', '22', '13651054608', 'IT', '2013-04-01'] 9 staff_dict[staff_info[3]] = staff_info 10 return staff_dict 11 12 13 def save_back_to_file(where): 14 s = staff_dict.values() 15 s2 = '' 16 for staff in s: 17 for info in staff: 18 s2 = s2 + info + ',' 19 s2 = s2.rstrip(',') + '\n' 20 s2.rstrip('\n') 21 22 f = open('%s' % where, mode="w", encoding='utf-8') 23 f.writelines(s2) 24 f.close() 25 26 27 def find_staff(how): 28 # 首先先筛选，输出需要打印的员工名单，即在staff_list中的index号 29 # how[0]为目标比较项，如age，name等，how[1]为比较符，如大于小于等于等等，how[2]为去比较的值，如年龄22，电话号码等 30 selected_staff = [] 31 if how[0] == 'staff_id' or how[0] == 'age': # 如果比较的项是员工ID或者年龄这类数字 32 if how[1] == '>': 33 for key, value in staff_dict.items(): 34 if int(value[title_list.index(how[0])]) > int(how[2]): # 筛选完毕 35 selected_staff.append(key) # 输出符合条件的所有staff的index的列表selected_staff_index 36 return selected_staff 37 38 elif how[1] == '<': 39 for key, value in staff_dict.items(): 40 if int(value[title_list.index(how[0])]) < int(how[2]): # 筛选完毕 41 selected_staff.append(key) 42 return selected_staff 43 44 elif how[1] == '=': 45 for key, value in staff_dict.items(): 46 if int(value[title_list.index(how[0])]) == int(how[2]): # 筛选完毕 47 selected_staff.append(key) 48 return selected_staff 49 50 elif how[0] == 'enroll_date' and how[1] == 'like': # 比较入职日期，日期目前只有like+年份的操作 51 for key, value in staff_dict.items(): 52 if value[title_list.index(how[0])].startswith(how[2]): # 筛选完毕 53 selected_staff.append(key) 54 return selected_staff 55 56 elif how[0] == 'name' and how[1] == '=': # 比较姓名 57 target_name = ' '.join(how[2:]) # 把已经拆分成两个列表元素的姓与名合并起来 58 for key, value in staff_dict.items(): 59 if value[title_list.index(how[0])] == target_name: # 筛选完毕 60 selected_staff.append(key) 61 return selected_staff 62 63 elif how[0] in title_list and how[1] == '=': # 如果条件是除上述以外的项目，如比较电话、部门 64 for key, value in staff_dict.items(): 65 if value[title_list.index(how[0])] == how[2]: # 筛选完毕 66 selected_staff.append(key) 67 return selected_staff 68 69 else: 70 print('对不起，您的输入有误，请重新输入！') 71 72 73 def print_info(what, list): # 如print_info('name,age', [1, 3, 8])或者print_info('*', [1, 3, 8]) 74 75 # 首先是识别what参数传入进来需要显示的项目名称， 76 index_list = [] 77 s = '' 78 global title_list # 呼叫全局变量，用作后续对比 79 subset_flag = True # 标识符，查看用户输入信息是否合法 80 81 if what == '*': # 输入为*号，即打印所有信息 82 index_list = range(len(title_list)) 83 84 else: 85 what = what.split(',') # 如果传进去what是'age, name'的话，这里就变成了['age', 'name'] 86 for element in what: # 先看一下用户传入的需打印的title是不是都是可打印的title，与title_list进行匹配 87 if element in title_list: 88 index_list.append(title_list.index(element)) # 将需打印的title在title_list当中的索引做成一个新列表，供打印使用 89 else: 90 subset_flag = False 91 92 if subset_flag: # 如果没有问题，都是合法的title 93 for i in selected_staff: # 将每个要打印的员工循环 94 for index in index_list: # 将每个员工的每个要打印的项目循环 95 s = s + title_list[index] + ' ' + staff_dict[i][index] + ' ' # 将同一个员工的信息加在同一行 96 97 s = s + '\n' # 每个员工循环完毕，换行 98 99 print(s) # 打印全部需要打印的信息 100 else: 101 print('对不起，您的输入有误，请重新输入！') 102 103 104 def add_staff(words, staff_dict): 105 new_staff_id = str(len(staff_dict) + 1) # 新的staff_id就是现有的数量+1，为了后面写数据方便，格式转成字符串 106 r_detailed_info = words[3].split(',') # 将除了姓以外的所有数据处理，结果如[Li', '25', '134435344', 'IT', '2015-10-29'] 107 r_detailed_info[0] = words[2] + ' ' + r_detailed_info[0] # 将姓氏加上 108 r_detailed_info.insert(0, new_staff_id) # 插入新的staff_id 109 if r_detailed_info[3] in staff_dict: 110 print('对不起，您想要添加的手机号已经存在，无法重复添加！') 111 else: 112 staff_dict[r_detailed_info[3]] = r_detailed_info # 将新的员工信息，手机号作为键，所有信息的列表作为值，插入字典 113 114 save_back_to_file(words[1]) 115 print('新员工信息%s，共1条已加入系统' % r_detailed_info) 116 117 118 def del_staff(selected_staff, staff_dict): 119 for staff in selected_staff: # 将查找到的员工信息删除 120 deleted_staff = staff_dict.pop(staff) 121 122 id_bigger_than_del_list = find_staff(['staff_id', '>', words[6]]) # 删除后只影响它后面的staff_id，需要-1 123 for key in id_bigger_than_del_list: # 更新staff_id 124 staff_dict[key][0] = str(int(staff_dict[key][0]) - 1) 125 126 save_back_to_file(words[2]) 127 print('老员工信息%s，共1条已删除' % deleted_staff) 128 129 130 def update_staff(selected_staff, staff_dict): 131 global title_list 132 for staff in selected_staff: # 将查找到的员工信息更新 133 staff_dict[staff][title_list.index(words[3])] = words[5] 134 save_back_to_file(words[1]) 135 print('所有%s为%s的员工，%s已经变更为%s，共变更%s条' % (words[7], words[9], words[3], words[5], len(selected_staff))) 136 137 138 # --------------------------------------------函数区结束，下面为实现代码区--------------------------------------------- 139 title_list = ['staff_id', 'name', 'age', 'phone', 'dept', 'enroll_date'] 140 141 while True: 142 search_instruction = input('please type in your searching instruction:').strip().replace('"', '') 143 words = search_instruction.split( 144 ' ') # words = ['find', 'name,age', 'from', 'staff_table', 'where', 'age', '>', '22'] 145 146 if words[0] == 'find' and 8 <= len(words) <= 9: # 命令的第一个字符串识别是哪一类操作，如果为find，开始进行find功能，将关键参数传入函数，函数命令元素一般为8个，查姓名为9个 147 148 staff_dict = read_file_as_dict(words[3]) # 调用read_file函数读取文档，生成对应的列表文件 149 # 列表类似staff_list = [['1', 'Alex Li', '22', '13651054608', 'IT', '2013-04-01'], 150 # ['2', 'Jack Wang', '28', '13451024608', 'HR', '2015-01-07']] 151 152 selected_staff = find_staff(words[5:]) # words[5:]为条件str语句的列表['age', '>', '22'] 153 # 输出的这个selected_staff的列表内容为[1, 2, 3, 5]等等的被筛选后需要打印信息的员工在staff_list里面的索引 154 155 # print(selected_staff) 156 157 print_info(words[1], selected_staff) # 然后是利用传入进来的list，打印list里每个人的what项目的信息，每人一行 158 159 print('这条语句查询了%s条' % len(selected_staff)) 160 161 elif words[0] == 'add' and len( 162 words) == 4: # 姓占1个长度，['add', 'staff_table', 'Alex', 'Li,25,134435344,IT,2015-10-29'] 163 staff_dict = read_file_as_dict(words[1]) # 将change_file函数的返回值字典赋予staff_dict 164 add_staff(words, staff_dict) 165 166 elif words[0] == 'del' and len(words) == 7: # ['del', 'from', 'staff_table', 'where', 'id', '=', '3']，命令长度为7 167 168 staff_dict = read_file_as_dict(words[2]) 169 words[4] = 'staff_id' # 将id改为staff_id以配合功能查询 170 selected_staff = find_staff(words[4:]) # 用find_staff功能进行查询 171 if not selected_staff: # 如果selected_staff值为空， 172 print('对不起，系统没有查询到，请重新查询！') 173 else: # 如果selected_staff值不为空， 174 del_staff(selected_staff, staff_dict) 175 176 elif words[0] == 'update' and len( 177 words) == 10: # 如['update', 'staff_table', 'set', 'dept', '=', 'Market', 'where', 'dept', '=', 'IT'] 178 staff_dict = read_file_as_dict(words[1]) 179 selected_staff = find_staff(words[7:]) 180 if not selected_staff: # 如果selected_staff值为空， 181 print('对不起，系统没有查询到，请重新查询！') 182 else: # 如果selected_staff值不为空， 183 update_staff(selected_staff, staff_dict) 184 185 elif search_instruction == 'q': 186 exit('感谢您的使用，再见！') 187 188 else: 189 print('对不起，系统无法识别您的命令，请重新输入！')
staff_table:
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