1 import os 2 def print_pos(key_dict): 3 keys = key_dict.keys() 4 keys = sorted(keys) # 由于字典是无序的，我们这里对行数进行排序 5 for each_key in keys: 6 print('关键字出现在第 %s 行，第 %s 个位置。' % (each_key, str(key_dict[each_key]))) 7 8 def pos_in_line(line, key): 9 pos = [] 10 begin = line.find(key) #查找关键字在每一行中的位置 赋值给begin 11 while begin != -1: 12 pos.append(begin + 1) # 用户的角度是从1开始数 13 begin = line.find(key, begin+1) # 从下一个位置继续查找 14 15 return pos 16 17 def search_in_file(file_name, key): 18 f = open(file_name) #打开文本文件 19 count = 0 # 记录行数 20 key_dict = dict() # 定义字典，用户存放key所在具体行数对应具体位置 21 22 for each_line in f: #遍历文本文件中的每一行 23 count += 1 #行数记录 24 if key in each_line: #如果关键字在某一行中 25 pos = pos_in_line(each_line, key) # key在每行对应的位置 调用pos_in_line函数 传入这个行 和 关键字 26 key_dict[count] = pos 27 28 f.close() 29 return key_dict 30 31 def search_files(key, detail): #第一步调用search_file函数 把 key, detail 参数传入进去 32 all_files = os.walk(os.getcwd()) #遍历当前目录下的所有文档，返回1目录，2包含路径，3包含文件 将返回的结果赋值给all_files 33 txt_files = [] #定义列表txt_files 34 35 for i in all_files: #遍历all_files 注意这里的i为单次遍历的三元组 36 for each_file in i[2]: #遍历三元组的第三个值，即遍历文件，这里的each_file 为单次遍历的文件 37 if os.path.splitext(each_file)[1] == '.txt': # 根据后缀判断是否文本文件 38 each_file = os.path.join(i[0], each_file) #如果each_file 是文本文件 则将该文件的路径名称以及文件名称结合 并赋值给each_file 39 txt_files.append(each_file) #列表txt_files 中追加each_file 此时的each_file为全路径含文件名 列表中追加的文本文件全路径 40 41 for each_txt_file in txt_files:#在列表txt_file中 遍历所有的文本文件 each_txt_file为单次遍历的文本文件名 42 key_dict = search_in_file(each_txt_file, key)#调用seach_in_file函数，传入单次遍历的文本文件名each_txt_file和用户输入的关键字key 43 if key_dict: 44 print('================================================================') 45 print('在文件【%s】中找到关键字【%s】' % (each_txt_file, key)) 46 if detail in ['YES', 'Yes', 'yes']: 47 print_pos(key_dict) 48 49 key = input('请将该脚本放于待查找的文件夹内，请输入关键字：') #接受用户输入的关键字 key为用户输入的结果 50 detail = input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：' % key) #请示用户是否打印 detail为请示结果 51 search_files(key, detail) #调用函数search_files 传入关键字 key, detail