Python 中的 __init_subclass__

在 Python 中，__init_subclass__ 方法是一种特殊方法，允许我们初始化或自定义类的创建，特别是对于子类。

__init_subclass__ 的关键特征

以下是 '__init_subclass__' 方法的一些关键方面

  • 自动调用：__init_subclass__ 方法在创建子类时会自动调用，无需手动调用此方法。

  • 可通过参数自定义：为了在创建子类时传递其他信息或配置选项，此方法可以接受关键字参数。

  • 继承和重写：Python 的 __init_subclass__ 函数提供了一种通过继承和重写来自定义和控制子类生成的方法。

  • 强制约束：当子类被定义时，我们可以在 __init_subclass__ 函数中强制约束，从而对子类施加特定的要求。

定义 __init_subclass__

在类内部定义 __init_subclass__ 的方式与其他方法一样，但它应该以 cls 作为其第一个参数，它指的是正在创建的类（子类）。

示例

class Base:
   def __init_subclass__(cls, **kwargs):
      super().__init_subclass__(**kwargs)
      print(f"Creating subclass: {cls.__name__}")

__init_subclass__ 的工作原理

当您定义一个从 Base 类或父类继承的新类时，__init_subclass__ 方法会自动调用。

示例

在下面的示例中，SubClass1 和 SubClass2 被创建为 Base 的子类，然后触发 __init_subclass__ 方法。

class Base:
   def __init_subclass__(cls, **kwargs):
      super().__init_subclass__(**kwargs)
      print(f'Subclass created: {cls.__name__}')

class SubClass1(Base):
   pass

class SubClass2(Base):
   pass

输出

Subclass created: SubClass1
Subclass created: SubClass2

自定义子类

通过使用 __init_subclass__ 方法，我们可以自定义子类的创建。

示例

下面的示例代码定义了我们可以要求所有子类都具有一定的属性，但是如果我们尝试创建一个没有 required_attribute 属性的子类，它将引发 TypeError 错误。

class Base:
 # This method is automatically called when a subclass is defined
    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        if not hasattr(cls, 'required_method'):
            raise TypeError(f"{cls.__name__} must implement 'required_method'.")

# SubClass1 implements 'required_method'
class SubClass1(Base):
    def required_method(self):
        pass

# This will raise a TypeError because 'required_method' is not implemented
class SubClass2(Base):
    pass

输出

TypeError: SubClass2 must implement 'required_method'.

更新于: 2024年11月13日

2K+ 浏览量

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告