allow default value in enums

Author: minenwerfer

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = 'normalize_json'
-version = '0.0.16'
+version = '0.0.17'
 authors = [
   { name = 'João Gabriel Santos', email = '[email protected]' }
 ]

src/normalize_json/normalize.py

@@ -261,19 +261,21 @@ def translate(
             if node.get('array') and not isinstance(value, list):
                 value = [value]
 
+            default = node.get('default')
             if err := check_types(node, value, modifiers):
-                raise ValueError('check_types @ %s (got "%s", expected "%s")' % (original_name, *err))
+                if not default:
+                    raise ValueError('check_types @ %s (got "%s", expected "%s")' % (original_name, *err))
 
             if (node_enum := node.get('enum')) and value != None:
                 if node.get('array'):
                     if not isinstance(value, list):
                         raise TypeError()
                     value = [
-                        node.get('enum', {}).get(v)
+                        node.get('enum', {}).get(v, default)
                         for v in typing.cast(list[str], value)
                     ]
                 else:
-                    value = node_enum.get(typing.cast(typing.Any, value))
+                    value = node_enum.get(typing.cast(typing.Any, value), default)
 
             ret[original_name] = value
 

tests/translate.py

@@ -131,15 +131,16 @@
                 'pago': 'paid',
                 'pendente': 'pending',
                 'recusado': 'refused',
-            }
+            },
         },
         'statuses': {
             'array': True,
             'enum': {
                 'pago': 'paid',
                 'pendente': 'pending',
                 'recusado': 'refused',
-            }
+            },
+            'default': 'pending'
         },
         'items': {
             'map': 'items',
@@ -181,6 +182,7 @@
     'statuses': [
         'pendente',
         'recusado',
+        'invalid',
     ],
     'items': {
         'meta': {
@@ -229,6 +231,7 @@ def test_translate_enums(self):
         self.assertEqual(result['status'], 'paid')
         self.assertEqual(result['statuses'][0], 'pending')
         self.assertEqual(result['statuses'][1], 'refused')
+        self.assertEqual(result['statuses'][2], 'pending')
 
 
     def test_translate_substitute(self):