domainiac 8.0.0__tar.gz → 8.0.1__tar.gz

{domainiac-8.0.0 → domainiac-8.0.1}/PKG-INFO

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

{domainiac-8.0.0 → domainiac-8.0.1}/domainiac/managers/__init__.py

@@ -1,6 +1,7 @@
 from .masterdata_manager import MasterdataManager
 from .metering_manager import MeteringManager
 from .nwp_manager import NWPManager
+from .outage_manager import OutageManager
 from .plant_manager import PlantManager
 from .resource_manager import ResourceManager
 from .unit_manager import UnitManager

domainiac-8.0.1/domainiac/managers/outage_manager.py

@@ -0,0 +1,47 @@
+import datamazing.pandas as pdz
+import pandas as pd
+
+
+class OutageManager:
+    """
+    Manager which simplifies the process of getting outage 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
+
+    def get_plant_outage_time_series(self) -> pd.DataFrame:
+        df_outage = self.db.query("scheduleOutage")
+
+        df_outage = df_outage[~df_outage["is_unapproved"]]
+
+        df_outage = df_outage.dropna(subset="plant_gsrn")
+
+        df_times = self.time_interval.to_range(self.resolution).to_frame(
+            index=False, name="time_utc"
+        )
+
+        df_outage = pdz.merge(
+            df_times,
+            df_outage,
+            left_time="time_utc",
+            right_period=("start_time_utc", "end_time_utc"),
+        )
+
+        df_outage = df_outage.filter(
+            [
+                "plant_gsrn",
+                "time_utc",
+            ]
+        )
+
+        df_outage = df_outage.reset_index(drop=True)
+
+        return df_outage

{domainiac-8.0.0 → domainiac-8.0.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "domainiac"
-version = "8.0.0"
+version = "8.0.1"
 description = "Package for working with Energinet data, but with specialized functions used for Enigma."
 authors = ["Team Enigma <gridop-enigma@energinet.dk>"]
 packages = [