Python生成臨時文件和文件夾的幾種方法

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

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

1 python生成臨時文件

在Python中生成臨時文件并在程序退出時自動刪除，有以下幾種方法：

使用 tempfile.NamedTemporaryFile

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

import tempfile
import os

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

使用 tempfile.TemporaryDirectory

如果需要臨時目錄及其內容自動清理：

import tempfile
import os

# 創(chuà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"臨時目錄: {temp_dir}")
    # 目錄和其中的文件在with塊結束時自動刪除

手動注冊退出處理函數(shù)

如果需要更多控制權，可以手動注冊清理函數(shù)：

import tempfile
import os
import atexit

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

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

atexit.register(cleanup_temp_file)

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

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

使用 tempfile.mkstemp

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

import tempfile
import os

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

2 python生成臨時文件夾

生成臨時文件夾，在程序退出時自動刪除；

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ù)注冊為程序退出時要執(zhí)行的函數(shù)
atexit.register(cleanup)

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

您可能感興趣的文章: