Python生成临时文件和文件夹的几种方法

import tempfile
import os

# 创建临时文件，程序退出时自动删除
with tempfile.NamedTemporaryFile(delete=True) as temp_file:
    print(f"临时文件路径: {temp_file.name}")
    # 写入数据
    temp_file.write(b"Hello, temporary file!")
    temp_file.flush()  # 确保数据写入磁盘
    
    # 文件在with块结束时自动删除

import tempfile
import os

# 创建临时目录，程序退出时自动删除
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块结束时自动删除

import tempfile
import os
import atexit

# 创建临时文件
temp_file = tempfile.NamedTemporaryFile(delete=False)
temp_file_path = temp_file.name
temp_file.close()  # 关闭文件以便后续操作

# 注册退出时的清理函数
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}")

import tempfile
import os

# 创建临时文件
fd, temp_path = tempfile.mkstemp()
try:
    print(f"临时文件路径: {temp_path}")
    # 使用文件描述符写入数据
    os.write(fd, b"Hello, mkstemp!")
finally:
    os.close(fd)  # 关闭文件描述符
    
# 手动删除文件
os.unlink(temp_path)

import tempfile

# 最佳实践示例
with tempfile.NamedTemporaryFile(suffix='.txt', prefix='temp_', delete=True) as f:
    f.write(b"临时数据")
    f.flush()
    # 文件路径: f.name
    # 文件会在with块结束时自动删除

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 函数注册为程序退出时要执行的函数
atexit.register(cleanup)