junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-07-12 13:29:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

bug fixes to get runner working

Browse source
This commit is contained in:
Daniel Roth 2026-04-08 07:42:00 +00:00
parent d229e2faf8
commit 7cd4d4c5b3
2 changed files with 18 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
backend/app/db/models/uploaded_file.py
View file

@ -39,9 +39,21 @@ class UploadedFile(Base):
hubspot_listing_id = Column(BigInteger, nullable=True) hubspot_listing_id = Column(BigInteger, nullable=True)
file_type = Column( file_type = Column(
SqlEnum(FileTypeEnum, name="file_type", create_type=False), nullable=True 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( file_source = Column(
SqlEnum(FileSourceEnum, name="file_source", create_type=False), nullable=True SqlEnum(
FileSourceEnum,
name="file_source",
create_type=False,
values_callable=lambda enum_cls: [e.value for e in enum_cls],
),
nullable=True,
) )

6
backend/ecmk_fetcher/upload.py
View file

@ -31,7 +31,9 @@ def upload_file_to_sharepoint(
def upload_file_to_s3_and_update_db( def upload_file_to_s3_and_update_db(
bucket: str, file_path: str, hubspot_listing_id: str, file_type: FileTypeEnum bucket: str, file_path: str, hubspot_listing_id: str, file_type: FileTypeEnum
) -> None: ) -> None:
key: str = f"documents/hubspot_listing_id/{hubspot_listing_id}" filename: str = os.path.basename(file_path)
key: str = f"documents/hubspot_listing_id/{hubspot_listing_id}/{filename}"
upload_file_to_s3(file_path, bucket, key) upload_file_to_s3(file_path, bucket, key)
uploaded_file = UploadedFile( uploaded_file = UploadedFile(
@ -40,7 +42,7 @@ def upload_file_to_s3_and_update_db(
s3_upload_timestamp=datetime.now(timezone.utc), s3_upload_timestamp=datetime.now(timezone.utc),
hubspot_listing_id=hubspot_listing_id, hubspot_listing_id=hubspot_listing_id,
file_source=FileSourceEnum.ECMK.value, file_source=FileSourceEnum.ECMK.value,
file_type=file_type, file_type=file_type.value,
) )
with db_session() as session: with db_session() as session: