domainiac 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

domainiac/__init__.py

@@ -1 +1 @@
-from .managers import MeteringManager, PlantManager, UnitManager
+from .managers import MeteringManager, PlantManager, ResourceManager, UnitManager

domainiac/managers/__init__.py

@@ -1,4 +1,5 @@
 from .masterdata_manager import MasterdataManager
 from .metering_manager import MeteringManager
 from .plant_manager import PlantManager
+from .resource_manager import ResourceManager
 from .unit_manager import UnitManager

domainiac/managers/masterdata_manager.py

@@ -19,23 +19,30 @@ class MasterdataManager:
         self.resolution = resolution
 
     @typechecked
-    def get_data(
+    def get_operational_entities(
         self,
         table: str,
         filters: dict = {},
-        columns: list = [],
     ) -> pd.DataFrame:
-        """Gets the units for a given unit type.
-        Filters for units valid at the end of time interval.
-        Filters by default for units in operation.
-        """
+        """Gets the operational data for a given table."""
 
         filters["standing_entity_state"] = "InOperation"
         df = self.db.query(table, filters=filters)
 
-        for key in filters.keys():
-            if key not in df.columns:
-                raise KeyError(f"Column {key} not found in {table}")
+        return df
+
+    @typechecked
+    def get_data(
+        self,
+        table: str,
+        filters: dict = {},
+        columns: list = [],
+    ) -> pd.DataFrame:
+        """Gets the data for a given table.
+        Filters for rows valid at the end of time interval.
+        """
+        df = self.get_operational_entities(table, filters)
+
         for column in columns:
             if column not in df.columns:
                 raise KeyError(f"Column {column} not found in {table}")

domainiac/managers/plant_manager.py

@@ -37,3 +37,25 @@ class PlantManager(MasterdataManager):
         Filters by default for plants in operation.
         """
         return self.get_data("masterdataPlant", filters=filters, columns=columns)
+
+    def get_installed_power_timeseries(self, gsrn: str) -> pd.DataFrame:
+        """Gets the installed power timeseries for a plant."""
+
+        df_times = self.time_interval.to_range(self.resolution).to_frame(
+            index=False, name="time_utc"
+        )
+
+        # explode plant to time series
+        df_plant = self.get_operational_entities("masterdataPlant")
+        df_plant = df_plant.query(f"masterdata_gsrn == '{gsrn}'")
+
+        df_plant = pdz.merge(
+            df_times,
+            df_plant,
+            left_time="time_utc",
+            right_period=("valid_from_date_utc", "valid_to_date_utc"),
+        )
+
+        return df_plant.filter(["time_utc", "installed_power_MW"]).reset_index(
+            drop=True
+        )

domainiac/managers/resource_manager.py

@@ -0,0 +1,65 @@
+import datamazing.pandas as pdz
+import pandas as pd
+from typeguard import typechecked
+
+
+class ResourceManager:
+    """
+    Manager which simplifies the process of handling resource schedules.
+    """
+
+    def __init__(
+        self,
+        db: pdz.Database,
+        time_interval: pdz.TimeInterval,
+    ) -> None:
+        self.db = db
+        self.time_interval = time_interval
+
+    @typechecked
+    def get_resource_schedules(
+        self,
+        resource_gsrns: list,
+    ) -> pd.DataFrame:
+        """Gets resource schedules for a given list of resource gsrns."""
+        df_resource_schedules = self.db.query(
+            table_name="resourceSchedule",
+            time_interval=self.time_interval,
+            filters={"resource_gsrn": resource_gsrns},
+        )
+        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",
+            ]
+        )
+        return df_resource_schedules
+
+    @typechecked
+    def get_latest_resource_plan(
+        self,
+        resource_gsrns: list,
+    ) -> 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_latest_created_time = pdz.group(
+            df=df_resource_schedules, by=["resource_gsrn", "time_utc"]
+        ).agg({"created_time_utc": max})
+
+        df_resource_latest = df_latest_created_time.merge(
+            df_resource_schedules, on=list(df_latest_created_time.columns)
+        )
+
+        return df_resource_latest

{domainiac-0.1.0.dist-info → domainiac-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: domainiac
-Version: 0.1.0
+Version: 0.1.2
 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.2.dist-info/RECORD

@@ -0,0 +1,10 @@
+domainiac/__init__.py,sha256=7-KoUot9OZgzMoOLhuHInlvyt9EfL5QjuZagTbAJksU,82
+domainiac/managers/__init__.py,sha256=QzkZNv8_3G0TPcfQia45RWBnASYR6WIuY6jlFBeley0,220
+domainiac/managers/masterdata_manager.py,sha256=QPybE3QTAmGFX0WhyKpg1ybqbxuY4ufZ0D3r7wS4dQs,1463
+domainiac/managers/metering_manager.py,sha256=NXGHuNjWiAUE08fGIPAfvwbzlKESbKaO5uyxLNyZKC0,995
+domainiac/managers/plant_manager.py,sha256=IUPmTgE99Y7tGASOgXRIeCQmqGe4J2V6yBOlkvcphKA,1823
+domainiac/managers/resource_manager.py,sha256=LJvAI6y1RcUgNrMS5iGMqV1Y4X2Pe_ocT_qgffh7HPU,1919
+domainiac/managers/unit_manager.py,sha256=PjtLag75PRDlX57lxbYo6hundkQy37jC170R4YKJw7U,1085
+domainiac-0.1.2.dist-info/METADATA,sha256=jo1JyI-of5vSawtuQUrZnxtr6Q0ozDrb6jlmEDUjQHw,465
+domainiac-0.1.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+domainiac-0.1.2.dist-info/RECORD,,

domainiac-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-domainiac/__init__.py,sha256=ezC_vpooWhMA-M0M2GxZiF0DFc-KoCeVaNhlR4Ysnfg,65
-domainiac/managers/__init__.py,sha256=ZwpdZMaQ7JxVWOlH2wKAxezy_JNni8qaGzfVFtMkIVo,174
-domainiac/managers/masterdata_manager.py,sha256=h_KRtXLRAniPDy94EkWGC5cB93RYXQe3Zt_r_p4mWe0,1388
-domainiac/managers/metering_manager.py,sha256=NXGHuNjWiAUE08fGIPAfvwbzlKESbKaO5uyxLNyZKC0,995
-domainiac/managers/plant_manager.py,sha256=LW9S_HECpR-lljkopY1IcYpU9XgPDqhQxFH-wPAvQIU,1083
-domainiac/managers/unit_manager.py,sha256=PjtLag75PRDlX57lxbYo6hundkQy37jC170R4YKJw7U,1085
-domainiac-0.1.0.dist-info/METADATA,sha256=qmiB4NB9x1CY9zJMnYiNTPbdPXUVa0HWpX2NvJPpMXM,465
-domainiac-0.1.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-domainiac-0.1.0.dist-info/RECORD,,