|
楼主 |
发表于 2021-10-16 18:02:22
|
显示全部楼层
另外附上小甲鱼代码,望有大神可以指导一下,这些代码是怎么一个逻辑写出来的呢
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) #查找参数在字符串中第一次出现的位置,找到返回索引值,找不到返回-1
while begin != -1: #也就是找到了
pos.append(begin + 1) # 用户的角度是从1开始数,这里比较细节
begin = line.find(key, begin+1) # 从下一个位置继续查找,这里也要对find函数有一定的了解!
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在每行对应的位置,又分析得知,这里pos是一个列表
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': # 根据后缀判断是否文本文件,splitext是分离文件名和拓展名的
each_file = os.path.join(i[0], each_file) #i[0]是指定路径,这里是把路径和each_file平起来
txt_files.append(each_file) #把当前目录里面所有的txt文件用列表装起来
for each_txt_file in txt_files:
key_dict = search_in_file(each_txt_file, key) #调用前面的函数,开始寻找txt里面的关键字
if key_dict: #只要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)
|
|