как я могу использовать fileinput для редактирования нескольких файлов?

Вот как я это делаю, добавляя к вашему коду:

 import os import fnmatch import fileinput def openFiles(dir): filePaths = [] for root, dirs, files in os.walk(dir): for textFile in fnmatch.filter(files, "*.txt"): filepath = os.path.join(root, textFile) filePaths.append(filepath) return filePaths def prefixLines(filepaths, chartoPrefix, prefixWith): res = '' for filepath in filepaths: # Read file with open(filepath, 'r') as f: for line in f: if line.startswith(chartoPrefix): res += prefixWith + line else: res += line # Write to file with open(filepath, 'w') as f: f.write(res) res = '' # Rest res prefixLines(openFiles(r'/home/test1/'), 'xe', '#') 

prefixLines страдает от многих недостатков:

• Поскольку мы читаем все строки файлов и сохраняем их в res , у нас может закончиться нехватка памяти для больших файлов.

• Если каким-то образом программист забыл отступ res = '' в правом блоке или если res был полностью опущен, а код работал с фактическими файлами, которые нужны пользователю, вы в конечном итоге записываете содержимое предыдущего файла чтения в следующий файл и последний файл будет содержать все прочитанные файлы. Вот почему вы используете этот код в тестовой среде или используете его с осторожностью.

Этот код служит только для того, чтобы продемонстрировать, как вы могли бы достичь желаемых эффектов, префикс строк файла, который начинается с строки с другой строкой. Поэтому рекомендуется небольшое улучшение этого кода. Например, вместо того, чтобы читать все содержимое файла и сохранять их в res вы могли бы просто сохранить номер строки, который должен быть префикс, и тем самым исключить необходимость загрузки всех данных в память. enumerate также может помочь вернуть номер файла, он возвращает итерабельность в 2.7. Исключая res , мы сохраняем не только память, но и устраняем недостаток в пуле 2.

Я закончил это так. Но я использую классы в своем основном коде, поэтому он разбит на 2 функции вместо одного. В моем основном коде я использовал список для хранения всех путей к файлам и использовал fileinput, чтобы открыть каждый путь for line in fileinput.FileInput(pathlist, inplace=1): do something. из списка таким образом for line in fileinput.FileInput(pathlist, inplace=1): do something. Я благодарю @direprobs за ее ответ, поскольку она пролила некоторый свет на то, как я должен это делать.

 import fnmatch import fileinput import os import sys def openFiles(): for dirpath, dirs, files in os.walk('/home/test1/'): for filename in fnmatch.filter(files, "*.txt"): filepaths = os.path.join(dirpath, filename) for line in fileinput.FileInput(filepaths, inplace=1): if line.startswith("xe"): add = "# {}".format(line) line = line.replace(line, add) sys.stdout.write(line) fileinput.close() openFiles()