创建一个临时目录以存储用于计算的文件

Question

A python script downloads 2 tables from a database, does some calculations and creates a new one, and uploads it. There are intermediate files involved and to store all the files I create a folder called “temp” in the current working directory. Before the program exits it deletes temp. This doesn't work very well because

1. If there already is a folder named temp it will be deleted
2. If the program crashes before the last step of deleting temp, the next time the program runs it crashes because files with the same name exist from the previous execution.

I imagine this is a relatively common scenario. How is it normally handled?

Answer 1

there is a tempfile module just for this

import tempfile
new_dir = tempfile.mkdtemp() #makes a new temp folder guaranteed to be unique(in the os-specific temp location)
...
os.remove(new_dir)

really I dont think you even need to remove the new_dir as it will automagically be gone on next reboot usually(*i think)