在我的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(BooleanField
,CharField
,{{1}中20个字段类的所有等等)?
我看到的唯一方法是使用元编程与inspect模块:
PositiveIntegerField
我不习惯看到这样的代码,甚至不知道它是否会起作用。我希望我可以将属性添加到基类Field类,并让它继承到所有子类,但我不知道如何做到这一点。
有什么想法吗?
答案 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__"])