Python中能否按命名约定批量创建同类型注册装饰器？

我们当前有一批结构重复的注册装饰器，用于将导入的函数或类注册到对应组件类型中，示例代码如下：

def register_crawler(func):
    register(func, "crawler")
    return func

def register_model(func):
    register(func, "model")
    return func

def register_generator(func):
    register(func, "generator")
    return func

所有装饰器最终都调用辅助函数register完成实际注册逻辑（存储导入对象的引用）。我们希望通过组件列表+命名约定批量定义这类装饰器，同时保留@register_crawler这种直观的显式装饰器写法，拒绝改用带参数的装饰器形式（如@register("crawler")）。

可行实现方案 #

可以通过动态生成函数并绑定到当前模块命名空间的方式实现，同时需规避循环中的闭包陷阱：

方法1：利用默认参数绑定组件类型 #

import sys

# 示例辅助注册函数（实际逻辑根据需求调整）
def register(func, component_type):
    # 这里写实际的注册逻辑，比如将函数存入对应组件的注册表
    print(f"已注册 {func.__name__} 为 {component_type} 组件")

# 定义需要生成装饰器的组件列表
components = ["crawler", "model", "generator", "processor"]

# 获取当前模块的引用
current_module = sys.modules[__name__]

for comp in components:
    # 通过默认参数将当前循环的组件名绑定到装饰器函数中，避免闭包陷阱
    def register_decorator(func, comp_type=comp):
        register(func, comp_type)
        return func
    # 生成装饰器名称并绑定到当前模块
    decorator_name = f"register_{comp}"
    setattr(current_module, decorator_name, register_decorator)

方法2：使用functools.partial简化实现 #

如果觉得默认参数的写法不够直观，可通过functools.partial固定register的组件类型参数：

import sys
from functools import partial

# 示例辅助注册函数
def register(func, component_type):
    print(f"已注册 {func.__name__} 为 {component_type} 组件")

components = ["crawler", "model", "generator", "processor"]

current_module = sys.modules[__name__]

for comp in components:
    # 固定组件类型参数，生成偏函数作为装饰器
    register_decorator = partial(register, component_type=comp)
    # 绑定到当前模块
    decorator_name = f"register_{comp}"
    setattr(current_module, decorator_name, register_decorator)

使用验证 #

定义完成后，即可像原来一样使用显式装饰器：

@register_crawler
def custom_crawler():
    pass

@register_model
def custom_model():
    pass

运行后会输出：

已注册 custom_crawler 为 crawler 组件
已注册 custom_model 为 model 组件

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