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

identify site notes files after file upload so we can save them to epc_property table

Browse source
This commit is contained in:
Daniel Roth 2026-04-28 16:33:53 +00:00
parent 0069fc3b7b
commit 1bc8343738
2 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
backend/app/db/models/epc_property.py
View file

@ -20,6 +20,7 @@ class EpcPropertyModel(SQLModel, table=True):
id: Optional[int] = Field(default=None, primary_key=True) id: Optional[int] = Field(default=None, primary_key=True)
property_id: Optional[int] = Field(default=None) property_id: Optional[int] = Field(default=None)
portfolio_id: Optional[int] = Field(default=None) portfolio_id: Optional[int] = Field(default=None)
uploaded_file_id: Optional[int] = Field(default=None)
# Identity / admin # Identity / admin
uprn: Optional[int] = Field(default=None) uprn: Optional[int] = Field(default=None)

15
backend/pashub_fetcher/handler/handler.py
View file

@ -8,6 +8,7 @@ from backend.app.config import get_settings
from backend.app.db.connection import db_session from backend.app.db.connection import db_session
from backend.app.db.models.uploaded_file import ( from backend.app.db.models.uploaded_file import (
FileSourceEnum, FileSourceEnum,
FileTypeEnum,
UploadedFile, UploadedFile,
) )
from backend.pashub_fetcher.core_files import infer_file_type from backend.pashub_fetcher.core_files import infer_file_type
@ -141,8 +142,22 @@ def upload_job_to_s3_and_update_db(
with db_session() as session: with db_session() as session:
session.add_all(uploaded_files) session.add_all(uploaded_files)
# Ensure IDs are generated
session.flush()
results = [
{"file": file.s3_file_key, "type": file.file_type, "id": file.id}
for file in uploaded_files
]
session.commit() session.commit()
for result in results:
if FileTypeEnum(result["type"]) == FileTypeEnum.RD_SAP_SITE_NOTE:
# upload site notes to epc_property table
continue
pass pass