From aadcc56d32547addb06af616a192b5c1446b03fd Mon Sep 17 00:00:00 2001
From: Khalim Conn-Kowlessar <kconnkowlessar@gmail.com>
Date: Wed, 2 Oct 2024 14:33:29 +0100
Subject: [PATCH] adding database queries to Outputs class

---
 backend/Outputs.py | 102 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 102 insertions(+)
 create mode 100644 backend/Outputs.py

diff --git a/backend/Outputs.py b/backend/Outputs.py
new file mode 100644
index 00000000..a846965b
--- /dev/null
+++ b/backend/Outputs.py
@@ -0,0 +1,102 @@
+from sqlalchemy.orm import sessionmaker
+
+from backend.app.db.connection import db_engine
+from backend.app.db.models.portfolio import PropertyModel, PropertyDetailsEpcModel
+from backend.app.db.models.recommendations import Recommendation, Plan, PlanRecommendations, Scenario
+
+
+class Outputs:
+    FORMATS = ["mds"]
+
+    def __init__(self, format, portfolio_id):
+        """
+        This class handles the creation of standard outputs for the backend. For example, creation of
+        an excel output, to be used for the MDS data sheet, required by E.ON
+
+        :param format: The format of the output, e.g. mds
+        :param portfolio_id: The id of the portfolio for which the output is being created
+        """
+
+        if format not in self.FORMATS:
+            raise ValueError("Invalid format, should be one of {}".format(self.FORMATS))
+
+        self.format = format
+        self.portfolio_id = portfolio_id
+
+        # Connect to the database
+        self.session = sessionmaker(bind=db_engine)()
+
+    def get_properties_from_db(self):
+        # Get properties and their details for a specific portfolio
+        self.session.begin()
+        properties_query = self.session.query(
+            PropertyModel,
+            PropertyDetailsEpcModel
+        ).join(
+            PropertyDetailsEpcModel,
+            PropertyModel.id == PropertyDetailsEpcModel.property_id
+        ).filter(
+            PropertyModel.portfolio_id == self.portfolio_id  # Filter by portfolio ID
+        ).all()
+
+        # Transform properties data to include all fields dynamically
+        properties_data = [
+            {**{col.name: getattr(prop.PropertyModel, col.name) for col in PropertyModel.__table__.columns},
+             **{col.name: getattr(prop.PropertyDetailsEpcModel, col.name) for col in
+                PropertyDetailsEpcModel.__table__.columns}}
+            for prop in properties_query
+        ]
+
+        self.session.close()
+        return properties_data
+
+    def get_plans_from_db(self):
+
+        self.session.begin()
+
+        plans_query = self.session.query(Plan).all()
+        # Transform plans data to include all fields dynamically
+        plans_data = [
+            {col.name: getattr(plan, col.name) for col in Plan.__table__.columns}
+            for plan in plans_query
+        ]
+
+        self.session.close()
+        return plans_data
+
+    def export_mds(self):
+        """
+        This function will export the data in the MDS format
+        Core data required:
+        - Property address
+        - Property postcode
+        - uprn
+        - recommended measures
+        - pre-EPC
+        - pre-SAP
+        - pre Heat Demand
+        - Property Type
+        - Built form
+        - Wall type
+        - Tenure
+        - Fuel type
+        - Estimated bill
+        - Recommended measures
+        - Post EPC
+        - Post heat demand
+        - Bill savings
+        - Kwh savings
+        """
+
+        properties_data = self.get_properties_from_db()
+
+        plans_data = self.get_plans_from_db()
+
+        plan_ids = [plan['id'] for plan in plans_data]
+
+    def export(self):
+        """
+        This function will export the data in the required format
+        """
+        if self.format == "mds":
+            self.export_mds()