add type for sub-class of declarative_base()

input_file to str

Author: MaxNumerique

src/opengeodeweb_back/data.py

@@ -1,14 +1,17 @@
 from sqlalchemy import String, JSON
 from .database import database
 import uuid
-from typing import List
+from typing import List, Type
 
+DatabaseModel : Type = database.Model
 
 class Data(database.Model):
     __tablename__ = "datas"
 
     id = database.Column(
-        String, primary_key=True, default=lambda: str(uuid.uuid4()).replace("-", "")
+        String,
+        primary_key=True,
+        default=lambda: str(uuid.uuid4()).replace("-", "")
     )
     name = database.Column(String, nullable=False)
     native_file_name = database.Column(String, nullable=False)
@@ -22,9 +25,11 @@ class Data(database.Model):
     def create(
         name: str,
         geode_object: str,
-        input_file: str, 
+        input_file: str,
         additional_files: List[str]
-    ) -> 'Data':
+    ) -> "Data":
+        if input_file is None:
+            input_file = []
         if additional_files is None:
             additional_files = []
 

src/opengeodeweb_back/utils_functions.py

@@ -214,7 +214,7 @@ def generate_native_viewable_and_light_viewable_from_file(
     temp_data_entry = Data.create(
         name="temp",
         geode_object=geode_object,
-        input_file=[input_filename],
+        input_file=input_filename,
         additional_files=[],
     )