domainiac 7.0.0__py3-none-any.whl → 8.0.1__py3-none-any.whl

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/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/modeling/plant.py

@@ -13,6 +13,7 @@ class Plant:
     price_area: str
     coordinate: Coordinate
     installed_power_MW: float
+    power_system_resource_type: str
 
     @classmethod
     def plants_from_df(cls, df: pd.DataFrame) -> list["Plant"]:
@@ -30,6 +31,7 @@ class Plant:
                     altitude=row["altitude"],
                 ),
                 installed_power_MW=row["installed_power_MW"],
+                power_system_resource_type=row["power_system_resource_type"],
             )
             plants.append(plant)
         return plants

{domainiac-7.0.0.dist-info → domainiac-8.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: domainiac
-Version: 7.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-7.0.0.dist-info → domainiac-8.0.1.dist-info}/RECORD

@@ -1,16 +1,17 @@
 domainiac/__init__.py,sha256=Wuix4jscZD4sy7YB0RAedAo_eiRW5AcPlef58lKvg6s,207
-domainiac/managers/__init__.py,sha256=x7NZxFAuJig2TZGJ71niwMyPyDvxjaiE647BJQU0qD8,256
+domainiac/managers/__init__.py,sha256=7N-iEsFidupPi1u3EURfeWAoQ-NdrzAwR13faA_VWUs,298
 domainiac/managers/masterdata_manager.py,sha256=H-Dk-dvZZ3bqgV66Xf1V4GQnQWprzLqSKdfD1MPLQgo,2223
 domainiac/managers/metering_manager.py,sha256=sOPss_vpe3GKCRm0U-Obolq9GJPoysimW-a7iXhEhtY,984
 domainiac/managers/nwp_manager.py,sha256=S6mLMJ_J2-0hFB_xZmcGMW25Oo_eVqTVooh-GnN41Ro,10027
+domainiac/managers/outage_manager.py,sha256=SiUBrW4fAspcR71SQsykQ8_Eg_vsHIy4N4A_xcxSZxw,1161
 domainiac/managers/plant_manager.py,sha256=aqJ04QB-QI83zIllQiyTrlh_SKR6CYI4bY4z4RlBL3g,5083
 domainiac/managers/resource_manager.py,sha256=QMkNcrgyXLLlkCR9kg4-nFjTaYHtlsldex-b_8lAmO0,3730
 domainiac/managers/unit_manager.py,sha256=DYF-AiBHzUGZJE5XlTLcszvgIQkSzZs_s9o35l4bkVk,1219
 domainiac/modeling/__init__.py,sha256=Ac5feYZ9gVLayU1Cb1ACqOyDrmvcxoOvZuUkOeXVKf8,101
 domainiac/modeling/nwp.py,sha256=PolrBdQn8W-e1M0_pefvmLn2mr4HT0NqlYlkyCV0dds,438
-domainiac/modeling/plant.py,sha256=SvJmpx-a_WXStQhlRPWbcQNovSaCGU5N2oV8eBdvneE,1728
+domainiac/modeling/plant.py,sha256=FMdoBZpjvg68xmWlDdbTNVtnAjH7FH1OAz4vQgB4d5Y,1842
 domainiac/wrappers/__init__.py,sha256=vZOw9maXgVoAvudZqioD-GTPkgPI6fm7_CUELQcR_-g,43
 domainiac/wrappers/cache_wrapper.py,sha256=jDg-Lt_y-YzItyP-74tGULOxb07s_CcutmOHP1Ie00Q,355
-domainiac-7.0.0.dist-info/METADATA,sha256=38NsSJNtDHhairTH-Uebk-CugBJjmhLZ0mPID8fmTDQ,510
-domainiac-7.0.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-domainiac-7.0.0.dist-info/RECORD,,
+domainiac-8.0.1.dist-info/METADATA,sha256=meEPdMeUIocm1bj8Hmj0jJlewECQ5Ee3N-Svg8oVtdg,510
+domainiac-8.0.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+domainiac-8.0.1.dist-info/RECORD,,