如何使用Python tempfile模块为临时文件设置自定义完整名称

tempfile.NamedTemporaryFile() 设计目标是生成全局唯一临时文件，避免多进程/多线程场景下的命名冲突，因此前缀、后缀之间的随机字符是默认强制生成的，没有直接参数可以屏蔽该逻辑。

以下是两种可直接落地的实现方案： #

方案1：直接拼接临时目录与自定义文件名（最简单） #

如果你的场景可以保证自定义的文件名不会重复（比如示例中用到秒级时间戳，1秒内不会多次生成文件），可以直接获取系统临时目录路径，自行拼接完整文件名后写入：

import tempfile
import datetime
import os

boiler = b'''
import numpy as np
import matplotlib.pyplot as plt
from astropy.io import ascii


def main():
    """
    """


def func():
    """
    """

    return


if __name__ == '__main__':
    main()
'''


def createFile():
    pref = datetime.datetime.now().strftime("%y%m%d_%H%M%S")
    # 获取系统默认临时目录
    temp_dir = tempfile.gettempdir()
    # 拼接完整自定义文件名
    filename = f"{pref}.py"
    filepath = os.path.join(temp_dir, filename)
    # 直接写入文件
    with open(filepath, "wb") as f:
        f.write(boiler)
    return filepath


filepath = createFile()
print(filepath)

生成的文件路径格式为 /tmp/210925_104246.py，完全符合需求。 #

方案2：保留tempfile机制，生成后重命名 #

如果需要复用tempfile的权限管理、冲突检测等能力，可以先创建临时文件，再重命名为目标名称，重命名前可以增加存在性判断避免覆盖已有文件：

def createFile():
    pref = datetime.datetime.now().strftime("%y%m%d_%H%M%S")
    temp_dir = tempfile.gettempdir()
    target_filepath = os.path.join(temp_dir, f"{pref}.py")
    # 先创建临时文件
    with tempfile.NamedTemporaryFile(delete=False, suffix=".py", prefix=pref, dir=temp_dir) as f:
        temp_filepath = f.name
        f.write(boiler)
    # 重命名为目标文件名
    os.rename(temp_filepath, target_filepath)
    return target_filepath

注意：如果同一秒内多次调用生成函数，会出现重名覆盖的问题，可根据需求增加毫秒级时间戳、序号等逻辑避免冲突。

内容的提问来源于stack exchange，提问作者Gabriel