mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
identify site notes files after file upload so we can save them to epc_property table
This commit is contained in:
parent
0069fc3b7b
commit
1bc8343738
2 changed files with 16 additions and 0 deletions
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue