Python查找字符串高亮显示

#!/usr/bin/env python

#_*_ coding:utf-8 _*_

user_file = 'userlist.txt'

info = raw_input('Input the info to search: ')

f = file(user_file,'rb')

count = 0

for line  in f.xreadlines():

    line = line.strip('\n').split()

    if info in line:

        line = ["\033[31m %s \033[0m" % info if i == info else i for i in line]

        for i in line:

            print i,

        print '\n'

        count += 1

if count == 0:

    print 'the info is not exists in your txt'

else:

    print 'match %d lines'%(count)

f.close()

~