python - 在 Python 中删除临时文件

Question

这是我现有的压缩文件夹的代码，我主要是从这里的帮助中整理出来的：

#!/usr/bin/env python

import os
import sys
import datetime

now = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M")
target_dir = '/var/lib/data'
temp_dir='/tmp'

zip = zipfile.ZipFile(os.path.join(temp_dir, now+".zip"), 'w', zipfile.ZIP_DEFLATED)
rootlen = len(target_dir) + 1
for base, dirs, files in os.walk(target_dir):
   for file in files:
      fn = os.path.join(base, file)
      zip.write(fn, fn[rootlen:])

如果我想删除我刚刚在操作结束时创建的 zip 文件，最好的命令是这个吗？

os.remove.join(temp_dir, now+".zip")

Answer 1

会os.remove(os.path.join(temp_dir, now + ".zip"))好的。

但是，您应该确保它在您希望的每种情况下都正确执行。

如果要在任何情况下将其删除，您可以这样做

create it
try:
    work with it
finally:
    remove it

但在这种情况下，您也可以使用该tempfile模块：

import tempfile
with tempfile.NamedTemporaryFile(suffix='.zip') as t:
    z = zipfile.ZipFile(t.name, 'w') # re-create it
    do stuff with z
# after the with clause, the file is gone.

但是，如果您希望它仅在特殊情况下（成功，错误，...）消失，您会被卡住，os.remove(os.path.join(temp_dir, now+".zip"))并且您应该在要删除文件时使用它。

try:
    do_stuff
except VerySpecialException:
    os.remove(os.path.join(temp_dir, now+".zip")) # do that here for a special exception?
    raise # re-raise
except: # only use a bare except if you intend to re-raise
    os.remove(os.path.join(temp_dir, now+".zip")) # or here for all exceptions?
    raise # re-raise
else:
    os.remove(os.path.join(temp_dir, now+".zip")) # or here for success?

Answer 2

这将是这样做的方法：

os.remove(os.path.join(temp_dir, now+".zip"))

玩得开心，迈克

Answer 3

考虑使用 Python 的内置临时文件库。它在关闭时处理临时文件的清理。