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)