#!/usr/bin/env python # -*- coding:utf-8 -*- # __author: 金鑫 # 2017/8/8 """ 1,Alex,22,13651054608,IT 2,Egon,23,13304320533, Tearcher 3,沛齐,23,13304320533, Tearcher 4,金鑫,23,13304320533, Tearcher """ import os #引用OS模块 user_dic = {'username': None, 'password': None, #创建了一个用户信息字典 'login': True } flag = False name_list = ['id', 'name', 'age', 'phone', 'job'] #创建列名列表 check_conditions = ['>', '<', '=', 'like'] #创建筛选条件列表 def auth(func): #认证装饰器 def wrapper(*args, **kwargs): with open('user-pwd', 'r', encoding='utf8') as f: #r模式打开账号密码文件 user = eval(f.read()) #user-pwd文件是个字典,用eval取出来 count = 0 #计数器刚开始为0 while count < 3: #最多登录三次 username = input('请输入用户名:').strip() #让用户输入账号 password = input('请输入密码:').strip() #让用户输入密码 if user.get(username) and password == user[username]: #如果用户的账号密码吻合进入下面操作(没有也不报错) user_dic['username'] = username #更改用户信息字典 user_dic['password'] = password #更改用户信息字典 user_dic['login'] = True #更改用户信息字典 ret = func(*args, **kwargs) #执行函数 return ret #将返回值返回给函数的调用者 else: #账号密码输入错误走这里 count += 1 #计数器加1 if count == 3: #如果三次都输错 print('用户名密码不正确,您已经没有输入机会,请重新登录') #打印内容 else: #输错次数不到三次 print('用户名密码不正确,您还有%s输入机会' % (3 - count)) #打印内容 return wrapper #装饰器闭包 def get_last_id(): #获取id表id """ 此函数是获取id表中的id :return: 返回值是最后的id,如果第一次给员工信息表添加数据,id为0 """ with open('id表', encoding='utf-8') as f: #r模式打开id表文件 list_id = f.read() #读取id表,并赋值给 list_id return int(list_id) if list_id else 0 #如果id表有内容就返回整数的id,没有就返回0 def add(): #增加函数 """ 此函数是为员工信息表添加新的数据,关键点在添加新数据之后,你的id表的id会随之改变。 :return: None """ old_last_id = get_last_id() #执行获取id表的函数,并赋值给old_last_id input_info = input('按照下列格式输入信息:' '姓名,年龄,电话,职业').strip() #让用户输入 new_last_id = str(int(old_last_id)+1) #让id表的数字加1 add_info = new_last_id + ',' + input_info #给用户输入的信息添加序号 with open('员工信息表', 'a', encoding='utf-8') as f2: #a模式打开员工信息表 # 此三元运算符是判断员工信息表中如果是第一次添加内容直接添加,否则 换行添加。 f2.write(add_info) if old_last_id == 0 else f2.write(' '+add_info) print('您已成功添加,请继续操作') #打印操作成功 with open('id表', 'w', encoding='utf-8') as f3: #w模式打开id表 f3.write(new_last_id) #写入新的id号 def pop(): #删除函数 """ 此函数的功能是删除员工信息表中的数据,关键点: 1,删除的id不是员工信息表最后的id,则不涉及id表中的id更改。 2,删除的id是员工信息表最后的id,则删除之后必须将id表中的id自增1,以便下次添加数据从新的id开始。 3,flag的设置是因为我在打开员工信息表的过程中,不能删除此员工信息表, 所以设置标志位,等操作完成之后,在删除原文件,重新命名新文件。 :return: """ pop_id = input('请填写你想删除的信息id') #获取用户想删除的id号 with open('员工信息表', encoding='utf-8') as f: #r模式打开员工信息表 for line in f: #循环员工信息表的每一行 line_list = line.strip().split(',') #把每一行以','切割,并赋值 if pop_id == line_list[0]: #如果就是用户要删除的行,执行下面操作 # 判断删除的id是否在文件中,设置标志位,等文件操作完成后,在删除文件,重命名文件。 global flag #给flag定义全局变量 flag = True last_id = get_last_id() #获取id表的id if pop_id == last_id: #如果要删除的ID就是最后一行 with open('id表', 'w', encoding='utf-8') as f3: #w模式打开id表 f3.write(str(int(pop_id)+1)) #给id表加1 with open('员工信息表', encoding='utf-8') as f1, open('员工信息表.bak', 'w', encoding='utf-8') as f2: #r模式打开员工信息表,w模式打开员工信息表.bak for line1 in f1: #循环员工信息表的每一行 f2.write(line1) if pop_id != line1.strip().split(',')[0] else f2.write('') #如果删除id不等于这一行就写入备份文件,如果这行就是要删除的那一行就写一个'' break #跳出循环 else: #如果输入的序号没有在员工信息表里找到 print('您输入的序号不存在,请重新输入') #输入内容 if flag: os.remove('员工信息表') #删除源文件 os.rename('员工信息表.bak', '员工信息表') #给新文件改名 def update(): """ 此函数的功能:修改员工信息 语法:set 列名=“新的值” where 条件 先用where查找对应人的信息,再使用set来修改列名对应的值为“新的值” 关键点:check_conditions, name_list 这两个列表的运用 check_conditions 为 where 关键字后面的比较条件:> < = like name_list 为员工信息表的列名,'id', 'name', 'age', 'phone', 'job' 索引与员工信息表每行信息的每列一一对应。 思路: 1,先通过 where 分割,得到content(如 set name = 太白),cond(如 id > 3) 2,循环 check_conditions(> < = like) 通过 条件 分割 得到结果:name是id,key是 > , cond 是 3 3,对 content进行操作,得到要修改的列名:column_name,要修改的值update_value 4,接下来就用到了文件改的思想,如果要修改文件,需要对将原文件读取,一行一行的写入新文件,将需要改动的行写入, 然后将原文件删除,新文件重命名成原文件,这样就做到了文件的改的操作。 5,关键点:check_dict 这是一个if条件的字典,这样设置是因为,不管条件为 < > = 或者like 他们只要满足条件,下面进行的操作是一样的,所以只是条件不同,那么根据key的不同,执行check_dict不同的条件, 然后进行相同的内容即可。 :return: None """ pass def check(): #查询函数 """ :param: 可以进行查询,支持三种语法: select 列名1,列名2,… where 列名条件 支持:大于小于等于,还要支持模糊查找。 示例: check_info = select name, age where age>22 select * where job=IT select * where phone like 133 关键点:check_conditions, name_list 这两个列表的运用 check_conditions 为 where 关键字后面的比较条件:> < = like name_list 为员工信息表的列名,'id', 'name', 'age', 'phone', 'job' 索引与员工信息表每行信息的每列一一对应。 思路: 1,先通过 where 分割,得到content(如 select name,id),cond(如 id > 3) 2,循环 check_conditions(> < = like) 通过 条件 分割 得到结果:name是id,key是 > , cond 是 3 3,对 content进行操作,得到要修改的列名:column_name,要修改的值update_value 4,接下来读文件,而读文件分两种:select * (全部) 和select name,id,.. 部分,所以走两条路。 5,关键点:check_dict 这是一个if条件的字典,这样设置是因为,不管条件为 < > = 或者like 他们只要满足条件,下面进行的操作是一样的,所以只是条件不同,那么根据key的不同,执行check_dict不同的条件, 然后进行相同的内容即可。 :return: """ try: #用户输入错误内容不报错 check_info = input('请输入查询语句:').strip() #让用户输入查询语句 content, condition = check_info.split('where') #将用户的查询语句以where切割,分别赋值给content和condition # print(content,condition) if 'select' in content: #如果select在content中执行下面操作 for key in check_conditions: #循环创建的条件列表 if key in condition: #如果列表的内容在用户输入的条件中 index_check_conditions = check_conditions.index(key) #获得查询条件列表中条件的索引,并赋值index_check_conditions name, cond = condition.strip().split(key) #将筛选条件去除空格并以要比较的符号或like分割,将前后分别赋值给name和cond # content为 'select name,id,'或'select *,' # name 是 where后面的第一个关键字 # key 是比较运算符 # cond 是 实际需要比较的 关键字 # 例如:where id > 3 name是id,key是 > , cond 是 3 content = ''.join(content.split('select')[1:]) #将content以select分割并切片去除空字符串再转换成字符串 with open('员工信息表', encoding='utf-8') as f: #r模式打开员工信息表 for line in f: #逐行循环 if line.strip(): #如果这一行不为空 line_list = [i.strip() for i in line.strip().replace(',',',').split(',')] #将有内容的一行除空格,中文逗号换成英文逗号,分割逗号,去除空格并添加到列表 # 上面此举是为了解决add信息时,输入的有中文逗号,从而导致split出问题的问题。 # 例:line_list = ['3','Egon','23','13304320533','Tearcher'] index = name_list.index(name.strip()) #查找name在列名的索引,列名的顺序和员工信息表的表头一样,并赋值给index # index 查询条件的索引,比如你的查询条件是select * where id > 3 # name 就是 id 求出的index 就是id在name_list列表中的索引,他亦是line_list的id的索引 # 这样就可以方便比较了。 check_dict = { 0: int(line_list[index]) > int(cond), 1: int(line_list[index]) < int(cond), 2: line_list[index].strip() == cond.strip(), 3: cond.strip() in line_list[index].strip(), } #创建了一个筛选字典,字典的筛选条件和条件列表的顺序一样,也就是说key的索引就是字典的键 if check_dict[index_check_conditions]: #如果字典中有这个键值对 if content.strip() == "*": #如果查询语句为全部 print(line.strip()) #就去除空格打印这一行 else: #如果查询语句不为'*' if ',' in content.strip(): #如果查询语句有多个内容 select_name_list = content.strip().split(',') #将查询语句以逗号分割,并赋值给select_name_list select_name_list = [i for i in select_name_list if i != ''] #如果查询语句内容不为空就添加到列表里,并重新赋值给select_name_list str1 = '' #创建一个空字符串赋值 for names in select_name_list: #循环用户输入的查询语句列表 name_index = name_list.index(names.strip()) #寻找用户输入的查询语句在列名中的索引,并赋值给 name_index str1 = str1 + line_list[name_index] + ',' #根据索引查找之前分割后的每一行的列表中的内容,并隔开 print(str1) #打印用户要查询的内容 else: #只剩下用户要查找的单个条件 print(line_list[name_list.index(content)]) #直接根据索引去查找用户要查询的信息并打印 break else: #如果输入的条件不在列表里 print('您输入的不正确,请重新输入') #打印内容 else: #没有select print('您没有输入select或者select输入有误,请重新输入') #打印内容 except Exception: #不报错 print('您输入的查询语句有误,请重新输入。')#打印内容 def loginout():#写一个注销函数 print('感谢您登陆员工内部系统') user_dic['login'] = False #将登陆状态改为False @auth #装饰器 def main(): #主界面的函数 menu = """ 欢迎%s用户登录内部员工系统,请进行选择: 选项 1,增加员工信息 选项 2,删除员工信息 选项 3,更改员工信息 选项 4,查询员工信息 选项 5,退出内部员工系统 """ % (user_dic['username']) #创建一个打印信息的字符串 choice_dict = { '1': add, '2': pop, '3': update, '4': check, '5': loginout } #创建一个带函数名的字典 while user_dic['login']: #如果认证状态为True print(menu) #打印主菜单 option = input('请输入你的选项:').strip() #让用户输入选项 if option in choice_dict: #如果用户输入的内容在字典里 choice_dict[option]() #根据输入的内容执行对应的函数 else: print('请输入正确选项') #不在字典里就打印 if __name__ == '__main__': main() #执行函数