104 lines
3.3 KiB
Python
104 lines
3.3 KiB
Python
|
from typing import Any
|
||
|
|
from sqlmodel import Field, SQLModel, MetaData, JSON, TEXT, Relationship, Column
|
||
|
|
|
||
|
|
schema = 'gisaf_survey'
|
||
|
|
metadata = MetaData(schema=schema)
|
||
|
|
|
||
|
|
mapbox_type_mapping = {
|
||
|
|
'Point': 'symbol',
|
||
|
|
'Line': 'line',
|
||
|
|
'Polygon': 'fill',
|
||
|
|
}
|
||
|
|
|
||
|
|
class CategoryGroup(SQLModel, table=True):
|
||
|
|
metadata = metadata
|
||
|
|
name: str = Field(min_length=4, max_length=4,
|
||
|
|
default=None, primary_key=True)
|
||
|
|
major: str
|
||
|
|
long_name: str
|
||
|
|
|
||
|
|
class Admin:
|
||
|
|
menu = 'Other'
|
||
|
|
flask_admin_model_view = 'CategoryGroupModelView'
|
||
|
|
|
||
|
|
|
||
|
|
class CategoryModelType(SQLModel, table=True):
|
||
|
|
metadata = metadata
|
||
|
|
name: str = Field(default=None, primary_key=True)
|
||
|
|
|
||
|
|
class Admin:
|
||
|
|
menu = 'Other'
|
||
|
|
flask_admin_model_view = 'MyModelViewWithPrimaryKey'
|
||
|
|
|
||
|
|
|
||
|
|
class CategoryBase(SQLModel):
|
||
|
|
metadata = metadata
|
||
|
|
|
||
|
|
class Admin:
|
||
|
|
menu = 'Other'
|
||
|
|
flask_admin_model_view = 'CategoryModelView'
|
||
|
|
|
||
|
|
name: str | None = Field(default=None, primary_key=True)
|
||
|
|
description: str | None
|
||
|
|
group: str = Field(min_length=4, max_length=4,
|
||
|
|
foreign_key="CategoryGroup.name", index=True)
|
||
|
|
#group_: CategoryGroup = Relationship()
|
||
|
|
minor_group_1: str = Field(min_length=4, max_length=4, default='----')
|
||
|
|
minor_group_2: str = Field(min_length=4, max_length=4, default='----')
|
||
|
|
status: str = Field(min_length=1, max_length=1)
|
||
|
|
custom: bool | None
|
||
|
|
auto_import: bool = True
|
||
|
|
model_type: str = Field(max_length=50,
|
||
|
|
foreign_key='CategoryModelType.name', default='Point')
|
||
|
|
long_name: str | None = Field(max_length=50)
|
||
|
|
style: str | None = Field(sa_column=Column(TEXT))
|
||
|
|
symbol: str | None = Field(max_length=1)
|
||
|
|
mapbox_type_custom: str | None = Field(max_length=32)
|
||
|
|
mapbox_paint: dict[str, Any] | None = Field(sa_column=Column(JSON, none_as_null=True))
|
||
|
|
mapbox_layout: dict[str, Any] | None = Field(sa_column=Column(JSON, none_as_null=True))
|
||
|
|
viewable_role: str | None
|
||
|
|
extra: dict[str, Any] | None = Field(sa_column=Column(JSON, none_as_null=True))
|
||
|
|
|
||
|
|
|
||
|
|
class Category(CategoryBase, table=True):
|
||
|
|
name: str = Field(default=None, primary_key=True)
|
||
|
|
|
||
|
|
|
||
|
|
class CategoryRead(CategoryBase):
|
||
|
|
name: str
|
||
|
|
domain = 'V' # Survey
|
||
|
|
|
||
|
|
@property
|
||
|
|
def layer_name(self):
|
||
|
|
"""
|
||
|
|
ISO compliant layer name (see ISO 13567)
|
||
|
|
:return: str
|
||
|
|
"""
|
||
|
|
return '{self.domain}-{self.group:4s}-{self.minor_group_1:4s}-{self.minor_group_2:4s}-{self.status:1s}'.format(self=self)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def table_name(self):
|
||
|
|
"""
|
||
|
|
Table name
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
if self.minor_group_2 == '----':
|
||
|
|
return '{self.domain}_{self.group:4s}_{self.minor_group_1:4s}'.format(self=self)
|
||
|
|
else:
|
||
|
|
return '{self.domain}_{self.group:4s}_{self.minor_group_1:4s}_{self.minor_group_2:4s}'.format(self=self)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def raw_survey_table_name(self):
|
||
|
|
"""
|
||
|
|
Table name
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
if self.minor_group_2 == '----':
|
||
|
|
return 'RAW_{self.domain}_{self.group:4s}_{self.minor_group_1:4s}'.format(self=self)
|
||
|
|
else:
|
||
|
|
return 'RAW_{self.domain}_{self.group:4s}_{self.minor_group_1:4s}_{self.minor_group_2:4s}'.format(self=self)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def mapbox_type(self):
|
||
|
|
return self.mapbox_type_custom or mapbox_type_mapping[self.model_type]
|