python找人脚本_Python开发文件搜索脚本

#!/usr/bin/env python

#coding:utf-8

import os, sys

#判断位置参数

argv_num = len(sys.argv)

if argv_num != 3:

print ("请至少输入2位参数:")

print ("########################################")

print ("EXample: python %s \033[1;31mfilename file_path\033[0m" % sys.argv[0])

print ("########################################")

exit()

def search(root_dir, searc):

try:

#获取搜索路径的整个列表目录及文件名

for (abs_path, dirs, files) in os.walk(root_dir):

for filename in files:

if searc in filename:

#搜索文件名的索引位置变量

searc_changdu = len(searc)

find_qian_num = filename.find(searc)

find_zongchang_num = searc_changdu + find_qian_num

#列表分片前段高亮显示变量

qian_1 = filename[:searc_changdu]

hou_1 = filename[searc_changdu:]

#列表分片中段高亮显示变量

qian_2 = filename[:find_qian_num]

hou_2 = filename[find_qian_num:find_zongchang_num]

#列表分片后段高亮显示变量

zhong = filename[find_qian_num:find_zongchang_num]

hou_3 = filename[find_zongchang_num:]

#绝对路径及文件名高亮输出

if find_qian_num == 0:

print ("%s/\033[1;31m%s\033[0m%s" % (abs_path,qian_1,hou_1))

elif find_zongchang_num == len(filename):

print ("%s/%s\033[1;31m%s\033[0m" % (abs_path,qian_2,hou_2))

else:

print ("%s/%s\033[1;31m%s\033[0m%s" % (abs_path,qian_2,zhong,hou_3))

#用户中断执行(通常是输入^C)

except KeyboardInterrupt, e:

print ("中断执行")

search(sys.argv[2], sys.argv[1])

执行效果展示