domainiac 9.3.0__tar.gz → 9.3.1__tar.gz

{domainiac-9.3.0 → domainiac-9.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: domainiac
-Version: 9.3.0
+Version: 9.3.1
 Summary: Package for working with Energinet data, but with specialized functions used for Enigma.
 Author: Team Enigma
 Author-email: enigma@energinet.dk
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
-Requires-Dist: datamazing (>=5.1.6)
+Requires-Dist: datamazing (>=6.0.0)
 Requires-Dist: pandas (>=2.2.0)
 Requires-Dist: pvlib (>=0.13.1)
 Requires-Dist: scikit-learn (>=1.3.0)

{domainiac-9.3.0 → domainiac-9.3.1}/domainiac/__init__.py

@@ -1,4 +1,5 @@
 from .managers import (
+    AvailabilityManager,
     MeteringManager,
     NWPManager,
     PlantManager,

{domainiac-9.3.0 → domainiac-9.3.1}/domainiac/managers/__init__.py

@@ -1,3 +1,4 @@
+from .availability_manager import AvailabilityManager
 from .masterdata_manager import MasterdataManager
 from .metering_manager import MeteringManager
 from .nwp_manager import NWPManager

domainiac-9.3.1/domainiac/managers/availability_manager.py

@@ -0,0 +1,58 @@
+import datamazing.pandas as pdz
+import pandas as pd
+
+from .outage_manager import OutageManager
+
+
+class AvailabilityManager:
+    """
+    Manager which simplifies the process of getting availability data.
+    """
+
+    def __init__(
+        self,
+        db: pdz.Database,
+        time_interval: pdz.TimeInterval,
+        resolution: pd.Timedelta,
+    ) -> None:
+        self.db = db
+        self.time_interval = time_interval
+        self.resolution = resolution
+        self.outage_manager = OutageManager(db, time_interval, resolution)
+
+    def get_availability(self) -> pd.DataFrame:
+        df_availability = self.db.query("scheduleAvailability", self.time_interval)
+        df_outage = self.outage_manager.get_plant_outage_time_series()
+
+        # Rename resource_gsrn to plant_gsrn for consistent merge
+        df_availability = df_availability.rename(
+            columns={
+                "resource_gsrn": "plant_gsrn",
+                "schedule_resource_capacity_max_MW": "capacity_max_MW",
+                "schedule_resource_capacity_min_MW": "capacity_min_MW",
+            }
+        )
+
+        required_columns = [
+            "time_utc",
+            "plant_gsrn",
+            "capacity_max_MW",
+            "capacity_min_MW",
+        ]
+
+        # Create an indicator column in outage to mark where outages exist
+        df_outage["has_outage"] = True
+
+        # Merge availability with outage data
+        df_available = df_availability.merge(
+            df_outage, on=["plant_gsrn", "time_utc"], how="outer"
+        )
+
+        # Where there's an outage, set capacity to zero
+        has_outage = df_available["has_outage"].fillna(False)
+        df_available.loc[has_outage, "capacity_max_MW"] = 0
+        df_available.loc[has_outage, "capacity_min_MW"] = 0
+
+        # Return only the required columns
+        df_available = df_available[required_columns]
+        return df_available

{domainiac-9.3.0 → domainiac-9.3.1}/domainiac/managers/outage_manager.py

@@ -22,7 +22,7 @@ class OutageManager:
 
         df_outage = df_outage[~df_outage["is_unapproved"]]
 
-        df_outage = df_outage.dropna(subset="plant_gsrn")
+        df_outage = df_outage.dropna(subset=["plant_gsrn"])
 
         # make placeholder dataframe to explode outage
         # data to all time points for all plants
@@ -34,12 +34,12 @@ class OutageManager:
 
         df_idx = df_times.merge(df_plants, how="cross")
 
-        df_outage = pdz.merge(
+        df_outage = pdz.merge_point_interval(
             df_idx,
             df_outage,
             on=["plant_gsrn"],
-            left_time="time_utc",
-            right_period=("start_time_utc", "end_time_utc"),
+            left_point="time_utc",
+            right_interval=("start_time_utc", "end_time_utc"),
         )
 
         df_outage = df_outage.filter(

{domainiac-9.3.0 → domainiac-9.3.1}/domainiac/managers/plant_manager.py

@@ -72,11 +72,11 @@ class PlantManager(MasterdataManager):
         df_plant = self.get_operational_entities("masterdataPlant")
         df_plant = df_plant.query(f"masterdata_gsrn == '{gsrn}'")
 
-        df_plant = pdz.merge(
+        df_plant = pdz.merge_point_interval(
             df_times,
             df_plant,
-            left_time="time_utc",
-            right_period=("valid_from_date_utc", "valid_to_date_utc"),
+            left_point="time_utc",
+            right_interval=("valid_from_date_utc", "valid_to_date_utc"),
         )
 
         return df_plant.filter(["time_utc", "installed_power_MW"]).reset_index(

{domainiac-9.3.0 → domainiac-9.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "domainiac"
-version = "9.3.0"
+version = "9.3.1"
 description = "Package for working with Energinet data, but with specialized functions used for Enigma."
 authors = [{ name = "Team Enigma", email = "enigma@energinet.dk" }]
 requires-python = ">=3.10"
@@ -11,7 +11,7 @@ requires-poetry = ">=2.2"
 
 [tool.poetry.dependencies]
 pandas = ">=2.2.0"
-datamazing = ">=5.1.6"
+datamazing = ">=6.0.0"
 typeguard = ">=4.2.1"
 scikit-learn = ">=1.3.0"
 scipy = ">=1.15.3"