refactor: track yaml hierarchy context and info in YamlModel (#1005)

Author: jsolaas

src/libecalc/application/graph_result.py

@@ -17,7 +17,7 @@ def __init__(
         self.emission_results = emission_results
         self.variables_map = variables_map
 
-    def get_subgraph(self, node_id: str) -> "GraphResult":
+    def get_installation(self, node_id: str) -> "GraphResult":
         subgraph = self.graph.get_node(node_id).get_graph()
 
         return GraphResult(

src/libecalc/domain/energy/emitter.py

@@ -1,4 +1,5 @@
 import abc
+from uuid import UUID
 
 from libecalc.core.result.emission import EmissionResult
 from libecalc.domain.energy.component_energy_context import ComponentEnergyContext
@@ -14,6 +15,9 @@ class Emitter(abc.ABC):
     @abc.abstractmethod
     def id(self) -> str: ...
 
+    @abc.abstractmethod
+    def get_id(self) -> UUID: ...
+
     @abc.abstractmethod
     def evaluate_emissions(
         self,

src/libecalc/domain/energy/energy_component.py

@@ -1,4 +1,5 @@
 import abc
+from uuid import UUID
 
 from libecalc.common.component_type import ComponentType
 
@@ -12,6 +13,9 @@ class EnergyComponent(abc.ABC):
     @abc.abstractmethod
     def id(self) -> str: ...
 
+    @abc.abstractmethod
+    def get_id(self) -> UUID: ...
+
     @abc.abstractmethod
     def get_component_process_type(self) -> ComponentType: ...
 

src/libecalc/domain/infrastructure/emitters/venting_emitter.py

@@ -1,4 +1,5 @@
 import abc
+from uuid import UUID
 
 import numpy as np
 
@@ -71,13 +72,15 @@ def __init__(self, oil_volume_rate: OilVolumeRate, emissions: list[VentingVolume
 class VentingEmitter(Emitter, EnergyComponent, abc.ABC):
     def __init__(
         self,
+        id: UUID,
         path_id: PathID,
         expression_evaluator: ExpressionEvaluator,
         component_type: ComponentType,
         user_defined_category: dict[Period, ConsumerUserDefinedCategoryType],
         emitter_type: VentingType,
         regularity: Regularity,
     ):
+        self._uuid = id
         self._path_id = path_id
         self.expression_evaluator = expression_evaluator
         self.component_type = component_type
@@ -86,6 +89,9 @@ def __init__(
         self.regularity = regularity
         self.emission_results: dict[str, EmissionResult] | None = None
 
+    def get_id(self) -> UUID:
+        return self._uuid
+
     @property
     def name(self) -> str:
         return self._path_id.get_name()

src/libecalc/domain/infrastructure/energy_components/asset/asset.py

@@ -1,3 +1,5 @@
+from uuid import UUID
+
 from libecalc.common.component_type import ComponentType
 from libecalc.domain.energy import EnergyComponent
 from libecalc.domain.infrastructure.energy_components.installation.installation import Installation
@@ -8,13 +10,18 @@
 class Asset(EnergyComponent):
     def __init__(
         self,
+        id: UUID,
         path_id: PathID,
         installations: list[Installation],
     ):
+        self._uuid = id
         self._path_id = path_id
         self.installations = installations
         self.component_type = ComponentType.ASSET
 
+    def get_id(self) -> UUID:
+        return self._uuid
+
     @property
     def id(self):
         return self._path_id.get_name()

src/libecalc/domain/infrastructure/energy_components/electricity_consumer/electricity_consumer.py

@@ -1,4 +1,5 @@
 from typing import Literal
+from uuid import UUID
 
 from libecalc.common.component_type import ComponentType
 from libecalc.common.consumption_type import ConsumptionType
@@ -35,6 +36,7 @@ def get_process_system(self, event: ProcessChangedEvent) -> ProcessSystem | None
 
     def __init__(
         self,
+        id: UUID,
         path_id: PathID,
         regularity: Regularity,
         user_defined_category: dict[Period, ConsumerUserDefinedCategoryType],
@@ -43,6 +45,7 @@ def __init__(
         expression_evaluator: ExpressionEvaluator,
         consumes: Literal[ConsumptionType.ELECTRICITY] = ConsumptionType.ELECTRICITY,
     ):
+        self._uuid = id
         self._path_id = path_id
         self.regularity = regularity
         self.user_defined_category = user_defined_category
@@ -52,6 +55,9 @@ def __init__(
         self.component_type = component_type
         self.consumer_results: dict[str, EcalcModelResult] = {}
 
+    def get_id(self) -> UUID:
+        return self._uuid
+
     @property
     def id(self) -> str:
         return self._path_id.get_name()

src/libecalc/domain/infrastructure/energy_components/fuel_consumer/fuel_consumer.py

@@ -1,3 +1,5 @@
+from uuid import UUID
+
 from libecalc.common.component_type import ComponentType
 from libecalc.common.consumption_type import ConsumptionType
 from libecalc.common.temporal_model import TemporalModel
@@ -36,6 +38,7 @@ def get_process_system(self, event: ProcessChangedEvent) -> ProcessSystem | None
 
     def __init__(
         self,
+        id: UUID,
         path_id: PathID,
         regularity: Regularity,
         user_defined_category: dict[Period, ConsumerUserDefinedCategoryType],
@@ -44,6 +47,7 @@ def __init__(
         energy_usage_model: TemporalModel[ConsumerFunction],
         expression_evaluator: ExpressionEvaluator,
     ):
+        self._uuid = id
         assert fuel is not None
         self._path_id = path_id
         self.regularity = regularity
@@ -55,6 +59,9 @@ def __init__(
         self.consumer_results: dict[str, EcalcModelResult] = {}
         self.emission_results: dict[str, EmissionResult] | None = None
 
+    def get_id(self) -> UUID:
+        return self._uuid
+
     @property
     def name(self):
         return self._path_id.get_name()

src/libecalc/domain/infrastructure/energy_components/generator_set/generator_set_component.py

@@ -1,4 +1,5 @@
 from typing import Literal
+from uuid import UUID
 
 import numpy as np
 from numpy.typing import NDArray
@@ -39,6 +40,7 @@ class GeneratorSetEnergyComponent(Emitter, EnergyComponent):
 
     def __init__(
         self,
+        id: UUID,
         path_id: PathID,
         user_defined_category: dict[Period, ConsumerUserDefinedCategoryType],
         generator_set_model: TemporalModel[GeneratorSetModel],
@@ -50,6 +52,7 @@ def __init__(
         max_usage_from_shore: ExpressionType | None = None,
         component_type: Literal[ComponentType.GENERATOR_SET] = ComponentType.GENERATOR_SET,
     ):
+        self._uuid = id
         self._path_id = path_id
         self.user_defined_category = user_defined_category
         self.regularity = regularity
@@ -63,6 +66,9 @@ def __init__(
         self.consumer_results: dict[str, EcalcModelResult] = {}
         self.emission_results: dict[str, EmissionResult] | None = None
 
+    def get_id(self) -> UUID:
+        return self._uuid
+
     @property
     def id(self) -> str:
         return self._path_id.get_name()

src/libecalc/domain/infrastructure/energy_components/installation/installation.py

@@ -1,3 +1,5 @@
+from uuid import UUID
+
 from libecalc.common.component_type import ComponentType
 from libecalc.common.variables import ExpressionEvaluator
 from libecalc.domain.energy import EnergyComponent
@@ -27,6 +29,7 @@ class Installation(EnergyComponent):
 
     def __init__(
         self,
+        id: UUID,
         path_id: PathID,
         regularity: Regularity,
         hydrocarbon_export: HydrocarbonExport,
@@ -35,6 +38,7 @@ def __init__(
         venting_emitters: list[VentingEmitter] | None = None,
         user_defined_category: InstallationUserDefinedCategoryType | None = None,
     ):
+        self._uuid = id
         self._path_id = path_id
         self.hydrocarbon_export = hydrocarbon_export
         self.regularity = regularity
@@ -49,6 +53,9 @@ def __init__(
             venting_emitters = []
         self.venting_emitters = venting_emitters
 
+    def get_id(self) -> UUID:
+        return self._uuid
+
     @property
     def name(self) -> str:
         return self._path_id.get_name()

src/libecalc/presentation/exporter/infrastructure.py

@@ -403,6 +403,6 @@ def get_from_type(self, exportable_type: ExportableType) -> list[Exportable]:
             assert_never(exportable_type)
 
         return [
-            InstallationExportable(self._graph_result.get_subgraph(installation_id))
+            InstallationExportable(self._graph_result.get_installation(installation_id))
             for installation_id in self._graph_result.graph.get_nodes_of_type(component_type)
         ]