Annikka/evennia
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added raw_queryset keyword to returns_typeclass_list decorated funcs

Browse source
This commit is contained in:
Jonathan Piacenti 2015-02-08 21:19:27 -06:00 • committed by Griatch
parent 5a811a2d89
commit 3d7ac9e365
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
evennia/typeclasses/managers.py
View file

@ -23,7 +23,12 @@ def returns_typeclass_list(method):
""" """
def func(self, *args, **kwargs): def func(self, *args, **kwargs):
self.__doc__ = method.__doc__ 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) return update_wrapper(func, method)