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

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

View file

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

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.models.uploaded_file import (
FileSourceEnum,
FileTypeEnum,
UploadedFile,
)
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:
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()
for result in results:
if FileTypeEnum(result["type"]) == FileTypeEnum.RD_SAP_SITE_NOTE:
# upload site notes to epc_property table
continue
pass