Inline builder._add_child_field

Scille · Jul 20, 2020 · 36e77a9 · 36e77a9
1 parent 2fd5272
commit 36e77a9
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/umongo/builder.py b/umongo/builder.py
@@ -92,10 +92,6 @@ def find_id_field(fields_dict):
     return 'id'
 
 
-def _add_child_field(name, fields_dict):
-    fields_dict['cls'] = fields.StringField(attribute='_cls', default=name, dump_only=True)
-
-
 def _collect_schema_attrs(template):
     """
     Split dict between schema fields and non-fields elements and retrieve
@@ -293,7 +289,9 @@ def build_from_template(self, template):
 
         if base_tmpl_cls is not MixinDocumentTemplate:
             if is_child:
-                _add_child_field(name, schema_fields)
+                schema_fields['cls'] = fields.StringField(
+                    attribute='_cls', default=name, dump_only=True
+                )
         schema_cls = self._build_schema(template, schema_bases, schema_fields, schema_non_fields)
         nmspc['Schema'] = schema_cls
         schema = schema_cls()