Convert magns into floats

gisce · Oct 8, 2024 · 5285c2c · 5285c2c
1 parent 6140cd0
commit 5285c2c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cchloader/adapters/corbagen.py b/cchloader/adapters/corbagen.py
@@ -13,7 +13,7 @@ class CorbaGenBaseAdapter(Schema):
     @pre_load
     def fix_numbers(self, data):
         for attr, field in self.fields.items():
-            if isinstance(field, fields.Integer):
+            if isinstance(field, (fields.Integer, fields.Float)):
                 if not data.get(attr):
                     data[attr] = None
         return data

diff --git a/cchloader/models/corbagen.py b/cchloader/models/corbagen.py
@@ -9,8 +9,8 @@ class CorbaGenSchema(Schema):
     name = fields.String(position=0, required=True)
     local_timestamp = fields.DateTime(position=1, format='%Y-%m-%d %H:%M', required=True)
     season = fields.Integer(position=2, validate=OneOf([0, 1]), required=True)
-    generacio = fields.Integer(position=3, required=True)
-    autocons = fields.Integer(position=4, required=True)
-    excedent = fields.Integer(position=5, required=True)
+    generacio = fields.Float(position=3, required=True)
+    autocons = fields.Float(position=4, required=True)
+    excedent = fields.Float(position=5, required=True)
 
 CorbaGenSchema()