domainiac 0.1.1__tar.gz → 0.1.3__tar.gz

{domainiac-0.1.1 → domainiac-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: domainiac
-Version: 0.1.1
+Version: 0.1.3
 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/domainiac/__init__.py

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

{domainiac-0.1.1 → domainiac-0.1.3}/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-0.1.1 → domainiac-0.1.3}/domainiac/managers/masterdata_manager.py

@@ -19,24 +19,16 @@ class MasterdataManager:
         self.resolution = resolution
 
     @typechecked
-    def get_operational_entities(
-        self,
-        table: str,
-        filters: dict = {},
-    ) -> pd.DataFrame:
+    def get_operational_entities(self, table: str, filters: dict = {}) -> pd.DataFrame:
         """Gets the operational data for a given table."""
 
         filters["standing_entity_state"] = "InOperation"
         df = self.db.query(table, filters=filters)
-
         return df
 
     @typechecked
     def get_data(
-        self,
-        table: str,
-        filters: dict = {},
-        columns: list = [],
+        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.

{domainiac-0.1.1 → domainiac-0.1.3}/domainiac/managers/plant_manager.py

@@ -28,6 +28,7 @@ class PlantManager(MasterdataManager):
             "datahub_gsrn_e18",
             "installed_power_MW",
             "price_area",
+            "is_tso_connected",
             "valid_from_date_utc",
             "valid_to_date_utc",
         ],

domainiac-0.1.3/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.1 → domainiac-0.1.3}/pyproject.toml

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

domainiac-0.1.1/domainiac/__init__.py

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