Python生成臨時(shí)文件和文件夾的幾種方法

更新時(shí)間：2026年03月10日 09:15:34 作者：mahuifa

這篇文章主要介紹了在Python中生成臨時(shí)文件和臨時(shí)文件夾的方法,包括使用tempfile模塊中的不同函數(shù),以及推薦的做法,需要的朋友可以參考下

1 python生成臨時(shí)文件

在Python中生成臨時(shí)文件并在程序退出時(shí)自動(dòng)刪除，有以下幾種方法：

使用 tempfile.NamedTemporaryFile

這是最常用的方法，可以自動(dòng)管理臨時(shí)文件的創(chuàng)建和刪除：

import tempfile
import os

# 創(chuàng)建臨時(shí)文件，程序退出時(shí)自動(dòng)刪除
with tempfile.NamedTemporaryFile(delete=True) as temp_file:
    print(f"臨時(shí)文件路徑: {temp_file.name}")
    # 寫(xiě)入數(shù)據(jù)
    temp_file.write(b"Hello, temporary file!")
    temp_file.flush()  # 確保數(shù)據(jù)寫(xiě)入磁盤(pán)
    
    # 文件在with塊結(jié)束時(shí)自動(dòng)刪除

使用 tempfile.TemporaryDirectory

如果需要臨時(shí)目錄及其內(nèi)容自動(dòng)清理：

import tempfile
import os

# 創(chuàng)建臨時(shí)目錄，程序退出時(shí)自動(dòng)刪除
with tempfile.TemporaryDirectory() as temp_dir:
    temp_file_path = os.path.join(temp_dir, "test.txt")
    with open(temp_file_path, "w") as f:
        f.write("Hello, temporary directory!")
    print(f"臨時(shí)目錄: {temp_dir}")
    # 目錄和其中的文件在with塊結(jié)束時(shí)自動(dòng)刪除

手動(dòng)注冊(cè)退出處理函數(shù)

如果需要更多控制權(quán)，可以手動(dòng)注冊(cè)清理函數(shù)：

import tempfile
import os
import atexit

# 創(chuàng)建臨時(shí)文件
temp_file = tempfile.NamedTemporaryFile(delete=False)
temp_file_path = temp_file.name
temp_file.close()  # 關(guān)閉文件以便后續(xù)操作

# 注冊(cè)退出時(shí)的清理函數(shù)
def cleanup_temp_file():
    if os.path.exists(temp_file_path):
        os.unlink(temp_file_path)
        print(f"已刪除臨時(shí)文件: {temp_file_path}")

atexit.register(cleanup_temp_file)

# 使用臨時(shí)文件
with open(temp_file_path, "w") as f:
    f.write("Hello, manual cleanup!")

print(f"臨時(shí)文件路徑: {temp_file_path}")

使用 tempfile.mkstemp

創(chuàng)建臨時(shí)文件并獲得文件描述符：

import tempfile
import os

# 創(chuàng)建臨時(shí)文件
fd, temp_path = tempfile.mkstemp()
try:
    print(f"臨時(shí)文件路徑: {temp_path}")
    # 使用文件描述符寫(xiě)入數(shù)據(jù)
    os.write(fd, b"Hello, mkstemp!")
finally:
    os.close(fd)  # 關(guān)閉文件描述符
    
# 手動(dòng)刪除文件
os.unlink(temp_path)

2 python生成臨時(shí)文件夾

生成臨時(shí)文件夾，在程序退出時(shí)自動(dòng)刪除；

def get_temp_dir() -> str:
    return os.path.join(get_main_dir(), TEMP_DIR)


def cleanup():
    temp_dir = get_temp_dir()
    if os.path.exists(temp_dir):
        shutil.rmtree(temp_dir)
        

temp_dir = get_temp_dir()
os.makedirs(temp_dir,  exist_ok=True)
#將 cleanup 函數(shù)注冊(cè)為程序退出時(shí)要執(zhí)行的函數(shù)
atexit.register(cleanup)

到此這篇關(guān)于Python生成臨時(shí)文件和文件夾的幾種方法的文章就介紹到這了,更多相關(guān)Python生成臨時(shí)文件和文件夾內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: