小甲鱼Python 第30讲课后习题看不懂

树精树蛙

小甲鱼，30讲课后习题最后一题调用太复杂看不懂

import os                           #读取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)     #从下一个位置继续查找，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)

费劲了好久加了注释，可是发现第一个函数全程没有被调用过，而且运行报错。。。

树精树蛙

报错如下：

请将该脚本放于待查找的文件夹内，请输入关键字：从明天起
请问是否需要打印关键字【从明天起】在文件中的具体位置（YES/NO）：yes
Traceback (most recent call last):
  File "C:/Users/think/AppData/Local/Programs/Python/Python35-32/1.py", line 57, in <module>
    search_files(key, detail)
  File "C:/Users/think/AppData/Local/Programs/Python/Python35-32/1.py", line 47, in search_files
    key_dict = search_in_file(each_txt_file, key) #调用函数
  File "C:/Users/think/AppData/Local/Programs/Python/Python35-32/1.py", line 26, in search_in_file
    for each_line in f:            #逐行提取
UnicodeDecodeError: 'gbk' codec can't decode byte 0x99 in position 261: illegal multibyte sequence

而且感觉逻辑上怪怪的

树精树蛙

还有41行看不懂

子非吾

同样是遇到运行出错。而且搞不懂第二个函数line在哪儿被赋值了

keydnal_aaron

这个测试的文本里面是英文字符串，如果环境不同，注意下文本内容的编码方式，我的编程环境是centos7+python3.6.4

from os import walk,getcwd
from os.path import join

def search_file():
    '''
    查找当前目录底下的所有文件！
    return返回的结果为列表：‘/路径/文件.txt’
    '''
    file_list=[]
    for each_dir_file in walk(getcwd()):
        if each_dir_file[2] != []:
            for each_file in each_dir_file[2]:
                if each_file[-4:] == '.txt':
                    temppath=join(each_dir_file[0],each_file)
                    file_list.append(temppath)
    return file_list

def chazhao(filename,zfc):
    '''
    传入两个参数,文件名,要查找字符串
    查找匹配字符串的行
    生成结果以字典形式返回
    如果文件没有匹配字符串的行，返回空字典
    :param filename:
    :param zfc:
    :return:
    '''
    count1 = 0
    dict_line_weizhi = {}
    file = open(filename)
    for each_line in file:
        count1 += 1
        if zfc in each_line:
            weizhi=[]
            begin_zfc = each_line.find(zfc)
            while begin_zfc != -1:
                weizhi.append(begin_zfc)
                begin_zfc = each_line.find(zfc, begin_zfc + 1)
            dict_line_weizhi.setdefault(count1,weizhi)
        else:
            dict_line_weizhi={}
    file.close()
    return dict_line_weizhi

def zfc_line_weizhi(zfc,daying):
    '''
    接收用户输入需要查找的字符串zfc
    接收用户输入是否确认查询daying
    确认查询，返回查询结果
    确认不查询，退出程序
    :param zfc:
    :param daying:
    :return:
    '''
    if (daying == 'YES') or (daying == 'yes') or (daying == 'Yes'):
        file_list = search_file()
        for each_file in file_list:
            result = chazhao(each_file, zfc)
            if result != {}:
                print('在文件' + '[' + each_file + ']' + "中找到关键字" + '[' + zfc + ']')
                for each_key in result.keys():
                    print('关键字出现在第' + str(each_key) + '行,第' + str(result[each_key]) + '个位置！')
    else:
        print('退出查询!')


if __name__ =='__main__':
    zfc = input('请将该代码放于待查找的文件夹内，请输入关键字:').strip()
    daying = input('请问是否打印关键字' + zfc + '在文件中的具体位置（YES/NO）:').strip()
    zfc_line_weizhi(zfc,daying)