텍스트 파일 내용 교체 (csv이용)

한글패치 관련 짧은 글들

텍스트 파일 내용 교체 (csv이용)

Snowyegret 2021. 8. 9. 09:58

출처: https://www.studytonight.com/python-howtos/search-and-replace-a-text-in-a-file-in-python

# importing the regex module
import re
import csv


# defining the replace method
def replace(file_path, text, subs, flags=0):
    with open(file_path, "r+", encoding='utf-8') as file:
        # read the file contents
        file_contents = file.read()
        text_pattern = re.compile(re.escape(text), flags)
        file_contents = text_pattern.sub(subs, file_contents)
        file.seek(0)
        file.truncate()
        file.write(file_contents)


def reformatt_string(input_text):
    return '\"val\": \"'+input_text+'\",'


file_path = "dialogues-jp.txt"
csv_path = "test.csv"

f = open(csv_path,'r', encoding='utf-8')
rdr = csv.reader(f)

for findincsv in rdr:
    string_replaced = reformatt_string(findincsv[0])
    new_string = reformatt_string(findincsv[1])
    replace(file_path, string_replaced, new_string)

def reformatt_string(input_text):

    return '\"val\": \"'+input_text+'\",'

검색/대체할 텍스트의 앞/뒷부분을 input_text의 앞/뒤에 넣는다.

file_path = "dialogues-jp.txt"

csv_path = "test.csv"

수정할 원본 txt파일과, 수정할 단어가 담겨있는 csv파일의 경로 (첫번째 라인이 원어, 두번째 라인이 번역본)

for findincsv in rdr:
    string_replaced = reformatt_string(findincsv[0])
    new_string = reformatt_string(findincsv[1])
    replace(file_path, string_replaced, new_string)

findincsv[0]

findincsv[1] -> 첫 번째 라인을 두번째 라인으로 대체한다는 것이므로, 상황에 맞게 수정 가능

사용예시) \n 개행문자가 들어있는 텍스트 파일의 경우

아래 텍스트 파일을 csv로 만들 것인데, \n 개행문자가 있을 시 파이썬에서 줄바꿈이 되므로 미리 변환시켜놓자.