Pythoninit_subclass实战:子类一定义就自动进注册表,告别手动维护 dict写插件系统、命令分发、序列化器路由时,几乎都会遇到同一个痛点:你有一个基类,一堆子类,还需要一个「名字 → 类」的注册表,好在运行时按字符串把请求路由到对应实现。绝大多数人的第一反应是手写一个字典:HANDLERS{}classJsonExporter:...classCsvExporter:...classXmlExporter:...HANDLERS[json]JsonExporter HANDLERS[csv]CsvExporter HANDLERS[xml]XmlExporter# 新增一个导出器就得回来补这一行这段代码有个致命的隐患:注册表和类定义是分离的。每加一个子类,你都得记得回到这里补登记。忘了补——而这几乎必然会发生——运行时才报KeyError: xml,而且报错点离真正的原因(漏登记)十万八千里。Python 3.6 起提供的__init_subclass__钩子,能让「定义子类」这个动作本身就触发注册,从根上消灭「忘了登记」这类 bug。手动注册表错在哪先把手动方案的问题说透,才知道我们到底要解决什么:注册和定义分离:类写在 A 文件,登记写在 B 文件(或同文件另一处),新增时容易漏。登记顺序有坑:如果登记写在模块顶部、类定义在下面,直接NameError。靠约定,不靠机制:全指望后人记得补一行,code review 也未必抓得到。有人会想到用类装饰器解决:defregister(name):defdeco(cls):HANDLERS[name]clsreturnclsreturndecoregister(json)classJsonExporter:...这好一点,登记和定义写一起了。但还是得每个子类都手动贴装饰器,漏贴一样出问题。我们想要的是:只要你继承了基类,就自动被登记,一行多余代码都不用写。init_subclass:定义子类时自动触发__init_subclass__是定义在父类上的一个钩子方法。它的特殊之处:每当有一个新的子类被定义(注意,是「定义」这个 class 语句执行时,不是实例化时),Python 就会调用父类的__init_subclass__,并把那个新子类作为参数传进来。classExporter:registry:dict[str,type[Exporter]]{}# 子类的类名会被自动传给 cls;这里必须是隐式 classmethoddef__init_subclass__(cls,/,format_name:str,**kwargs):super().__init_subclass__(**kwargs)# 别吞掉给上层的参数Exporter.registry[format_name]clsdefexport(self,data)-str:raiseNotImplementedErrorclassJsonExporter(Exporter,format_namejson):defexport(self,data)-str:importjsonreturnjson.dumps(data,ensure_asciiFalse)classCsvExporter(Exporter,format_namecsv):defexport(self,data)-str:rowsdataifisinstance(data,list)else[data]keyslist(rows[0].keys())lines[,.join(keys)]lines[,.join(str(r[k])forkinkeys)forrinrows]return\n.join(lines)关键点解释清楚:__init_subclass__隐式就是 classmethod,不用你自己写classmethod装饰器,Python 会替你处理。class JsonExporter(Exporter, format_namejson)里的format_namejson不是普通继承语法,而是关键字类参数,它会原样传给__init_subclass__的format_name形参。这就是自动登记名字的来源。那个/表示cls是仅位置参数,是官方推荐写法,防止有人误传clsxxx。一定要调super().__init_subclass__(**kwargs),把剩余关键字参数传给上层,否则多层继承或和别的钩子协作时会炸。现在用起来极其干净:defexport(fmt:str,data)-str:try:clsExporter.registry[fmt]exceptKeyError:raiseValueError(f不支持的格式:{fmt},已注册:{list(Exporter.registry)})fromNonereturncls().export(data)print(export(json,{name:阿明,age:29}))# {name: 阿明, age: 29}print(Exporter.registry)# {json: class ...JsonExporter, csv: class ...CsvExporter}从此新增一个导出器,只需要写class XmlExporter(Exporter, format_namexml),注册表自动多一项。漏登记这个 bug 类别被彻底消除了——因为登记就是继承的副作用,你不可能「继承了但没登记」。进阶一:校验子类必须实现接口自动注册只是起点。__init_subclass__拿到的是类对象,所以你可以在子类定义的那一刻就做校验,而不是等到实例化甚至调用时才崩。这能把错误提前到「导入模块」阶段:classExporter:registry:dict[str,type[Exporter]]{}def__init_subclass__(cls,/,format_name:str,**kwargs):super().__init_subclass__(**kwargs)# 子类必须自己实现 export,否则定义时就报错,不给它上线的机会ifexportnotincls.__dict__:raiseTypeError(f{cls.__name__}必须实现 export 方法)ifformat_nameinExporter.registry:raiseValueError(f格式名{format_name!r}已被 f{Exporter.registry[format_name].__name__}占用)Exporter.registry[format_name]clsif export not in cls.__dict__这个判断很讲究:用cls.__dict__而不是hasattr(cls, export),是因为基类本身就有export,hasattr永远为真,检测不出「子类没重写」。只有查子类自己的__dict__才能确认它真的重写了。现在,如果谁写了个空壳子类:classBrokenExporter(Exporter,format_namebroken):pass# 一 import 这个模块就直接 TypeError: BrokenExporter 必须实现 export 方法模块导入即失败,连部署上线的机会都没有。重复格式名也会当场拦下,不会出现「后定义的悄悄覆盖先定义的」这种诡异问题。进阶二:让插件目录自动全部注册自动注册有个前提常被忽略:子类所在的模块必须被 import 过,class语句才会执行,钩子才会触发。如果你的插件散在plugins/目录下一堆文件里,而主程序从没 import 它们,注册表就是空的。解法是启动时扫描并导入整个包:importimportlibimportpkgutildefload_plugins(package_name:strplugins)-None:把 package 下所有模块都 import 一遍,触发它们的子类注册。packageimportlib.import_module(package_name)formodinpkgutil.iter_modules(package.__path__,package.__name__.):importlib.import_module(mod.name)# import 的副作用就是执行 class 语句load_plugins(plugins)print(已加载导出器:,list(Exporter.registry))pkgutil.iter_modules遍历包内所有子模块,importlib.import_module逐个导入。导入的副作用就是执行模块里的class语句,__init_subclass__顺势把每个子类登记进表。这样你往plugins/里丢一个新文件,重启就自动生效,主程序一个字都不用改——这才是插件系统该有的样子。和元类比:为什么优先选init_subclass老一辈 Python 里,这类「拦截子类创建」的需求都用元类(metaclass)实现:classRegistryMeta(type):def__new__(mcs,name,bases,ns,**kwargs):clssuper().__new__(mcs,name,bases,ns)ifbases:# 跳过基类自己REGISTRY[kwargs[format_name]]clsreturncls元类能干,但代价大:概念门槛高、和其他用元类的库(比如 ABC、ORM、Pydantic v1)容易撞车——因为一个类只能有一个元类,两个都用元类的基类没法多继承。__init_subclass__是普通方法,不占用元类这个「稀缺资源」,能和 ABC、dataclass等和平共处。官方的态度很明确:能用__init_subclass__解决的,就别动元类。元类只留给真正需要改类创建流程本身的极端场景。小结手动维护「名字 → 类」注册表的根本问题是注册与定义分离,靠人记忆补登记,必然漏。__init_subclass__定义在父类上,每定义一个子类就自动触发,让注册成为继承的副作用,从机制上消灭「忘了登记」。它隐式是 classmethod;子类用class Sub(Base, keyx)的关键字类参数传登记信息;务必调super().__init_subclass__(**kwargs)。拿到子类对象后可顺手做接口校验(用cls.__dict__判断是否真重写)、查重名,把错误提前到导入阶段。插件散在多文件时,启动用pkgutilimportlib扫描导入,触发全部注册。优先用它而非元类:不占元类名额,能和 ABC/dataclass 共存。一句话记忆点:注册表不是你去填的,是子类「一出生」就自己报到的。