python腳本執行后清理/ tmp目錄

Question

我正在使用python並在.py文件中調用外部實用程序“ send-data”，如下所示，

cmd=['send-data', '-f', file_name]
res = subprocess.run(cmd,stdout=subprocess.PIPE, input=b'a')

我無法控制send-data實用程序，它會在/ tmp目錄下使用隨機名稱創建文件，如響應字符串所示。

'a)bort, e)dit or s)end? send-data: the file remains in /tmp/pbadLWX .'

想知道每次執行后如何刪除/ tmp文件夾下的文件“ pbadLWX”？ 考慮每次文件名都可以不同嗎？

Answer 1

您可以通過re模塊解析輸出，然后通過os.remove刪除文件：

import os
import re

f_name = re.search(r'\s(/tmp/[a-zA-Z]+)\s', res.stdout.decode('ascii')).group(1)
os.remove(f_name)