覆盖模块中所有类的__init__

时间:2014-01-23 20:47:24

标签: python django inheritance reflection metaprogramming

在我的Django项目中,我希望我的所有模型字段都有一个名为documentation的附加参数。 (它类似于verbose_namehelp_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(BooleanFieldCharField,{{1}中20个字段类的所有等等)?

我看到的唯一方法是使用元编程与inspect模块:

PositiveIntegerField

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

有什么想法吗?

1 个答案:

答案 0 :(得分:1)

确实 - 你是他正确的轨道。 在python中,内省是正常的事情,你甚至不需要使用inspect模块,因为它“我正在使用内省和元编程,我必须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__"])