uuids method

ie3-institute · May 2, 2024 · 7016016 · 7016016
1 parent 0f20953
commit 7016016
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/pypsdm/models/result/participant/dict.py b/pypsdm/models/result/participant/dict.py
@@ -25,6 +25,8 @@
 
 
 class EntitiesResultDictMixin:
+    def uuids(self) -> set[str]:
+        return {key.uuid for key in self.keys()}  # type: ignore
 
     @classmethod
     @abstractmethod
@@ -142,7 +144,6 @@ def from_csv_for_entity(
 
 
 class EmsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin):
-
     def __init__(self, data: dict[EntityKey, ComplexPower]):
         for key, value in data.items():
             if not isinstance(key, EntityKey):
@@ -167,7 +168,6 @@ def entity_type(cls) -> EntitiesEnum:
 
 
 class LoadsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin):
-
     def __init__(self, data: dict[EntityKey, ComplexPower]):
         for key, value in data.items():
             if not isinstance(key, EntityKey):
@@ -192,7 +192,6 @@ def entity_type(cls) -> EntitiesEnum:
 
 
 class FixedFeedInsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin):
-
     def __init__(self, data: dict[EntityKey, ComplexPower]):
         for key, value in data.items():
             if not isinstance(key, EntityKey):
@@ -217,7 +216,6 @@ def entity_type(cls) -> EntitiesEnum:
 
 
 class PvsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin):
-
     def __init__(self, data: dict[EntityKey, ComplexPower]):
         for key, value in data.items():
             if not isinstance(key, EntityKey):
@@ -242,7 +240,6 @@ def entity_type(cls) -> EntitiesEnum:
 
 
 class WecsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin):
-
     def __init__(self, data: dict[EntityKey, ComplexPower]):
         for key, value in data.items():
             if not isinstance(key, EntityKey):
@@ -267,7 +264,6 @@ def entity_type(cls) -> EntitiesEnum:
 
 
 class StoragesResult(ComplexPowerWithSocDict[EntityKey], EntitiesResultDictMixin):
-
     def __init__(self, data: dict[EntityKey, ComplexPowerWithSoc]):
         for key, value in data.items():
             if not isinstance(key, EntityKey):