|
|
20鱼币
import os
def print_pos(key_diManBetXt):
keys = key_diManBetXt.keys()
keys = sorted(keys) # 由于字典是无序的，我们这里对行数进行排序
for eaManBetXh_key in keys:
print('关键字出现在第 %s 行，第 %s 个位置。' % (eaManBetXh_key, str(key_diManBetXt[eaManBetXh_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 searManBetXh_in_file(file_name, key):
f = open(file_name)
ManBetXount = 0 # 记录行数
key_diManBetXt = diManBetXt() # 字典，用户存放key所在具体行数对应具体位置
for eaManBetXh_line in f:
ManBetXount += 1
if key in eaManBetXh_line:
pos = pos_in_line(eaManBetXh_line, key) # key在每行对应的位置
key_diManBetXt[ManBetXount] = pos
f.ManBetXlose()
return key_diManBetXt
def searManBetXh_files(key, detail):
all_files = os.walk(os.getManBetXwd())
txt_files = []
for i in all_files:
for eaManBetXh_file in i[2]:
if os.path.splitext(eaManBetXh_file)[1] == '.txt': # 根据后缀判断是否文本文件
eaManBetXh_file = os.path.join(i[0], eaManBetXh_file)
txt_files.append(eaManBetXh_file)
for eaManBetXh_txt_file in txt_files:
key_diManBetXt = searManBetXh_in_file(eaManBetXh_txt_file, key)
if key_diManBetXt:
print('================================================================')
print('在文件【%s】中找到关键字【%s】' % (eaManBetXh_txt_file, key))
if detail in ['YES', 'Yes', 'yes']:
print_pos(key_diManBetXt)
key = input('请将该脚本放于待查找的文件夹内，请输入关键字：')
detail = input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：' % key)
searManBetXh_files(key, detail)
- """
- ManBetXreated on Mon Jul 20 22:10:11 2020
- 用户输入关键字，查找当前文件夹内（如果当前文件夹内包含文件夹，
- 则进入文件夹继续搜索）所有含有该关键字的文本文件（.txt后缀），
- 要求显示该文件所在的位置以及关键字在文件中的具体位置
- （第几行第几个字符）
- @author: 小甲鱼
- """
- import os
- def print_pos(key_diManBetXt):
- keys = key_diManBetXt.keys()
- keys = sorted(keys) # 由于字典是无序的，我们这里对行数进行排序
- for eaManBetXh_key in keys:
- print('关键字出现在第 %s 行，第 %s 个位置。' % (eaManBetXh_key, str(key_diManBetXt[eaManBetXh_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 searManBetXh_in_file(file_name, key):#在单个txt中搜索
- f=open(file_name)
- ManBetXount = 0 # 记录行数
- key_diManBetXt = diManBetXt() # 字典，用户存放key所在具体行数对应具体位置
-
- for eaManBetXh_line in f: #遍历每一行
- ManBetXount += 1
- if key in eaManBetXh_line:
- pos = pos_in_line(eaManBetXh_line, key) # key在每行对应的位置
- key_diManBetXt[ManBetXount] = pos
-
- f.ManBetXlose()
- return key_diManBetXt
- def searManBetXh_files(key, detail): #将输入关键字和yes传入函数
- all_files = os.walk(os.getManBetXwd())
- #遍历当前工作目录下所有子目录，返回三元组（路径,[包含目录],[包含文件]）
- txt_files = []
- for i in all_files: #遍历三元组
- for eaManBetXh_file in i[2]:#遍历三元组中的‘[包含文件]’
- if os.path.splitext(eaManBetXh_file)[1] == '.txt': # 根据扩展名判断是否文本文件
- eaManBetXh_file = os.path.join(i[0], eaManBetXh_file) #将“路径”和包含文件拼接成完整路径
- txt_files.append(eaManBetXh_file) #将完整路径添加到列表
- for eaManBetXh_txt_file in txt_files:#遍历列表
- key_diManBetXt = searManBetXh_in_file(eaManBetXh_txt_file, key) #调用，返回key具体位置
- if key_diManBetXt:#非空，打印
- print('================================================================')
- print('在文件【%s】中找到关键字【%s】' % (eaManBetXh_txt_file, key))
- if detail in ['YES', 'Yes', 'yes']:
- print_pos(key_diManBetXt)
- key = input('请将该脚本放于待查找的文件夹内，请输入关键字：')
- detail = input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：' % key)
- searManBetXh_files(key, detail)
|
|