关于30讲最后一题

540507521

import os


def search_files(key,detail):
    all_files = os.walk(os.getcwd())   # 遍历文件夹里的所有目录并返回元组
    txt_files = []

    for i in all_files:
        for each_file in i[2]:   # i[2]是元组里第3个元素
            if os.path.splitext(each_file)[1]:
                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_key(key_dict)


def search_in_file(file_name,key):
    file = open(file_name)
    count = 0    # 记录文件行数
    key_dict = dict()

    for each_line in file:
        count += 1
        if key in each_line:
            key_each_line = key_in_line(each_line,key)
            key_dict[count] = key_each_line
    file.close()
    return key_dict


def key_in_line(line,key):
    key_each_line = []
    begin = line.find(key)
    while begin != -1:
        key_each_line.append(begin + 1)  # 用户的角度是从1开始数
    return key_each_line


def print_key(key_dict):
    keys = key_dict.keys()
    keys = sorted(keys)
    for each_key in keys:
        print('关键字出现在第 %s 行，第 %s 个位置。' % (each_key,str(key_dict[each_key])))


key = input('请将该脚本放于待查找的文件夹内，请输入关键字：')
detail = input('请问是否需要打印关键字【%s】在文件中的具体位置(YES/NO)：' % key)
search_files(key,detail)


代码运行后为啥会出错啊

Traceback (most recent call last):
  File "E:\python\在文件中寻找关键字.py", line 54, in <module>
    search_files(key,detail)
  File "E:\python\在文件中寻找关键字.py", line 15, in search_files
    key_dict = search_in_file(each_txt_file,key)
  File "E:\python\在文件中寻找关键字.py", line 28, in search_in_file
    for each_line in file:
UnicodeDecodeError: 'gbk' codec can't decode byte 0x90 in position 52: illegal multibyte sequence

Stubborn

gbk应该是文本编码有问题哦