如何grep文件名並將其存儲在python輸出中

Question

我有一個包含多個文件的目錄，我只想grep test.uwsgi.log文件並刪除.uwsgi.log部分。 命名后，我將其存儲在輸出文件中。

這是我得到的：

import subprocess
testlist = “cd /root/test/ && ls | grep ‘uwsgi.log' |sed 's/\.uwsgi.log\>//g' > /root/test/testlist.txt"
output = subprocess.check_output(['bash','-c', testlist])

輸出為：

test1
test2
test3

有人可以教我如何在沒有子流程模塊的情況下實現這一目標，而僅使用python嗎？

Answer 1

單獨使用Python：

import os

for _, dirs, files in os.walk('/root/test/'):
    with open('output.txt', 'w') as out_file:
        out_file.writelines("{}\n".format(f) for f in files if f.endswith('uwsgi.log'))

output.txt符合預期：

test1
test2
test3

注意，我將\\n添加到列表推導中的每個元素。 因為writelines不會自動執行。