【Python】Python程序根据指定的文件夹存储文件。如果未指定文件夹，文件将保存在当前工作目录下。

#【Python】Python程序根据指定的文件夹存储文件。如果未指定文件夹，文件将保存在当前工作目录下。| 来源: 网络整理| 查看: 265

如果指定了文件夹，存入到指定文件夹，如果未指定文件夹，存到当前文件夹

import os import shutil def save_file(filename, folder=None): # 获取当前工作目录 current_folder = os.getcwd() if folder is None: # 如果未指定文件夹，则使用当前工作目录 folder = current_folder else: # 如果指定了文件夹，则将文件夹路径与当前工作目录拼接 folder = os.path.join(current_folder, folder) # 确保目标文件夹存在，如果不存在则创建 os.makedirs(folder, exist_ok=True) # 拼接完整的文件路径 file_path = os.path.join(folder, filename) # 进行文件保存操作，这里使用一个示例，将文件内容写入文件 with open(file_path, 'w') as file: file.write("This is a test file.") print(f"文件已保存至：{file_path}") # 示例用法 save_file("example.txt") # 存储到当前文件夹 save_file("example2.txt", "my_folder") # 存储到指定文件夹

个程序首先检查是否指定了文件夹，如果没有，则使用os.getcwd()获取当前工作目录。如果指定了文件夹，则将文件夹路径与当前工作目录拼接。然后，程序通过os.makedirs()创建目标文件夹（如果不存在的话）。接下来，使用os.path.join()将文件夹路径和文件名拼接成完整的文件路径。在示例中，程序使用open()函数创建一个文件对象，并将内容写入文件。你可以根据需要修改这一部分的代码，以实现自己的文件保存逻辑。最后，程序输出文件的保存路径。

【本文地址】

公司简介

联系我们