mypy type

Author: MaxNumerique

src/opengeodeweb_microservice/database/connection.py

@@ -23,7 +23,7 @@ def init_database(db_path: str = DATABASE_FILENAME) -> None:
         Base.metadata.create_all(engine)
 
 
-def get_session():
+def get_session() -> scoped_session:
     if scoped_session_registry is None:
         raise RuntimeError()
     return scoped_session_registry

src/opengeodeweb_microservice/database/data.py

@@ -3,6 +3,7 @@
 from .connection import get_session
 from .base import Base
 import uuid
+from typing import cast
 
 
 class Data(Base):
@@ -46,4 +47,4 @@ def create(
     @staticmethod
     def get(data_id: str) -> "Data | None":
         session = get_session()
-        return session.get(Data, data_id)
+        return cast("Data | None", session.get(Data, data_id))