domainiac 4.0.0__py3-none-any.whl → 5.0.0__py3-none-any.whl

domainiac/managers/plant_manager.py

@@ -41,14 +41,14 @@ class PlantManager(MasterdataManager):
             "is_tso_connected",
             "valid_from_date_utc",
             "valid_to_date_utc",
-            "net_component_id",
+            "primary_net_component_id",
         ]
         if not columns:
             columns = default_columns
 
         # TODO: masterdata_plant table doesn't have net_component_id column
         # Find a better way to do this in future.
-        plant_columns = [col for col in columns if col != "net_component_id"]
+        plant_columns = [col for col in columns if col != "primary_net_component_id"]
         df_plant = self.get_data(
             "masterdataPlant", filters=filters, columns=plant_columns
         )
@@ -119,14 +119,14 @@ class PlantManager(MasterdataManager):
 
     def _get_power_system_resource(self) -> pd.DataFrame:
 
-        df_unit = self.unit_manager.get_units(columns=["masterdata_gsrn"])
+        df_unit = self.unit_manager.get_units(
+            columns=["masterdata_gsrn", "capacity_min_MW", "capacity_max_MW"]
+        )
 
         df_psr_mapping = self.db.query("masterdataAggregatedUnit")[
             ["unit_gsrn", "net_component_id"]
         ]
 
-        df_unit = df_unit[["masterdata_gsrn", "plant_id"]]
-
         df = pd.merge(
             df_psr_mapping,
             df_unit,
@@ -134,6 +134,27 @@ class PlantManager(MasterdataManager):
             right_on="masterdata_gsrn",
             how="left",
             validate="1:m",
-        )[["net_component_id", "plant_id"]]
+        )
+
+        # for a small number of plants, the underlying unit can
+        # be associated with different net components. Too avoid
+        # this issue, we choose for each plant, the net component,
+        # for which the underlying units amounts to the largest
+        # capacity
+        df = pdz.group(df, by=["net_component_id", "plant_id"]).agg(
+            {"capacity_min_MW": "sum", "capacity_max_MW": "sum"}
+        )
+
+        df["capacity_range_MW"] = df["capacity_max_MW"] - df["capacity_min_MW"]
+
+        df = df.sort_values(
+            ["plant_id", "capacity_range_MW"], ascending=False
+        ).drop_duplicates(subset=["plant_id"], keep="first")
+
+        df = df.rename(
+            columns={
+                "net_component_id": "primary_net_component_id",
+            }
+        )
 
         return df

{domainiac-4.0.0.dist-info → domainiac-5.0.0.dist-info}/METADATA

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

{domainiac-4.0.0.dist-info → domainiac-5.0.0.dist-info}/RECORD

@@ -3,7 +3,7 @@ domainiac/managers/__init__.py,sha256=x7NZxFAuJig2TZGJ71niwMyPyDvxjaiE647BJQU0qD
 domainiac/managers/masterdata_manager.py,sha256=H-Dk-dvZZ3bqgV66Xf1V4GQnQWprzLqSKdfD1MPLQgo,2223
 domainiac/managers/metering_manager.py,sha256=NXGHuNjWiAUE08fGIPAfvwbzlKESbKaO5uyxLNyZKC0,995
 domainiac/managers/nwp_manager.py,sha256=qGr4c1ZbUnVRmhX_LmjaobBKN6BBXUn5bbYVKa_5lo4,9921
-domainiac/managers/plant_manager.py,sha256=5NJA8nxx-gTgAXXeVJHK9knwEL-_SiZnQa1tIePVP2g,4307
+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
@@ -11,6 +11,6 @@ domainiac/modeling/nwp.py,sha256=PolrBdQn8W-e1M0_pefvmLn2mr4HT0NqlYlkyCV0dds,438
 domainiac/modeling/plant.py,sha256=48KEcfiMk90fOAzOeI5qtf1FcGvpyBNIRmN2zvx9CLc,1699
 domainiac/wrappers/__init__.py,sha256=vZOw9maXgVoAvudZqioD-GTPkgPI6fm7_CUELQcR_-g,43
 domainiac/wrappers/cache_wrapper.py,sha256=jDg-Lt_y-YzItyP-74tGULOxb07s_CcutmOHP1Ie00Q,355
-domainiac-4.0.0.dist-info/METADATA,sha256=ayM8aSuYDSjgE1KM4GrCURc52_gMu60J_HCFSA65p-E,510
-domainiac-4.0.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-domainiac-4.0.0.dist-info/RECORD,,
+domainiac-5.0.0.dist-info/METADATA,sha256=x2KN4qk3ZS1-Mn1uQ9IHLGF7VE_YdoTQkWfq1WK3LiE,510
+domainiac-5.0.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+domainiac-5.0.0.dist-info/RECORD,,