如何创建目录并存储脚本生成文件,且支持目录存在时清空内容?
解决脚本生成文件存入指定目录并清空旧内容的问题
你的代码问题在于生成文件时没有指定目标目录的路径,导致文件创建在脚本运行的当前目录,而非你创建的new文件夹里。同时也未实现“目录存在则清空内容”的逻辑,下面是修正后的代码:
import os # 定义目标目录路径,方便后续修改 TARGET_DIR = "home/Documents/new" # 处理目标目录:不存在则创建,存在则清空文件 if os.path.exists(TARGET_DIR): # 遍历目录内的所有内容,仅删除文件(如需删除子目录可额外处理) for filename in os.listdir(TARGET_DIR): file_path = os.path.join(TARGET_DIR, filename) if os.path.isfile(file_path): os.remove(file_path) else: # 目录不存在则创建(支持多级目录创建) os.makedirs(TARGET_DIR) # 生成文件并写入目标目录 with open(os.path.join(TARGET_DIR, "demo1.txt"), "w") as f: f.write("welcome to python") with open(os.path.join(TARGET_DIR, "demo2.txt"), "w") as f: f.write("new world")
关键改动说明:
- 指定文件路径:用
os.path.join()拼接目标目录和文件名,确保文件写入new文件夹内,而非当前工作目录 - 清空已有文件:当目录存在时,遍历目录内的所有文件并删除,保留子目录(如果需要连子目录一起删除,可以改用
shutil.rmtree(TARGET_DIR)后重新创建目录) - 使用with语句:替代手动调用
close(),自动管理文件资源,代码更简洁安全
内容的提问来源于stack exchange,提问作者V_S
相关产品推荐
相关产品推荐

