python-查询员工信息表

python查询员工信息表

基本要求：

用户可以模糊查询员工信息

显示匹配了多少条，匹配字符需要高亮显示

代码：

1. #!/usr/env python
2. #coding:utf-8
3. import time
4. def BreakFlag(): #用于设置标志位
5. while True:
6. Break_Flag=raw_input('\t\t\t是否继续？（y/n）:')
7. if Break_Flag=='y' or Break_Flag=='n':
8. return Break_Flag
9. else:
10. print '''''\t\t输入错误，请重新输入!'''
11. Info_File=open(r'information.txt','r') #只读方式打开员工信息表文件
12. Employee_Info=Info_File.readlines()  #生成一个员工信息列表，列表中元素为员工信息字符串
13. Info_File.close() #关闭员工信息表文件
14. Break_Flag=''
15. print '''''\n\t\t\t\t\033[34;1m您好，欢迎来到员工信息查询系统！\033[0m\n'''
16. while Break_Flag!='n':
17. while True:
18. Search_Info=raw_input('''''\n\t\t\t请输入您需要查询的信息：''')
19. if len(Search_Info)>2: #判断输入字符长度，小于3个字符，则需要重新输入
20. break
21. else:
22. print '''''\n\t\t\t\033[31;1m您输入信息太少，请重新输入!\033[0m\n'''
23. count_number=0
24. Search_Info_List=[]
25. for i in Employee_Info:
26. if i.count(Search_Info)>0:
27. #将搜索到的字符串进行替换，并追加到Search_Info_List列表中
28. Search_Info_List.append(i.replace(Search_Info,'''''\033[42;31;1m%s\033[0m''' % Search_Info))
29. count_number+=1 #一条记录匹配成功
30. #count_number+=i.count(Search_Info)#统计每次搜寻到的字符串的个数，并进行累加
31. if count_number>0:
32. #若搜寻到字符串，将统计到的字符串总个数和包含字符串的列表进行屏幕打印
33. print '''''\n\t\t\t共查询到：\033[31;1m %s \033[0m条信息!\n''' % count_number
34. for i in Search_Info_List:
35. print i
36. else:
37. print '''''\n\t\t\t\033[31;1m没有您查找的信息!\033[0m\n'''
38. Break_Flag=BreakFlag()
39. for i in range(3):
40. print '''''\n\t\t\t\033[31;5m谢谢使用员工信息查询系统，%s秒后，退出系统\033[0m''' % (3-i)
41. time.sleep(1)
42. exit('\n\t\t\t\t您已经退出员工查询系统！')
43. #import sys
44. #sys.exit('\n\t\t\t\t您已经退出员工查询系统！')

 

假定员工信息存放在 information.txt文件中，如下：

song@ubuntu:~$ more information.txt
name:marry  age:23  sex:man  job:IT
name:song dada  age:23  sex:man  job:HR
name:zhang sanfeng  age:23  sex:man  job:IT
name:zhang wuji  age:23  sex:man  job:accountant
name:ling huchong  age:23  sex:man  job:IT
name:xiaoping  age:23  sex:man  job:manager
name:li yuanfang  age:23  sex:man  job:IT
name:feng feng  age:23  sex:man  job:boss
运行程序，示例如下：