117 lines
2.8 KiB
Python
117 lines
2.8 KiB
Python
from enum import Enum
|
|
from pydantic import BaseModel
|
|
from typing import Optional, Union, Any
|
|
|
|
class UserType(Enum):
|
|
|
|
employee = 1
|
|
occupant = 2
|
|
|
|
|
|
class Credentials(BaseModel):
|
|
|
|
person_id: int
|
|
person_name: str
|
|
|
|
|
|
class ApplicationToken(BaseModel):
|
|
# Application Token Object -> is the main object for the user
|
|
|
|
user_type: int = UserType.occupant.value
|
|
credential_token: str = ""
|
|
|
|
user_uu_id: str
|
|
user_id: int
|
|
|
|
person_id: int
|
|
person_uu_id: str
|
|
|
|
request: Optional[dict] = None # Request Info of Client
|
|
expires_at: Optional[float] = None # Expiry timestamp
|
|
|
|
# ID list of reachable event codes as "endpoint_code": ["UUID", "UUID"]
|
|
reachable_event_codes: Optional[dict[str, Any]] = None
|
|
|
|
# ID list of reachable applications as "page_url": ["UUID", "UUID"]
|
|
reachable_app_codes: Optional[dict[str, Any]] = None
|
|
|
|
|
|
class OccupantToken(BaseModel):
|
|
|
|
# Selection of the occupant type for a build part is made by the user
|
|
|
|
living_space_id: int # Internal use
|
|
living_space_uu_id: str # Outer use
|
|
|
|
occupant_type_id: int
|
|
occupant_type_uu_id: str
|
|
occupant_type: str
|
|
|
|
build_id: int
|
|
build_uuid: str
|
|
build_part_id: int
|
|
build_part_uuid: str
|
|
|
|
responsible_company_id: Optional[int] = None
|
|
responsible_company_uuid: Optional[str] = None
|
|
responsible_employee_id: Optional[int] = None
|
|
responsible_employee_uuid: Optional[str] = None
|
|
|
|
|
|
class CompanyToken(BaseModel):
|
|
|
|
# Selection of the company for an employee is made by the user
|
|
company_id: int
|
|
company_uu_id: str
|
|
|
|
department_id: int # ID list of departments
|
|
department_uu_id: str # ID list of departments
|
|
|
|
duty_id: int
|
|
duty_uu_id: str
|
|
|
|
staff_id: int
|
|
staff_uu_id: str
|
|
|
|
employee_id: int
|
|
employee_uu_id: str
|
|
bulk_duties_id: int
|
|
|
|
|
|
class OccupantTokenObject(ApplicationToken):
|
|
# Occupant Token Object -> Requires selection of the occupant type for a specific build part
|
|
|
|
available_occupants: dict = None
|
|
selected_occupant: Optional[dict] = None # Selected Occupant Type
|
|
|
|
@property
|
|
def is_employee(self) -> bool:
|
|
return False
|
|
|
|
@property
|
|
def is_occupant(self) -> bool:
|
|
return True
|
|
|
|
|
|
class EmployeeTokenObject(ApplicationToken):
|
|
# Full hierarchy Employee[staff_id] -> Staff -> Duty -> Department -> Company
|
|
|
|
companies_id_list: list[int] # List of company objects
|
|
companies_uu_id_list: list[str] # List of company objects
|
|
|
|
duty_id_list: list[int] # List of duty objects
|
|
duty_uu_id_list: list[str] # List of duty objects
|
|
|
|
selected_company: Optional[dict] = None # Selected Company Object
|
|
|
|
@property
|
|
def is_employee(self) -> bool:
|
|
return True
|
|
|
|
@property
|
|
def is_occupant(self) -> bool:
|
|
return False
|
|
|
|
|
|
TokenDictType = Union[EmployeeTokenObject, OccupantTokenObject]
|