Added raw_queryset keyword to returns_typeclass_list decorated funcs

This commit is contained in:
Jonathan Piacenti 2015-02-08 21:19:27 -06:00 committed by Griatch
parent 5a811a2d89
commit 3d7ac9e365

View file

@ -23,7 +23,12 @@ def returns_typeclass_list(method):
"""
def func(self, *args, **kwargs):
self.__doc__ = method.__doc__
return list(method(self, *args, **kwargs))
raw_queryset = kwargs.pop('raw_queryset', False)
result = method(self, *args, **kwargs)
if raw_queryset:
return result
else:
return list(result)
return update_wrapper(func, method)