|
20鱼币
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 = []
begin = line.find(key)
while begin != -1:
pos.append(begin + 1) # 用户的角度是从1开始数
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
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)
- """
- Created on Mon Jul 20 22:10:11 2020
- 用户输入关键字,查找当前文件夹内(如果当前文件夹内包含文件夹,
- 则进入文件夹继续搜索)所有含有该关键字的文本文件(.txt后缀),
- 要求显示该文件所在的位置以及关键字在文件中的具体位置
- (第几行第几个字符)
- @author: 小甲鱼
- """
- 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 = []
- begin = line.find(key)
- while begin != -1:
- pos.append(begin + 1) # 用户的角度是从1开始数
- begin = line.find(key, begin+1) # 从下一个位置继续查找
- return pos
- def search_in_file(file_name, key):#在单个txt中搜索
- 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
-
- f.close()
- return key_dict
- def search_files(key, detail): #将输入关键字和yes传入函数
- 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) #调用,返回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)
复制代码
|
|