junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-30 13:10:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use correct db session to write to db

Browse source
This commit is contained in:
Daniel Roth 2026-06-08 14:07:03 +00:00
parent aaeb339254
commit 3136d5a0f8
2 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
applications/magic_plan/local_handler/invoke_local_orchestrator.py
View file

@ -27,8 +27,8 @@ from infrastructure.s3.s3_client import S3Client
from orchestration.magic_plan_orchestrator import MagicPlanOrchestrator from orchestration.magic_plan_orchestrator import MagicPlanOrchestrator
from applications.magic_plan.magic_plan_trigger_request import MagicPlanTriggerRequest from applications.magic_plan.magic_plan_trigger_request import MagicPlanTriggerRequest
ADDRESS = "63 Dunkery Road, Wythenshawe, M22 0WR | EPC" ADDRESS = "20 Larch Way, Bromley, BR2 8DU | Retrofit Assessment"
HUBSPOT_DEAL_ID = "501851906250" HUBSPOT_DEAL_ID = "500328089833"
# ADDRESS = "33 Wallaby Way, Sydney" # ADDRESS = "33 Wallaby Way, Sydney"
# HUBSPOT_DEAL_ID = "123456789" # HUBSPOT_DEAL_ID = "123456789"

15
orchestration/magic_plan_orchestrator.py
View file

@ -15,7 +15,7 @@ from backend.app.db.models.uploaded_file import (
) )
from applications.magic_plan.address_matcher import find_matching_plan from applications.magic_plan.address_matcher import find_matching_plan
from infrastructure.postgres.config import PostgresConfig from infrastructure.postgres.config import PostgresConfig
from infrastructure.postgres.engine import make_engine, make_session from infrastructure.postgres.engine import make_engine, transactional_session
from infrastructure.s3.s3_client import S3Client from infrastructure.s3.s3_client import S3Client
from repositories.magic_plan.magic_plan_postgres_repository import ( from repositories.magic_plan.magic_plan_postgres_repository import (
MagicPlanPostgresRepository, MagicPlanPostgresRepository,
@ -69,13 +69,12 @@ class MagicPlanOrchestrator:
logger.info("Successfully save Plan json") logger.info("Successfully save Plan json")
engine = make_engine(PostgresConfig.from_env(os.environ)) engine = make_engine(PostgresConfig.from_env(os.environ))
session = make_session(engine) with transactional_session(engine) as session:
session.add(uploaded_file)
session.add(uploaded_file) session.flush()
session.flush() MagicPlanPostgresRepository(session).save(
MagicPlanPostgresRepository(session).save( plan, cast(int, uploaded_file.id)
plan, cast(int, uploaded_file.id) ) # TODO: refactor to use postgres Unit of Work
) # TODO: refactor to use postgres Unit of Work
logger.info("Successfully saved Plan to database") logger.info("Successfully saved Plan to database")
return plan return plan