Cosmetic: mostly typings

src/gisaf/models/authentication.py

@@ -1,10 +1,11 @@
 from sqlmodel import Field, SQLModel, Relationship
+from pydantic import BaseModel
 
 from gisaf.models.metadata import gisaf_admin
 
 
 class UserRoleLink(SQLModel, table=True):
-    __tablename__ = 'roles_users'
+    __tablename__: str = 'roles_users' # type: ignore
     __table_args__ = gisaf_admin.table_args
     user_id: int | None = Field(
         default=None,
@@ -56,7 +57,7 @@ class Role(RoleWithDescription, table=True):
 
 class UserReadNoRoles(UserBase):
     id: int
-    email: str | None
+    email: str | None # type: ignore
 
 
 class RoleRead(RoleBase):
@@ -70,5 +71,10 @@ class RoleReadNoUsers(RoleBase):
 
 class UserRead(UserBase):
     id: int
-    email: str | None
-    roles: list[RoleReadNoUsers] = []
+    email: str | None # type: ignore
+    roles: list[RoleReadNoUsers] = []
+
+
+# class ACL(BaseModel):
+#     user_id: int
+#     role_ids: list[int]

src/gisaf/models/bootstrap.py

@@ -16,4 +16,4 @@ class BootstrapData(BaseModel):
     geo: Geo = conf.geo
     measures: Measures = conf.measures
     redirect: str = conf.gisaf.redirect
-    user: UserRead | None = None
+    user: UserRead | None = None # type: ignore

src/gisaf/models/category.py

@@ -15,9 +15,9 @@ mapbox_type_mapping = {
 
 
 class CategoryGroup(BaseModel, table=True):
-    __tablename__ = 'category_group'
+    __tablename__: str = 'category_group' # type: ignore
     __table_args__ = gisaf_survey.table_args
-    name: str | None = Field(sa_type=String(4), default=None, primary_key=True)
+    name: str | None = Field(sa_type=String(4), default=None, primary_key=True) # type: ignore
     major: bool
     long_name: str
     categories: list['Category'] = Relationship(back_populates='category_group')
@@ -28,7 +28,7 @@ class CategoryGroup(BaseModel, table=True):
 
 
 class CategoryModelType(BaseModel, table=True):
-    __tablename__ = 'category_model_type'
+    __tablename__: str = 'category_model_type' # type: ignore
     __table_args__ = gisaf_survey.table_args
     name: str | None = Field(default=None, primary_key=True)
 
@@ -46,27 +46,27 @@ class CategoryBase(BaseModel):
     name: str | None = Field(default=None, primary_key=True)
     domain: ClassVar[str] = 'V'
     description: str | None
-    group: str = Field(sa_type=String(4),
-                       foreign_key=gisaf_survey.table('category_group.name'),
-                       index=True)
-    minor_group_1: str = Field(sa_type=String(4), default='----')
-    minor_group_2: str = Field(sa_type=String(4), default='----')
-    status: str = Field(sa_type=String(1))
+    group: str = Field(sa_type=String(4), # type: ignore
+                       foreign_key=gisaf_survey.table('category_group.name'), # type: ignore
+                       index=True) # type: ignore
+    minor_group_1: str = Field(sa_type=String(4), default='----') # type: ignore
+    minor_group_2: str = Field(sa_type=String(4), default='----') # type: ignore
+    status: str = Field(sa_type=String(1)) # type: ignore
     custom: bool | None
     auto_import: bool = True
-    gis_type: str = Field(sa_type=String(50),
-                          foreign_key=gisaf_survey.table('category_model_type.name'),
-                          default='Point')
-    long_name: str | None = Field(sa_type=String(50))
+    gis_type: str = Field(sa_type=String(50), # type: ignore
+                          foreign_key=gisaf_survey.table('category_model_type.name'), # type: ignore
+                          default='Point') # type: ignore
+    long_name: str | None = Field(sa_type=String(50)) # type: ignore
     style: str | None = Field(sa_type=TEXT)
-    symbol: str | None = Field(sa_type=String(1))
-    mapbox_type_custom: str | None = Field(sa_type=String(12))
-    mapbox_paint: dict[str, Any] | None = Field(sa_type=JSON(none_as_null=True))
-    mapbox_layout: dict[str, Any] | None = Field(sa_type=JSON(none_as_null=True))
+    symbol: str | None = Field(sa_type=String(1)) # type: ignore
+    mapbox_type_custom: str | None = Field(sa_type=String(12)) # type: ignore
+    mapbox_paint: dict[str, Any] | None = Field(sa_type=JSON(none_as_null=True)) # type: ignore
+    mapbox_layout: dict[str, Any] | None = Field(sa_type=JSON(none_as_null=True)) # type: ignore
     viewable_role: str | None
-    extra: dict[str, Any] | None = Field(sa_type=JSON(none_as_null=True))
+    extra: dict[str, Any] | None = Field(sa_type=JSON(none_as_null=True)) # type: ignore
 
-    @computed_field
+    @computed_field # type: ignore
     @property
     def layer_name(self) -> str:
         """
@@ -75,7 +75,7 @@ class CategoryBase(BaseModel):
         """
         return '{self.domain}-{self.group:4s}-{self.minor_group_1:4s}-{self.minor_group_2:4s}-{self.status:1s}'.format(self=self)
 
-    @computed_field
+    @computed_field # type: ignore
     @property
     def table_name(self) -> str:
         """
@@ -87,7 +87,7 @@ class CategoryBase(BaseModel):
         else:
             return '{self.domain}_{self.group:4s}_{self.minor_group_1:4s}_{self.minor_group_2:4s}'.format(self=self)
 
-    @computed_field
+    @computed_field # type: ignore
     @property
     def raw_survey_table_name(self) -> str:
         """
@@ -99,7 +99,7 @@ class CategoryBase(BaseModel):
         else:
             return 'RAW_{self.domain}_{self.group:4s}_{self.minor_group_1:4s}_{self.minor_group_2:4s}'.format(self=self)
 
-    @computed_field
+    @computed_field # type: ignore
     @property
     def mapbox_type(self) -> str:
         return self.mapbox_type_custom or mapbox_type_mapping[self.gis_type]

src/gisaf/models/dashboard.py

@@ -18,11 +18,11 @@ logger = logging.getLogger(__name__)
 try:
     import matplotlib.pyplot as plt
 except ImportError:
-    plt = None
+    plt = None # type: ignore
 
 
 class DashboardPageSource(Model, table=True):
-    __tablename__ = 'dashboard_page_source'
+    __tablename__ = 'dashboard_page_source' # type: ignore
     __table_args__ = gisaf.table_args
 
     id: str = Field(primary_key=True)
@@ -118,7 +118,7 @@ class DashboardPageCommon:
 
 
 class DashboardPage(Model, DashboardPageCommon, table=True):
-    __tablename__ = 'dashboard_page'
+    __tablename__ = 'dashboard_page' # type: ignore
     __table_args__ = gisaf.table_args
 
     class Admin:
@@ -179,7 +179,7 @@ class DashboardPage(Model, DashboardPageCommon, table=True):
 
 
 class DashboardPageSection(Model, DashboardPageCommon, table=True):
-    __tablename__ = 'dashboard_page_section'
+    __tablename__ = 'dashboard_page_section' # type: ignore
     __table_args__ = gisaf.table_args
 
     class Admin:
@@ -252,10 +252,10 @@ class DashboardPageSection(Model, DashboardPageCommon, table=True):
 
 
 class Widget(Model, table=True):
-    __tablename__ = 'widget'
+    __tablename__ = 'widget' # type: ignore
     __table_args__ = gisaf.table_args
     ## CREATE TABLE gisaf.widget (name char(50) not null PRIMARY KEY, title varchar, subtitle varchar, notebook varchar, content varchar, time timestamp);
-    name: str = Field(primary_key=True, sa_type=String(50))
+    name: str = Field(primary_key=True, sa_type=String(50)) # type: ignore
     title: str
     subtitle: str
     content: str

src/gisaf/models/geo_models_base.py

@@ -91,10 +91,10 @@ class BaseSurveyModel(BaseModel):
     @classmethod
     def selectinload(cls):
         return [
-            cls.equipment,
-            cls.surveyor,
-            cls.accuracy,
-            cls.project,
+            cls.equipment, # type: ignore
+            cls.surveyor, # type: ignore
+            cls.accuracy, # type: ignore
+            cls.project, # type: ignore
         ]
     # @classmethod
     # def dyn_join_with(cls):

src/gisaf/models/store.py

@@ -6,7 +6,12 @@ from gisaf.models.geo_models_base import GeoModel, RawSurveyBaseModel, GeoPointS
 class MapLibreStyle(BaseModel):
     ...
 
-class Store(BaseModel):
+
+class StoreNameOnly(BaseModel):
+    name: str
+
+
+class Store(StoreNameOnly):
     auto_import: bool
     # base_gis_type: str
     count: int | None = None
@@ -29,7 +34,6 @@ class Store(BaseModel):
     minor_group_2: str | None
     #model: GeoModel
     gis_type: str
-    name: str
     #name_letter: str
     #name_number: int
     #raw_model: GeoPointSurveyModel

src/gisaf/models/survey.py

@@ -1,9 +1,13 @@
 from enum import Enum
 
 from sqlmodel import Field, Relationship
+from pydantic import BaseModel
 
+from gisaf.config import BasketDefault
 from gisaf.models.models_base import Model
 from gisaf.models.metadata import gisaf_survey
+from gisaf.models.project import Project
+from gisaf.models.store import StoreNameOnly
 
 
 class Accuracy(Model, table=True):
@@ -63,7 +67,7 @@ class GeometryType(str, Enum):
 
 class AccuracyEquimentSurveyorMapping(Model, table=True):
     __table_args__ = gisaf_survey.table_args
-    __tablename__ = 'accuracy_equiment_surveyor_mapping'
+    __tablename__: str = 'accuracy_equiment_surveyor_mapping' # type: ignore
 
     class Admin:
         menu = 'Other'
@@ -91,4 +95,13 @@ class AccuracyEquimentSurveyorMapping(Model, table=True):
     #         'surveyor': Surveyor,
     #         'equipment': Equipment,
     #         'accuracy': Accuracy,
-    #     }
+    #     }
+
+class SurveyMeta(BaseModel):
+    projects: list[Project]
+    surveyors: list[Surveyor]
+    equipments: list[Equipment]
+    statuses: list[str]
+    stores_misc: list[StoreNameOnly]
+    stores_line_work: list[StoreNameOnly]
+    default: BasketDefault

src/gisaf/models/tags.py

@@ -19,7 +19,7 @@ class Tags(GeoPointModel, table=True):
     id: int | None = Field(primary_key=True, default=None)
     store: str = Field(index=True)
     ref_id: int = Field(index=True, sa_type=BigInteger)
-    tags: dict = Field(sa_type=MutableDict.as_mutable(HSTORE))
+    tags: dict = Field(sa_type=MutableDict.as_mutable(HSTORE)) # type: ignore
 
     def __str__(self):
         return '{self.store:s} {self.ref_id}: {self.tags}'.format(self=self)