如果指定了文件夹,存入到指定文件夹,如果未指定文件夹,存到当前文件夹
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()
函数创建一个文件对象,并将内容写入文件。你可以根据需要修改这一部分的代码,以实现自己的文件保存逻辑。最后,程序输出文件的保存路径。