mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import enum
|
|
from sqlalchemy import TIMESTAMP, BigInteger, Column, Text, Enum as SqlEnum
|
|
|
|
from backend.app.db.base import Base
|
|
|
|
|
|
class FileTypeEnum(enum.Enum):
|
|
PHOTO_PACK = "photo_pack"
|
|
SITE_NOTE = "site_note"
|
|
RD_SAP_SITE_NOTE = "rd_sap_site_note"
|
|
PAS_2023_VENTILATION = "pas_2023_ventilation"
|
|
PAS_2023_CONDITION = "pas_2023_condition"
|
|
PAS_SIGNIFICANCE = "pas_significance"
|
|
PAR_PHOTO_PACK = "par_photo_pack"
|
|
PAS_2023_PROPERTY = "pas_2023_property"
|
|
PAS_2023_OCCUPANCY = "pas_2023_occupancy"
|
|
ECMK_SITE_NOTE = "ecmk_site_note"
|
|
ECMK_RD_SAP_SITE_NOTE = "ecmk_rd_sap_site_note"
|
|
ECMK_SURVEY_XML = "ecmk_survey_xml"
|
|
|
|
|
|
class FileSourceEnum(enum.Enum):
|
|
PAS_HUB = "pas hub"
|
|
SHAREPOINT = "sharepoint"
|
|
HUBSPOT = "hubspot"
|
|
ECMK = "ecmk"
|
|
|
|
|
|
class UploadedFile(Base):
|
|
__tablename__ = "uploaded_files"
|
|
|
|
id = Column(BigInteger, primary_key=True, autoincrement=True)
|
|
|
|
s3_file_bucket = Column(Text, nullable=False)
|
|
s3_file_key = Column(Text, nullable=False)
|
|
s3_upload_timestamp = Column(TIMESTAMP(timezone=True), nullable=False)
|
|
|
|
landlord_property_id = Column(Text, nullable=True)
|
|
uprn = Column(BigInteger, nullable=True)
|
|
hubspot_listing_id = Column(BigInteger, nullable=True)
|
|
|
|
file_type = Column(
|
|
SqlEnum(
|
|
FileTypeEnum,
|
|
name="file_type",
|
|
create_type=False,
|
|
values_callable=lambda enum_cls: [e.value for e in enum_cls],
|
|
),
|
|
nullable=True,
|
|
)
|
|
|
|
file_source = Column(
|
|
SqlEnum(
|
|
FileSourceEnum,
|
|
name="file_source",
|
|
create_type=False,
|
|
values_callable=lambda enum_cls: [e.value for e in enum_cls],
|
|
),
|
|
nullable=True,
|
|
)
|