domainiac 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

domainiac/managers/resource_manager.py

@@ -6,57 +6,71 @@ from typeguard import typechecked
 class ResourceManager:
     """
     Manager which simplifies the process of handling resource schedules.
+    Scheduled resources are delivered in a 5 min resolution. If a higher
+    resolution is chosen,
+    the result will be the most conservative value in the time interval,
+    e.i. max of minimum capacity and min of maximum capacity.
+    If no resolution is given, the default is 5 min.
+    If resolution is less than 5 min, the resulting time series will
+    default to 5 min resolution.
     """
 
     def __init__(
         self,
         db: pdz.Database,
         time_interval: pdz.TimeInterval,
+        resolution: pd.Timedelta = pd.Timedelta("PT5M"),
     ) -> None:
         self.db = db
         self.time_interval = time_interval
+        self.resolution = resolution
 
     @typechecked
-    def get_resource_schedules(
-        self,
-        resource_gsrns: list,
-    ) -> pd.DataFrame:
+    def get_resource_schedules(self, resource_gsrn: str | list[str]) -> pd.DataFrame:
         """Gets resource schedules for a given list of resource gsrns."""
         df_resource_schedules = self.db.query(
-            table_name="resourceSchedule",
+            table_name="scheduleResourcePowerPlan",
             time_interval=self.time_interval,
-            filters={"resource_gsrn": resource_gsrns},
+            filters={"resource_gsrn": resource_gsrn},
         )
-        df_resource_schedules = df_resource_schedules.filter(
-            [
-                "market_participant",
-                "created_time_utc",
-                "price_area",
-                "resource_gsrn",
-                "main_fuel_type",
-                "is_sum_plan",
-                "time_utc",
-                "schedule_power_MW",
-                "schedule_capacity_min_MW",
-                "schedule_capacity_max_MW",
-            ]
+
+        if df_resource_schedules.empty:
+            raise ValueError(f"No resource schedules found for gsrn {resource_gsrn}.")
+
+        df_resource_resampled = (
+            pdz.group(
+                df_resource_schedules,
+                by=[
+                    "market_participant",
+                    "created_time_utc",
+                    "price_area",
+                    "resource_gsrn",
+                ],
+            )
+            .resample(on="time_utc", resolution=self.resolution)
+            .agg(
+                {
+                    "schedule_power_MW": "mean",
+                    "schedule_capacity_min_MW": "max",
+                    "schedule_capacity_max_MW": "min",
+                }
+            )
+            .dropna()
         )
-        return df_resource_schedules
+        return df_resource_resampled
 
     @typechecked
-    def get_latest_resource_plan(
+    def get_latest_resource_schedules(
         self,
-        resource_gsrns: list,
+        resource_gsrn: str | list[str],
     ) -> pd.DataFrame:
         """Gets the lastest resource schedules for a given list of resource gsrns."""
 
-        df_resource_schedules = self.get_resource_schedules(
-            resource_gsrns=resource_gsrns
-        )
+        df_resource_schedules = self.get_resource_schedules(resource_gsrn=resource_gsrn)
 
         df_latest_created_time = pdz.group(
             df=df_resource_schedules, by=["resource_gsrn", "time_utc"]
-        ).agg({"created_time_utc": max})
+        ).agg({"created_time_utc": "max"})
 
         df_resource_latest = df_latest_created_time.merge(
             df_resource_schedules, on=list(df_latest_created_time.columns)

{domainiac-0.1.3.dist-info → domainiac-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: domainiac
-Version: 0.1.3
+Version: 0.1.5
 Summary: Package for working with Energinet data, but with specialized functions used for Enigma.
 Author: Team Enigma
 Author-email: gridop-enigma@energinet.dk

{domainiac-0.1.3.dist-info → domainiac-0.1.5.dist-info}/RECORD

@@ -3,8 +3,8 @@ domainiac/managers/__init__.py,sha256=QzkZNv8_3G0TPcfQia45RWBnASYR6WIuY6jlFBeley
 domainiac/managers/masterdata_manager.py,sha256=5YI5K6fE8EFL5TE6OCn1FKIConlIrhKy3aj9yyKze90,1406
 domainiac/managers/metering_manager.py,sha256=NXGHuNjWiAUE08fGIPAfvwbzlKESbKaO5uyxLNyZKC0,995
 domainiac/managers/plant_manager.py,sha256=nDbc0_XFUITylbSCjfhjeVZtffjjSgFJb8BXGr4TJgo,1855
-domainiac/managers/resource_manager.py,sha256=LJvAI6y1RcUgNrMS5iGMqV1Y4X2Pe_ocT_qgffh7HPU,1919
+domainiac/managers/resource_manager.py,sha256=tu6ZLI3wm_9F-ReE7-2MBIDReZILQyLuKsITqAnJgyU,2691
 domainiac/managers/unit_manager.py,sha256=PjtLag75PRDlX57lxbYo6hundkQy37jC170R4YKJw7U,1085
-domainiac-0.1.3.dist-info/METADATA,sha256=hI1_mD-iY5_XPchJ4HP07aNV0zx1YQAAG5UUSGo5rks,465
-domainiac-0.1.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-domainiac-0.1.3.dist-info/RECORD,,
+domainiac-0.1.5.dist-info/METADATA,sha256=lCFHZgQNmVXwVGBNQV6uDIuMDt5J99rHgcp_YDNyAOk,465
+domainiac-0.1.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+domainiac-0.1.5.dist-info/RECORD,,