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('对不起，系统无法识别您的命令，请重新输入！')