使用__init_subclass__为子类方法添加后置处理的实现疑问

你当前的代码实现如下：

def log(return_value):
  print('Logged:', return_value)


class Car:
  @staticmethod
  def logged_run(run): 

    def wrap(self):
      result = run(self)
      log(result)
      return result

    return wrap

  def run(self):
    return 'Car'
  
  def __init_subclass__(cls) -> None:
    super().__init_subclass__()
    cls.run = cls.logged_run(cls.run)

class AutoCar(Car):
  def run(self):
    return 'AutoCar'



auto = AutoCar()
car_type = auto.run()

运行输出：

Logged: AutoCar

针对你的疑问逐一解答：

1. 当前实现是否存在问题？ #

当前实现能满足基础需求，但存在重复装饰的隐患：如果某个子类再被其他子类继承（比如class Tesla(AutoCar): pass），Car的__init_subclass__会再次对继承来的run方法进行装饰，导致调用Tesla().run()时日志被打印多次。

2. 是否需要调用super().__init_subclass__()？ #

建议保留这个调用。如果后续Car的继承层级扩展（比如混入其他定义了__init_subclass__的类，或者Car继承自某个非object的父类），super()能保证父类的__init_subclass__逻辑也被执行，是符合Python继承规范的良好习惯。即使当前Car直接继承object，调用它也不会有任何副作用。

3. __init_subclass__的参数列表是否需要**kwargs？ #

需要添加。如果子类定义时使用了自定义关键字参数（比如class AutoCar(Car, debug=True):），没有**kwargs的话会抛出参数不匹配的错误。修改后的方法签名应该是：

def __init_subclass__(cls, **kwargs) -> None:
    super().__init_subclass__(**kwargs)
    # 后续逻辑

这样能兼容子类定义时的任意关键字参数，提升代码的健壮性。

4. 更优的实现方式？ #

方式一：避免重复装饰 #

在装饰器中给包装后的函数添加标记，在__init_subclass__中判断是否已经装饰过，防止重复：

import functools

def log(return_value):
    print('Logged:', return_value)

class Car:
    @staticmethod
    def logged_run(run):
        # 检查是否已经被装饰过
        if hasattr(run, '_is_logged'):
            return run
        
        @functools.wraps(run)  # 保留原函数的元信息（名称、文档等）
        def wrap(self):
            result = run(self)
            log(result)
            return result
        
        wrap._is_logged = True
        return wrap

    def run(self):
        return 'Car'
    
    def __init_subclass__(cls, **kwargs) -> None:
        super().__init_subclass__(**kwargs)
        cls.run = cls.logged_run(cls.run)

class AutoCar(Car):
    def run(self):
        return 'AutoCar'

class Tesla(AutoCar):
    pass

Tesla().run()  # 只会打印一次日志

方式二：使用类装饰器（适合少量子类场景） #

如果子类数量不多，也可以用类装饰器直接给子类的run方法添加逻辑，但这种方式需要给每个子类手动添加装饰器，不如__init_subclass__自动化：

def add_logging(cls):
    original_run = cls.run
    @functools.wraps(original_run)
    def wrapped_run(self):
        result = original_run(self)
        log(result)
        return result
    cls.run = wrapped_run
    return cls

@add_logging
class AutoCar(Car):
    def run(self):
        return 'AutoCar'

方式三：使用抽象基类+装饰器（强制子类实现run） #

如果需要强制子类必须实现run方法，可以结合abc模块：

from abc import ABC, abstractmethod

class Car(ABC):
    @abstractmethod
    def run(self):
        pass
    
    @staticmethod
    def logged_run(run):
        # 同方式一的装饰器逻辑
        pass
    
    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        cls.run = cls.logged_run(cls.run)

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