From 3d7ac9e3655974d5d9adca5e02f257ac99a954e5 Mon Sep 17 00:00:00 2001 From: Jonathan Piacenti Date: Sun, 8 Feb 2015 21:19:27 -0600 Subject: [PATCH] Added raw_queryset keyword to returns_typeclass_list decorated funcs --- evennia/typeclasses/managers.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/evennia/typeclasses/managers.py b/evennia/typeclasses/managers.py index cc7126b28c..366f63aa16 100644 --- a/evennia/typeclasses/managers.py +++ b/evennia/typeclasses/managers.py @@ -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)