导入的Python函数无法保存输出至文件，原地调用正常的解决方案咨询

测试场景与问题 #

我尝试将Python处理的输出保存到文本文件，先后测试了三种方案：

方案1：导入函数传参执行（报错） #

从main_controller.py导入save_output_to_file函数并传入动态生成的urls列表，代码如下：

from controllers.save_output_to_file import save_output_to_file

[...]

print(urls)

save_output_to_file(urls)

执行后触发FileNotFoundError，无论output.txt是否预先存在。

方案2：预定义参数但未调用函数（无文件生成） #

在save_output_to_file.py中预定义urls并编写保存函数，代码如下：

from pathlib import Path


output_folder = Path('../containers/processed')
output_source = 'output.txt'
output_file = output_folder / output_source

urls = ['url4', 'url5', 'url6']

print(urls)


def save_output_to_file(urls):
    """Save URLs to a text file for further processing by bash script."""
    with open(output_file, 'w+', encoding='utf-8') as f:
        for url in urls:
            f.write(f'{url}\n')

执行后无报错，但未生成目标文件。

方案3：直接编写保存逻辑（功能正常） #

在test.py中直接编写文件保存代码，代码如下：

from pathlib import Path


output_folder = Path('../containers/processed')
output_source = 'models.txt'
output_file = output_folder / output_source

urls = ['url7', 'url8', 'url9']

print(urls)

with open(output_file, 'w+', encoding='utf-8') as f:
    for url in urls:  # 注：原代码中变量名ssh_urls应为urls，属于笔误
        f.write(f'{url}\n')

执行后功能正常，文件可正常创建或覆盖。

文件夹结构 #

root
  +- containers
       +- processed
            output.txt
  +- controllers
       main_controller.py
       save_output_to_file.py
       test.py

所有代码文件位于controllers目录，containers与controllers为同级目录，测试环境为WSL2。

需求 #

解决方案1的报错问题，并实现跨平台兼容。

问题根源与解决方案 #

1. 方案2无文件生成的原因：代码仅定义了函数但未调用，在文件末尾添加save_output_to_file(urls)即可执行保存逻辑。

2. 方案1 FileNotFoundError的核心问题：

   • 相对路径的基准是当前工作目录（CWD），而非代码文件所在目录。当从main_controller.py执行时，若启动脚本的工作目录不是controllers，../containers/processed会指向错误路径，导致找不到目标文件夹。
   • 跨平台兼容需避免依赖固定工作目录，应基于代码文件的绝对路径构建目标路径。

3. 修正后的方案1实现（跨平台兼容）：
   修改save_output_to_file.py，基于文件自身路径构建绝对路径：

   from pathlib import Path
   
   # 获取当前文件所在目录的绝对路径
   current_dir = Path(__file__).parent
   # 构建目标文件夹的绝对路径：当前目录 -> 上级目录 -> containers/processed
   output_folder = current_dir.parent / 'containers' / 'processed'
   output_source = 'output.txt'
   output_file = output_folder / output_source
   
   def save_output_to_file(urls):
       """Save URLs to a text file for further processing by bash script."""
       # 确保目标文件夹存在，不存在则创建（跨平台兼容）
       output_folder.mkdir(parents=True, exist_ok=True)
       with open(output_file, 'w', encoding='utf-8') as f:
           for url in urls:
               f.write(f'{url}\n')
   

   然后在main_controller.py中正常导入调用即可：

   from save_output_to_file import save_output_to_file  # 同目录下无需controllers前缀
   
   # 动态生成urls列表
   urls = ['url1', 'url2', 'url3']
   print(urls)
   save_output_to_file(urls)
   

关键优化点 #

• 使用Path(__file__).parent获取代码文件的绝对目录，彻底避免依赖工作目录变化。
• 调用output_folder.mkdir(parents=True, exist_ok=True)确保目标文件夹存在，即使上级目录不存在也会自动创建，同时避免重复创建报错。
• 用'w'模式替代'w+'，仅保留写入权限，更贴合当前场景需求。

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