From 48cbe474a48da637a753beacf8a46bf63da285d3 Mon Sep 17 00:00:00 2001
From: Vyacheslav Tverskoy <tsx@tsx.su>
Date: Wed, 12 May 2021 12:57:19 +0000
Subject: [PATCH] Support factories and serializers in schema directly

---
 cleancat/base.py | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/cleancat/base.py b/cleancat/base.py
index 1ff2bb0..1deca9d 100644
--- a/cleancat/base.py
+++ b/cleancat/base.py
@@ -678,6 +678,12 @@ def clean(self, value):
         )
 
 
+class SchemaOptions:
+    def __init__(self, factory=None, serializer=None):
+        self.factory = factory
+        self.serializer = serializer
+
+
 class Schema(object):
     """
     Base Schema class. Provides core behavior like fields declaration
@@ -745,6 +751,10 @@ def get_fields(cls):
                 fields[field_name] = field
         return fields
 
+    @classmethod
+    def _get_options(cls):
+        return getattr(cls, '__options__', SchemaOptions())
+
     @classmethod
     def obj_to_dict(cls, obj):
         """
@@ -851,7 +861,10 @@ def full_clean(self):
             self.errors = [e.args and e.args[0]]
 
         self.raise_on_errors()
-        return self.data
+        factory = self._get_options().factory
+        if factory is None:
+            return self.data
+        return factory(**self.data)
 
     def raise_on_errors(self):
         if self.field_errors or self.errors:
@@ -877,9 +890,14 @@ def external_clean(self, cls, raise_on_errors=True):
 
     def serialize(self):
         data = {}
+        serializer = self._get_options().serializer
+        if serializer is None:
+            original_data = self.data
+        else:
+            original_data = serializer(self.data)
         for field_name, field in self.fields.items():
             raw_field_name = field.raw_field_name or field_name
-            value = self.data[field_name]
+            value = original_data[field_name]
             data[raw_field_name] = field.serialize(value)
         return data