allow calling Model class methods wrapped

This commit is contained in:
Laura Klünder 2017-07-13 18:53:48 +02:00
parent 20924b9845
commit 11ff24cc49

View file

@ -89,7 +89,7 @@ class BaseWrapper:
def wrapper(*args, **kwargs):
return func(self, *args, **kwargs)
return wrapper
if isinstance(self, ModelInstanceWrapper) and not hasattr(models.Model, name):
if isinstance(self, (ModelInstanceWrapper, ModelWrapper)) and not hasattr(models.Model, name):
return value
raise TypeError('Can not call %s.%s wrapped!' % (type(self), name))
return value