python - 为模块中的所有类覆盖 __init__

Question

在我的 Django 项目中，我希望所有模型字段都有一个名为 documentation 的附加参数。 (它类似于 verbose_name 或 help_text，但用于内部文档。)

这看起来很简单:只需子类化并覆盖字段的 __init__:

def __init__(self, verbose_name=None, name=None, documentation=None, **kwargs):
    self.documentation = documentation
    super(..., self).__init__(verbose_name, name, **kwargs)

问题是如何使其适用于 django.db.models 中的所有 20 个字段类(BooleanField、CharField、PositiveIntegerField等)？

我看到的唯一方法是将元编程与检查模块一起使用:

import inspect
import sys
from django.db.models import *

current_module = sys.modules[__name__]

all_field_classes = [Cls for (_, Cls) in inspect.getmembers(current_module,
            lambda m: inspect.isclass(m) and issubclass(m, Field))]

for Cls in all_field_classes:
    Cls.__init__ = <???>

我不习惯看到这样的代码，甚至不知道它是否能工作。我希望我可以将属性添加到 Field 基类，并将其继承到所有子类，但我不知道如何做到这一点。

有什么想法吗？

Answer 1

确实 - 你走在正确的道路上。在Python中，自省(introspection)是很正常的事情，你甚至不需要使用inspect模块只是因为它“我正在使用内省(introspection)和元编程，我必须需要 inspect ):-)

不过，有一点不被认为是很好的做法，那就是 Monkey 修补 - 也就是说，如果您更改 django.db.models 中的类。本身，以便其他模块将从那里导入修改后的类并使用修改后的版本。 (请注意，在这种情况下:不推荐!= 将不起作用) - 因此您最好在自己的模块中创建所有新模型类，并从您自己的模块导入它们，而不是从 django.db.models 导入它们。

所以，一些事情:

from django.db import models

# A decorator to implement the behavior you want for the
# __init__ method
def new_init(func):
   def __init__(self, *args,  **kw):
       self.documentation = kw.pop("documentation", None)
       return func(self, *args, **kw)

for name, obj in models.__dict__.items():
    #check if obj is a class:
    if not isinstance(obj, type):
        continue

    # creates a new_init, retrieving the original one -
    # taking care for not to pick it as an unbound method -
    # check: http://pastebin.com/t1SAusPS
    new_init_method = new_init(obj.__dict__.get("__init__", lambda s:None))

    # dynamically creates a new sublass of obj, overriding just the __init__ method:
    new_class = type(name, (obj,), {"__init__": new_init_method}) 

    # binds the new class to this module's scope:
    globals().__setitem__(name, new_class)

或者如果您更喜欢使用猴子修补，因为它更容易:-p

from django.db import models

def new_init(func):
   def __init__(self, *args,  **kw):
       self.documentation = kw.pop("documentation", None)
       return func(self, *args, **kw)

for name, obj in models.__dict__.items():
    #check if obj is a class:
    if not isinstance(obj, type):
        continue

    obj.__init__ = new_init(obj.__dict__["__init__"])