Python中实现C++ template类模板功能的最优方案是什么

你原来用工厂函数返回动态类的思路并不是hack，类是Python的一等公民，动态创建类属于语言原生支持的常规操作。针对你需要生成类可继承的要求，以下是几种成熟的实现方案：

方案1：优化工厂函数实现（兼容性最高，无额外依赖） #

这种方案逻辑最简单，全Python版本通用，生成的类可以直接被继承：

def SomeTypeTemplate(x: int, y: int):
    class SomeType:
        # 模板参数绑定为类属性
        TPL_X = x
        TPL_Y = y
        
        def __init__(self, value):
            self.value = value
        
        def compute(self):
            return self.value * self.TPL_X + self.TPL_Y
    return SomeType

使用和继承示例：

# 传入模板参数得到具体类
SomeType2_3 = SomeTypeTemplate(2, 3)

# 正常继承生成的类
class DerivedType(SomeType2_3):
    def double_compute(self):
        return self.compute() * 2

obj = DerivedType(10)
print(obj.compute()) # 输出 2*10+3=23
print(obj.double_compute()) # 输出 46

方案2：基于__init_subclass__实现（写法更符合面向对象习惯） #

如果不想显式调用工厂函数，可以用Python 3.6+支持的__init_subclass__语法，实现更贴近原生类定义的模板效果：

class SomeTypeBase:
    TPL_X: int
    TPL_Y: int

    def __init_subclass__(cls, tpl_x: int, tpl_y: int, **kwargs):
        super().__init_subclass__(**kwargs)
        # 子类定义时自动绑定模板参数
        cls.TPL_X = tpl_x
        cls.TPL_Y = tpl_y

    def __init__(self, value):
        self.value = value

    def compute(self):
        return self.value * self.TPL_X + self.TPL_Y

使用和继承示例：

# 定义时直接传入模板参数
class SomeType2_3(SomeTypeBase, tpl_x=2, tpl_y=3):
    pass

# 继承无额外成本
class DerivedType(SomeType2_3):
    def double_compute(self):
        return self.compute() * 2

如果需要配合mypy等静态类型检查工具，可以结合typing.Generic实现带类型标注的模板类：

from typing import Generic, TypeVar
T = TypeVar("T")

class SomeTypeBase(Generic[T]):
    TPL_X: int
    TPL_Y: int
    def __init_subclass__(cls, tpl_x: int, tpl_y: int, **kwargs):
        super().__init_subclass__(**kwargs)
        cls.TPL_X = tpl_x
        cls.TPL_Y = tpl_y
    def __init__(self, value: T):
        self.value: T = value
    def compute(self) -> int:
        return self.value * self.TPL_X + self.TPL_Y

方案选择建议 #

• 如果你需要兼容旧版本Python、或者需要极高的灵活度，选择工厂函数方案
• 如果你偏好更规范的类定义写法、需要配合静态类型检查，选择__init_subclass__方案

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