python学习--2_2

user_info 信息存储文件内容

zhangsan    18612348765    sanic    [e-mail]      23
lishun      19235648308    shuns    [e-mail]         21
liusi       19283638273    sics     [e-mail]         30
feifei       29386201834    jalen    [e-mail]      32
huangdd     18273672822    duoduo   [e-mail]        25
qiuqiu      17356489382    qiuqiu   [e-mail]       29

代码块

#!/usr/bin/env python
# _*_ coding: utf-8 _*_

'''员工信息表
1.用户可以模糊查询员工信息
2.显示匹配了几条，匹配字符需要高亮显示
'''

user_file = open('user_info', 'r')
exit_flag = True
count = 0

while exit_flag:
    search = raw_input('输入你要查询的关键字(至少输入2位)：')
    if len(search) >= 2:
        search_info = []
        for user in user_file.readlines():
            if search in user:
                search_info.append(user.replace(search, '''\033[42;31;1m%s\033[0m''' %search))
                count += 1
        for i in search_info:
            print i
        print '总共匹配到%s条信息。' %count
    else:
        print '输入的关键字太少，请至少输入两位！'
        continue
    exit_flag = False

user_file.close()