如何批量实现Python模块的条件式import语句修改？

文件结构 #

packagename\
    mod_1.py
    mod_2.py
    script_1.py
    etc.

两种使用场景 #

• 作为JMP/JSL插件运行：通过script_1.py调用，script_1.py用import mod_1导入模块，mod_1.py需用import mod_2导入同目录模块。
• 作为packagename Python包发布：mod_1.py需用import packagename.mod_2导入同包模块。

当前实现的痛点 #

目前通过判断文件路径区分场景，每个模块都要写重复的分支导入代码：

if 'addin' in __file__:
    import mod_2
else:
    import packagename.mod_2

模块数量较多时，代码冗余度极高。

期望的简化方式 #

希望通过动态前缀实现统一导入逻辑，类似伪代码：

if 'addin' in __file__:
    import_prefix = ''
else:
    import_prefix = 'packagename.'

import import_prefix + mod_2

注：此前查阅《Python：导入子包或子模块》相关内容，仅覆盖部分需求，未完全解决该多场景适配问题。

可行简化方案 #

方案1：使用importlib动态导入 #

利用Python标准库importlib实现动态模块导入，只需一次判断即可批量导入模块：

import importlib

# 确定导入前缀
prefix = '' if 'addin' in __file__ else 'packagename.'

# 动态导入所需模块
mod_2 = importlib.import_module(f"{prefix}mod_2")
mod_3 = importlib.import_module(f"{prefix}mod_3")
# 其他模块依此类推

方案2：封装导入函数 #

如果模块数量极多，可封装一个通用导入函数，进一步简化代码：

import importlib

def get_module(module_name):
    prefix = '' if 'addin' in __file__ else 'packagename.'
    return importlib.import_module(f"{prefix}{module_name}")

# 调用函数导入模块
mod_2 = get_module('mod_2')
mod_3 = get_module('mod_3')
utils = get_module('utils')

方案3：调整导入路径（可选） #

若不想使用动态导入，可通过修改sys.path实现统一导入语法：

import sys
from os.path import abspath, dirname

# 获取当前模块所在目录
module_dir = dirname(abspath(__file__))
# 插件场景下，确保当前目录在导入路径最前
if 'addin' in __file__:
    if module_dir not in sys.path:
        sys.path.insert(0, module_dir)

# 统一使用包前缀导入（插件场景下路径调整后也能识别）
import packagename.mod_2 as mod_2
import packagename.mod_3 as mod_3

注：此方案需确保插件场景下packagename目录结构完整，且存在__init__.py（可为空文件）。

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