From 560af9e0cf7d7ee9e282ab4c6398783dd267738b Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Wed, 16 Aug 2023 12:10:24 +0100 Subject: [PATCH] Added property id to plan creation --- backend/app/db/models/recommendations.py | 1 + backend/app/plan/router.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/backend/app/db/models/recommendations.py b/backend/app/db/models/recommendations.py index 9c11ab83..60325562 100644 --- a/backend/app/db/models/recommendations.py +++ b/backend/app/db/models/recommendations.py @@ -44,6 +44,7 @@ class Plan(Base): id = Column(BigInteger, primary_key=True, autoincrement=True) portfolio_id = Column(BigInteger, ForeignKey(Portfolio.id), nullable=False) + property_id = Column(BigInteger, ForeignKey(PropertyModel.id), nullable=False) created_at = Column(TIMESTAMP, nullable=False, server_default=func.now()) is_default = Column(Boolean, nullable=False) diff --git a/backend/app/plan/router.py b/backend/app/plan/router.py index 96f591ba..dd66639f 100644 --- a/backend/app/plan/router.py +++ b/backend/app/plan/router.py @@ -245,6 +245,7 @@ async def trigger_plan(body: PlanTriggerRequest): # 2) the property details (epc) # 3) the recommendations + logger.info("Uploading recommendations to the database") # Upload property data for p in input_properties: property_details_epc = p.get_property_details_epc(portfolio_id=body.portfolio_id, rating_lookup=rating_lookup) @@ -265,6 +266,7 @@ async def trigger_plan(body: PlanTriggerRequest): new_plan_id = create_plan( { "portfolio_id": body.portfolio_id, + "property_id": p.id, "is_default": True } )