python txtファイルに一行ごと文字を追加したい

Question

実現したいこと

以下に示す例のようにa.txtを読み込み、一行ごとに文字を追加して更新を行いたいのですが、pythonで行う方法がわからないため教えていただけると幸いです。
よろしくお願いいたします。

a
b
c
d

a.py
b.py
c.py
d.py

Comments

[1T2R3M4]

調べたこと、試したことを質問に追記していただけませんか。

Answer 1

解決済みですが、アイデアがあるので回答します。
一時ファイルを作ってコピーする方法にすると取り回し易いのかなと思いました。

import shutil
import tempfile


dst_file = 'a.txt'

with open(dst_file, mode='r') as fin_out,   \
        tempfile.NamedTemporaryFile(mode='w', delete=False) as ftmp:
    
    for line in map(str.rstrip, fin_out):
        print(f'{line}.py', file=ftmp)

shutil.move(ftmp.name, dst_file)

大したサイズで無いのであれば、解決方法のとおりメモリ上で捌いても充分とは思います。
ちょっと綺麗にするならこんな感じ。

file_path = 'a.txt'

with open(file_path) as fin:
    lines = [
        '{}.py'.format(line) for line in map(str.rstrip, fin)
    ]
    
with open(file_path, 'w') as fout:
    print(*lines, sep='\n', file=fout)

Answer 2

以下のように、配列に読み込んで更新し上書きすることで解決することができました。

file_path = "aaa.txt"
with open(file_path) as f:
  lines = f.readlines()
lines = [line.rstrip('\n') for line in lines]
f.close()
print(lines)
for i in range (len(lines)):
  lines[i] = lines[i] + '.py'
with open('aaa.txt', 'w') as f:
    for d in lines:
        f.write("%s\n" % d)