多Python脚本调用：如何在R.py中复用P.py输入到if语句

Hey there! 刚好碰到过类似的需求，不想动R.py里那个不在函数内的if语句？给你几个低成本的方案：

方案1：通过环境变量传递参数 #

这是最省心的跨脚本传参方式，不用改函数调用逻辑，只需要给R.py加几行读取环境变量的代码（改动极小）：

• 在P.py里，把用户输入存入环境变量：

  import os
  user_input = "你的输入内容"  # 替换成P.py实际的输入值
  os.environ["R_SCRIPT_INPUT"] = user_input
  

• 在R.py的if语句所在位置，直接读取环境变量：

  import os
  # 原有if语句替换为环境变量判断
  if os.environ.get("R_SCRIPT_INPUT") == "目标匹配值":
      # 保留原来的业务逻辑
  

这个方案的好处是不影响原有函数调用链路，就算后续脚本结构调整也能正常工作。

方案2：修改sys.argv传递参数 #

如果你的输入是简单字符串，直接往sys.argv里追加内容就行，R.py可以直接读取命令行参数列表：

• 在P.py的开头，把输入追加到sys.argv：

  import sys
  user_input = "你的输入内容"
  sys.argv.append(user_input)
  

• 在R.py的if语句里，读取对应的argv索引（注意避开脚本本身的启动参数）：

  import sys
  # 假设输入是最后一个参数，用[-1]直接获取
  if sys.argv[-1] == "目标匹配值":
      # 保留原来的业务逻辑
  

这个方案几乎不用改代码，只要R.py导入sys模块就行，适合快速测试场景。

方案3：利用模块级变量传递 #

如果三个脚本是在同一个进程内运行（P调用N，N导入R），可以给R模块加个全局变量，通过N.py中转赋值：

• 在R.py的开头新增一个空变量（只加这一行，完全不碰原有if逻辑）：

  external_input = None  # 新增的模块级全局变量
  # 原有if语句替换为这个变量判断
  if external_input == "目标匹配值":
      # 保留原来的业务逻辑
  

• 在N.py里，先导入R模块赋值，再调用R的函数：

  import R
  # 先把从P获取到的输入赋值给R的模块变量
  R.external_input = 从P传递过来的输入内容
  # 再正常调用R的函数
  R.your_target_function()
  

这个方案适合同进程内的场景，参数传递更直接，不会污染环境变量或命令行参数。

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