import os
def print_pos(key_dict):#定义函数打印第几行第几个位置
keys = key_dict.keys()#将键赋值
keys = sorted(keys) # 由于字典是无序的,我们这里对行数进行排序
for each_key in keys:
print('关键字出现在第 %s 行,第 %s 个位置。' % (each_key, str(key_dict[each_key])))
def pos_in_line(line, key):#查找关键字在单行中的位置
pos = []#存放key在line中的位置
begin = line.find(key)#返回关键字的索引值,如果未匹配到的话,会返回-1
while begin != -1:#如果该行包含key
pos.append(begin + 1) # 用户的角度是从1开始数(因为python从0开始)
begin = line.find(key, begin+1) # 从下一个位置继续查找
return pos
def search_in_file(file_name, key):#统计单一文本文件内的关键字位置
f = open(file_name)
count = 0 # 记录行数
key_dict = dict() # 字典,用户存放key所在具体行数对应具体位置
for each_line in f:
count += 1
if key in each_line:
pos = pos_in_line(each_line, key) # key在每行对应的位置
key_dict[count] = pos#将pos作为key_dict[count]的键值
f.close()
return key_dict#返回哪一行哪个位置
def search_files(key, detail):#统计关键字在文建中的位置
all_files = os.walk(os.getcwd())#返回当前工作目录中的三元组(路径,包含的目录,包含的文件)
txt_files = []#储存合成的路径
for i in all_files:#遍历三元组
for each_file in i[2]:#遍历三元组中的包含文件
if os.path.splitext(each_file)[1] == '.txt': # 根据后缀判断是否文本文件
each_file = os.path.join(i[0], each_file)#将三元组中的路径和文件合成一个路径
txt_files.append(each_file)#将合成的路径添加
for each_txt_file in txt_files:#遍历合成的路径
key_dict = search_in_file(each_txt_file, key)#调用上一个函数
if key_dict:#如果路径内有关键字,则打印出
print('================================================================')
print('在文件【%s】中找到关键字【%s】' % (each_txt_file, key))
if detail in ['YES', 'Yes', 'yes']:
print_pos(key_dict)
key = input('请将该脚本放于待查找的文件夹内,请输入关键字:')
detail = input('请问是否需要打印关键字【%s】在文件中的具体位置(YES/NO):' % key)
search_files(key, detail)
这是我写的注释,望能够指正?
|