Поиск слова в документах на компьютере

import os
import os.path
import fileinput
 
from itertools import repeat
from collections import defaultdict
from pprint import pprint
 
 
top = '.'
results = defaultdict(list)
 
patterns = ['matplot', 'mpmath']
include = ['.py', 'txt']
 
openhook = fileinput.hook_encoded("utf8")
selection = lambda file:  file.endswith(tuple(include))
 
 
def hook_encoded(encoding=None, errors='strict'):
    if encoding:
        def openhook(filename, mode):
            return open(filename, mode, encoding=encoding, errors=errors)
    
    else:
        from cchardet import UniversalDetector
        
        def openhook(filename, mode):
            with open(filename, 'rb') as file:
                detector = UniversalDetector()
                for line in file:
                    detector.feed(line)
                    if detector.done: break
                detector.close()
                encoding = detector.result['encoding'] or 'utf8'
                # print(detector.result, filename)
            return open(filename, mode, encoding=encoding, errors=errors)
    
    return openhook
 
openhook = hook_encoded(errors='ignore')
 
for path, dirs, files in os.walk(top):
    # files = (file for file in files if file.endswith(tuple(include)))
    files = filter(selection, files)
    files = map(os.path.join, repeat(path), files)
    
    with fileinput.input(files, openhook=openhook) as f:
        for line in f:
            for pattern in patterns:
                if pattern in line:
                    fnl = (f.filename(), f.filelineno(), line.rstrip())
                    results[pattern].append(fnl)
 
pprint(results)

import os
import os.path
 
from itertools import repeat
from collections import defaultdict
from pprint import pprint
from cchardet import UniversalDetector
 
 
bufsize = 64 * 1024
 
top = '.'
results = defaultdict(list)
 
patterns = ['matplot', 'mpmath']
include = ['.py', 'txt']
 
selection = lambda file:  file.endswith(tuple(include))
 
 
def detect_encoding(filename):
    detector = UniversalDetector()
    with open(filename, 'rb') as file:
        for line in file:
            detector.feed(line)
            if detector.done: break
    detector.close()
    encoding = detector.result['encoding'] or 'utf8'
    # print(detector.result, filename)
    return encoding
 
 
for path, dirs, files in os.walk(top):
    # files = (file for file in files if file.endswith(tuple(include)))
    files = filter(selection, files)
    files = map(os.path.join, repeat(path), files)
    
    for filename in files:
        encoding = detect_encoding(filename)
        with open(filename, encoding=encoding, errors='ignore') as file:
            rightover = ''
            while True:
                buffer = file.read(bufsize)
                if not buffer:
                    break
                lines = buffer.split("\n")
                lines[0] = rightover + lines[0] 
                if len(buffer) == bufsize:
                    rightover = lines[-1]
                    del lines[-1]
                for lineno, line in enumerate(lines, 1):
                    for pattern in patterns:
                        if pattern in line:
                            fnl = (filename, lineno, line)
                            results[pattern].append(fnl)
                
pprint(results)