MonngoEmbedded

closeio · Sep 29, 2012 · 42cd1d7 · 42cd1d7
1 parent 9dce120
commit 42cd1d7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/cleancat/base.py b/cleancat/base.py
@@ -106,7 +106,6 @@ def has_value(self, value):
 
     def clean(self, value):
         value = super(Embedded, self).clean(value)
-
         return self.schema_class(value).full_clean()
 
     def is_valid(self):
@@ -135,6 +134,15 @@ def clean(self, value):
         return value
 
 # TODO move to separate module
+class MongoEmbedded(Embedded):
+    def __init__(self, document_class=None, *args, **kwargs):
+        self.document_class = document_class
+        super(MongoEmbedded, self).__init__(*args, **kwargs)
+
+    def clean(self, value):
+        value = super(MongoEmbedded, self).clean(value)
+        return self.document_class(**value)
+
 class MongoReference(Field):
     base_type = basestring
 

diff --git a/tests/__init__.py b/tests/__init__.py
@@ -136,7 +136,7 @@ class OptionalTagsSchema(Schema):
         self.assertValid(OptionalTagsSchema({}), {'tags': []})
 
 
-    # TODO: Test Email, Embedded, Choices, MongoReference, more Schema tests.
+    # TODO: Test Embedded, Choices, MongoReference, more Schema tests.
 
 if __name__ == '__main__':
     unittest.main()