Python如何向__init__.py传参？3.7命名空间包初始化传参方法

Great question! Let's break this down into two parts—regular Python packages and namespace packages in Python 3.7—since each has slightly different approaches to achieve that my_package(arg) callable behavior you want.

1. 普通Python包：让包可调用并接收参数 #

By default, Python modules (including packages) aren't callable, but we can work around this by replacing the module object in sys.modules with a custom callable class instance. Here's how to implement it:

Step 1: 包结构 #

my_package/
    __init__.py

Step 2: 在__init__.py中实现可调用逻辑 #

import sys

# 存储包级别的配置参数
_package_config = None

def _init_package(arg):
    """内部初始化逻辑，使用传入的参数"""
    global _package_config
    _package_config = arg
    # 在这里添加初始化代码（比如加载资源、初始化子模块）
    print(f"包已使用参数初始化：{arg}")

class _CallablePackage:
    def __call__(self, arg):
        _init_package(arg)
        # 返回包模块本身，让用户可以继续访问它的属性
        return sys.modules[__name__]

# 用我们的可调用实例替换默认模块
sys.modules[__name__] = _CallablePackage()

Step 3: 带参数使用包 #

import my_package

# 直接调用包来传递初始化参数
my_package("my_custom_argument")

# 之后可以访问存储的配置（或者在包内部使用它）
print(my_package._package_config)  # 输出: my_custom_argument

这种方法让你的包在被调用时像函数一样工作，同时在初始化后保留所有正常的模块属性。

2. Python 3.7命名空间包：传递初始化参数 #

命名空间包（遵循PEP 420）不需要顶层的__init__.py，这增加了一点复杂度。关键是为命名空间创建一个可调用的模块对象，并在导入之前或期间替换sys.modules中的对应条目。

Step 1: 命名空间包结构 #

src/
    my_package/
        sub_package_a/
            __init__.py
        # 没有顶层__init__.py（PEP 420命名空间包）

Step 2: 在子包的__init__.py中添加可调用逻辑 #

我们将用其中一个子包来设置可调用的命名空间：

import sys
import types

# 获取或创建命名空间包模块
namespace_name = "my_package"
ns_package = sys.modules.get(namespace_name)
if not ns_package:
    ns_package = types.ModuleType(namespace_name)
    sys.modules[namespace_name] = ns_package

class _CallableNamespace(types.ModuleType):
    def __init__(self, name):
        super().__init__(name)
        self._namespace_config = None

    def __call__(self, arg):
        self._namespace_config = arg
        # 使用参数初始化子包
        from my_package import sub_package_a
        sub_package_a._setup(self._namespace_config)
        print(f"命名空间包已使用参数初始化：{arg}")
        return self

# 用我们的可调用实例替换命名空间模块
sys.modules[namespace_name] = _CallableNamespace(namespace_name)

Step 3: 带参数使用命名空间包 #

import my_package

# 直接调用命名空间包来传递参数
my_package("namespace_init_arg")

# 访问配置或使用已初始化的子包
print(my_package._namespace_config)  # 输出: namespace_init_arg

命名空间包的重要注意事项： #

• 如果你的命名空间分布在多个位置，确保只有一个子包处理命名空间替换，避免冲突。
• 如果命名空间有多个子包，通过命名空间的__call__方法协调它们的初始化。

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