子进程解压缩源文件删除

Question

我正在使用python子进程来解压缩zip存档。我的代码如下：

subprocess.Popen(['unzip', '{}.zip'.format(inputFile), '-d', output_directory])

有没有解压命令可以在解压后删除zip源文件？如果没有，我如何通过管道将rm发送到subprocess.Popen，但要确保它先等待文件解压缩？

谢谢。

Answer 1

您可以在Shell中使用&&，只有当第一个命令成功时，它才会执行第二个命令：

import subprocess
import os
values = {'zipFile': '/tmp/simple-grid.zip', 'outDir': '/tmp/foo'}
command = 'unzip {zipFile} -d {outDir} && rm {zipFile}'.format(**values)
proc = subprocess.Popen(command, shell=True)
_ = proc.communicate()
print('Success' if proc.returncode == 0 else 'Error')

或者，如果unzip成功，则为os.remove()：

inputFile = values['zipFile']
output_directory = values['outDir']
proc = subprocess.Popen(
['unzip', '{}'.format(inputFile), '-d', output_directory]
)
_ = proc.communicate()  # communicate blocks!
if proc.returncode == 0:
os.remove(values['zipFile'])
print('Success' if not os.path.exists(inputFile) else 'Error')