diff --git a/backend/pashub_fetcher/pashub_client.py b/backend/pashub_fetcher/pashub_client.py index a3fa07714..2ca34367d 100644 --- a/backend/pashub_fetcher/pashub_client.py +++ b/backend/pashub_fetcher/pashub_client.py @@ -1,6 +1,6 @@ from collections import defaultdict import os -from typing import Dict, List, NamedTuple, Optional +from typing import Any, Dict, List, NamedTuple, Optional from datetime import datetime import requests @@ -133,7 +133,17 @@ class PashubClient: return None def get_rdsap_summary_by_job_id(self, job_id: str) -> RdSapSummary: - raise NotImplementedError + logger.info(f"Getting RdSAP Summary for job ID {job_id}") + url = f"{self.base}/jobs/{job_id}/rdSapSummary" + + r = self.session.get(url) + + data: Dict[str, Any] = r.json() + return RdSapSummary( + pre_sap_rating=data.get("preSapRating"), + pre_current_co2_emissions=data.get("preCurrentCo2Emissions"), + pre_current_energy_consumption=data.get("preCurrentEnergyConsumption"), + ) def _group_into_core_and_other_files( self,