Tìm kiếm và thay thế một dòng trong một tệp bằng Python

Tôi muốn lặp lại nội dung của tệp văn bản và thực hiện tìm kiếm và thay thế trên một số dòng và viết kết quả trở lại tệp. Trước tiên tôi có thể tải toàn bộ tệp trong bộ nhớ và sau đó ghi lại, nhưng đó có lẽ không phải là cách tốt nhất để làm điều đó.

Cách tốt nhất để làm điều này, trong đoạn mã sau đây là gì?

f = open(file)
for line in f:
    if line.contains('foo'):
        newline = line.replace('foo', 'bar')
        # how to write this newline back to the file

Tôi đoán một cái gì đó như thế này nên làm điều đó. Về cơ bản, nó ghi nội dung vào một tệp mới và thay thế tệp cũ bằng tệp mới:

from tempfile import mkstemp
from shutil import move, copymode
from os import fdopen, remove

def replace(file_path, pattern, subst):
    #Create temp file
    fh, abs_path = mkstemp()
    with fdopen(fh,'w') as new_file:
        with open(file_path) as old_file:
            for line in old_file:
                new_file.write(line.replace(pattern, subst))
    #Copy the file permissions from the old file to the new file
    copymode(file_path, abs_path)
    #Remove original file
    remove(file_path)
    #Move new file
    move(abs_path, file_path)

Cách ngắn nhất có lẽ là sử dụng mô-đun fileinput . Ví dụ: phần sau đây thêm số dòng vào một tệp, tại chỗ:

import fileinput

for line in fileinput.input("test.txt", inplace=True):
    print('{} {}'.format(fileinput.filelineno(), line), end='') # for Python 3
    # print "%d: %s" % (fileinput.filelineno(), line), # for Python 2

Điều gì xảy ra ở đây là:

1. Các tập tin ban đầu được chuyển đến một tập tin sao lưu
2. Đầu ra tiêu chuẩn được chuyển hướng đến tệp gốc trong vòng lặp
3. Do đó, bất kỳ printcâu lệnh nào ghi lại vào tập tin gốc

fileinputcó nhiều chuông và còi. Ví dụ, nó có thể được sử dụng để tự động hoạt động trên tất cả các tệp trong sys.args[1:]mà không cần bạn phải lặp lại một cách rõ ràng. Bắt đầu với Python 3.2, nó cũng cung cấp một trình quản lý bối cảnh thuận tiện để sử dụng trong một withcâu lệnh.

Mặc dù fileinputlà tuyệt vời cho các kịch bản bỏ đi, tôi sẽ cảnh giác khi sử dụng nó trong mã thực bởi vì phải thừa nhận rằng nó không dễ đọc hoặc quen thuộc. Trong mã thực tế (sản xuất), đáng để chi tiêu thêm một vài dòng mã để làm cho quy trình rõ ràng và do đó làm cho mã có thể đọc được.

Có hai lựa chọn:

1. Tệp không quá lớn và bạn chỉ có thể đọc toàn bộ vào bộ nhớ. Sau đó đóng tệp, mở lại ở chế độ ghi và ghi lại nội dung đã sửa đổi.
2. Các tập tin quá lớn để được lưu trữ trong bộ nhớ; bạn có thể chuyển nó sang một tập tin tạm thời và mở nó ra, đọc từng dòng một, viết lại vào tập tin gốc. Lưu ý rằng điều này đòi hỏi gấp đôi dung lượng lưu trữ.

Đây là một ví dụ khác đã được thử nghiệm và sẽ khớp với các mẫu tìm kiếm & thay thế:

import fileinput
import sys

def replaceAll(file,searchExp,replaceExp):
    for line in fileinput.input(file, inplace=1):
        if searchExp in line:
            line = line.replace(searchExp,replaceExp)
        sys.stdout.write(line)

Ví dụ sử dụng:

replaceAll("/fooBar.txt","Hello\sWorld!$","Goodbye\sWorld.")

Điều này sẽ làm việc: (chỉnh sửa tại chỗ)

import fileinput

# Does a list of files, and
# redirects STDOUT to the file in question
for line in fileinput.input(files, inplace = 1): 
      print line.replace("foo", "bar"),

Dựa trên câu trả lời của Thomas Watnedal. Tuy nhiên, điều này không trả lời chính xác phần trực tuyến của câu hỏi ban đầu. Hàm vẫn có thể thay thế trên cơ sở từng dòng

Việc triển khai này thay thế nội dung tệp mà không sử dụng các tệp tạm thời, do đó, quyền của tệp không thay đổi.

Ngoài ra re.sub thay vì thay thế, cho phép thay thế regex thay vì chỉ thay thế văn bản thuần túy.

Đọc tệp dưới dạng một chuỗi thay vì từng dòng cho phép khớp và thay thế nhiều dòng.

import re

def replace(file, pattern, subst):
    # Read contents from file as a single string
    file_handle = open(file, 'r')
    file_string = file_handle.read()
    file_handle.close()

    # Use RE package to allow for replacement (also allowing for (multiline) REGEX)
    file_string = (re.sub(pattern, subst, file_string))

    # Write contents to file.
    # Using mode 'w' truncates the file.
    file_handle = open(file, 'w')
    file_handle.write(file_string)
    file_handle.close()

Như lassevk gợi ý, hãy viết ra tệp mới khi bạn đi, đây là một số ví dụ mã:

fin = open("a.txt")
fout = open("b.txt", "wt")
for line in fin:
    fout.write( line.replace('foo', 'bar') )
fin.close()
fout.close()

Nếu bạn muốn một chức năng chung thay thế bất kỳ văn bản nào bằng một số văn bản khác, đây có thể là cách tốt nhất để đi, đặc biệt nếu bạn là người hâm mộ của regex:

import re
def replace( filePath, text, subs, flags=0 ):
    with open( filePath, "r+" ) as file:
        fileContents = file.read()
        textPattern = re.compile( re.escape( text ), flags )
        fileContents = textPattern.sub( subs, fileContents )
        file.seek( 0 )
        file.truncate()
        file.write( fileContents )

Một cách pythonic hơn sẽ là sử dụng các trình quản lý bối cảnh như mã dưới đây:

from tempfile import mkstemp
from shutil import move
from os import remove

def replace(source_file_path, pattern, substring):
    fh, target_file_path = mkstemp()
    with open(target_file_path, 'w') as target_file:
        with open(source_file_path, 'r') as source_file:
            for line in source_file:
                target_file.write(line.replace(pattern, substring))
    remove(source_file_path)
    move(target_file_path, source_file_path)

Bạn có thể tìm thấy đoạn trích đầy đủ ở đây .

Tạo một tệp mới, sao chép các dòng từ cũ sang mới và thực hiện thay thế trước khi bạn viết các dòng vào tệp mới.

Mở rộng câu trả lời của @ Kiran, mà tôi đồng ý là ngắn gọn và Pythonic hơn, điều này bổ sung các codec để hỗ trợ việc đọc và viết UTF-8:

import codecs 

from tempfile import mkstemp
from shutil import move
from os import remove


def replace(source_file_path, pattern, substring):
    fh, target_file_path = mkstemp()

    with codecs.open(target_file_path, 'w', 'utf-8') as target_file:
        with codecs.open(source_file_path, 'r', 'utf-8') as source_file:
            for line in source_file:
                target_file.write(line.replace(pattern, substring))
    remove(source_file_path)
    move(target_file_path, source_file_path)

Sử dụng câu trả lời của hamishmcn làm mẫu, tôi có thể tìm kiếm một dòng trong một tệp khớp với biểu thức chính quy của tôi và thay thế nó bằng chuỗi trống.

import re 

fin = open("in.txt", 'r') # in file
fout = open("out.txt", 'w') # out file
for line in fin:
    p = re.compile('[-][0-9]*[.][0-9]*[,]|[-][0-9]*[,]') # pattern
    newline = p.sub('',line) # replace matching strings with empty string
    print newline
    fout.write(newline)
fin.close()
fout.close()

fileinput khá đơn giản như đã đề cập trong các câu trả lời trước:

import fileinput

def replace_in_file(file_path, search_text, new_text):
    with fileinput.input(file_path, inplace=True) as f:
        for line in f:
            new_line = line.replace(search_text, new_text)
            print(new_line, end='')

Giải trình:

• fileinputcó thể chấp nhận nhiều tệp, nhưng tôi thích đóng từng tệp ngay khi nó được xử lý. Vì vậy, đặt đơn file_pathtrong withtuyên bố.
• printcâu lệnh không in bất cứ điều gì khi inplace=True, bởi vì STDOUTđang được chuyển tiếp đến tập tin gốc.
• end=''trong printtuyên bố là để loại bỏ các dòng mới trống trung gian.

Có thể được sử dụng như sau:

file_path = '/path/to/my/file'
replace_in_file(file_path, 'old-text', 'new-text')

nếu bạn loại bỏ thụt lề như bên dưới, nó sẽ tìm kiếm và thay thế trong nhiều dòng. Xem dưới đây ví dụ.

def replace(file, pattern, subst):
    #Create temp file
    fh, abs_path = mkstemp()
    print fh, abs_path
    new_file = open(abs_path,'w')
    old_file = open(file)
    for line in old_file:
        new_file.write(line.replace(pattern, subst))
    #close temp file
    new_file.close()
    close(fh)
    old_file.close()
    #Remove original file
    remove(file)
    #Move new file
    move(abs_path, file)