Support extra functions.

Author: iurisilvio

django_qserializer/serialization.py

@@ -1,3 +1,5 @@
+import types
+
 from django.db import models
 from django.db.models.query import ModelIterable
 from django.db.models.manager import BaseManager
@@ -142,7 +144,18 @@ def serialize(objs):
     return serializer.serialize(objs)
 
 
+class _FuncSerializer(BaseSerializer):
+    def __init__(self, func):
+        super().__init__()
+        self.func = func
+
+    def serialize_object(self, obj):
+        return self.func(obj)
+
+
 def _resolve_serializer(serializer):
-    if not isinstance(serializer, BaseSerializer):
+    if isinstance(serializer, types.FunctionType):
+        serializer = _FuncSerializer(serializer)
+    elif not isinstance(serializer, BaseSerializer):
         serializer = serializer()
     return serializer

django_qserializer/tests/test_serializer.py

@@ -57,22 +57,31 @@ def serialize_object(self, obj):
                 'myattr': obj.company.name
             }
 
+    def func(obj):
+        return {
+            'seats': 32,
+        }
+
     class S(BaseSerializer):
         extra = {
             'myattr': Attr,
+            'func': func,
         }
 
         def serialize_object(self, obj):
             return {
                 'plate': obj.plate,
             }
 
+    serializer = S(extra=['myattr', 'func'])
+
     with django_assert_num_queries(1):
-        bus = Bus.objects.to_serialize(S(extra=['myattr'])).first()
+        bus = Bus.objects.to_serialize(serializer).first()
 
     expected = {
         'plate': 'BUSER',
         'myattr': 'Hurricane Cart',
+        'seats': 32,
     }
 
     with django_assert_num_queries(0):

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = django_qserializer
-version = 0.2.2
+version = 0.2.3
 url = https://github.com/buserbrasil/django-qserializer
 license = MIT
 author = Iuri de Silvio