zepben.ewb 1.0.0b1__py3-none-any.whl

zepben/ewb/services/diagram/diagrams.py

@@ -0,0 +1,107 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["DiagramService"]
+
+from typing import Dict, List, Optional
+
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.diagram_object import DiagramObject
+from zepben.ewb.services.common.base_service import BaseService
+from zepben.ewb.services.common.meta.metadata_collection import MetadataCollection
+
+
+class DiagramService(BaseService):
+
+    def __init__(
+        self,
+        metadata: Optional[MetadataCollection] = None
+    ):
+        super().__init__("diagram", metadata)
+
+        self._diagram_objects_by_diagram_mrid: Dict[str, Dict[str, DiagramObject]] = dict()
+        self._diagram_objects_by_identified_object_mrid: Dict[str, Dict[str, DiagramObject]] = dict()
+        self._diagram_object_indexes: List[Dict[str, Dict[str, DiagramObject]]] = list()
+
+    def get_diagram_objects(self, mrid: str) -> List[DiagramObject]:
+        """
+        Get `DiagramObject`'s from the service associated with the given mRID.
+
+        `DiagramObject`'s are indexed by its `DiagramObject.mrid`, its `DiagramObject.diagram.mrid`,
+        and its `DiagramObject.identifiedObjectMRID`'s (if present).
+
+        If you request a `DiagramObject` by its mRID you will receive a List with a single entry, otherwise
+        the list will contain as many `DiagramObject`'s as have been recorded against the provided mRID.
+        `mrid` The mRID to look up in the service.
+        Returns A list of `DiagramObject`'s associated with `mrid`.
+        """
+        obj = self.get(mrid, DiagramObject, None)
+        if obj is not None:
+            return [obj]
+
+        for index in self._diagram_object_indexes:
+            if mrid in index:
+                return list(index[mrid].values())
+
+        return []
+
+    def add_diagram_object(self, diagram_object: DiagramObject):
+        """
+        Associate a `DiagramObject` with this service.
+        The `DiagramObject` will be indexed by its `Diagram` and its `IdentifiedObject` (if present).
+
+        `diagram_object` The `DiagramObject` to add.
+        Returns True if the `DiagramObject` was successfully associated with the service.
+        """
+        return self.add(diagram_object) and self._add_index(diagram_object)
+
+    def remove(self, diagram_object: DiagramObject) -> bool:
+        """
+        Disassociate a `DiagramObject` with the service. This will remove all indexing of the `DiagramObject` and it
+        will no longer be able to be found via the service.
+
+        `diagram_object` The `DiagramObject` to disassociate with this service.
+        Returns True if the `DiagramObject` was removed successfully.
+        """
+        return super().remove(diagram_object) and self._remove_index(diagram_object)
+
+    def _add_index(self, diagram_object: DiagramObject) -> bool:
+        """
+        Index a `DiagramObject` against its associated [Diagram] and [IdentifiedObject].
+
+        `diagram_object` The `DiagramObject` to remove from the indexes.
+        Returns True if the index was updated.
+        """
+        if diagram_object.diagram:
+            self._diagram_objects_by_diagram_mrid.setdefault(diagram_object.diagram.mrid, dict())[diagram_object.mrid] = diagram_object
+
+        io_mrid = diagram_object.identified_object_mrid
+        if io_mrid:
+            self._diagram_objects_by_identified_object_mrid.setdefault(io_mrid, dict())[diagram_object.mrid] = diagram_object
+
+        return True
+
+    def _remove_index(self, diagram_object: DiagramObject) -> bool:
+        """
+        Remove the indexes of a `DiagramObject`.
+
+        `diagram_object` The `DiagramObject` to remove from the indexes.
+        Returns True if the index was updated.
+        """
+        if diagram_object.diagram.mrid:
+            diagram_map = self._diagram_objects_by_diagram_mrid[diagram_object.diagram.mrid]
+            if diagram_map is not None:
+                del diagram_map[diagram_object.mrid]
+                if not diagram_map:
+                    del self._diagram_objects_by_diagram_mrid[diagram_object.diagram.mrid]
+
+        io_mrid = diagram_object.identified_object_mrid
+        if io_mrid is not None:
+            io_map = self._diagram_objects_by_identified_object_mrid[io_mrid]
+            if io_map is not None:
+                del io_map[diagram_object.mrid]
+                if not io_map:
+                    del self._diagram_objects_by_identified_object_mrid[io_mrid]
+
+        return True

zepben/ewb/services/diagram/translator/__init__.py

@@ -0,0 +1,11 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+from zepben.protobuf.cim.iec61970.base.diagramlayout.DiagramObject_pb2 import DiagramObject
+from zepben.protobuf.cim.iec61970.base.diagramlayout.Diagram_pb2 import Diagram
+
+
+Diagram.mrid = lambda self: self.io.mRID
+DiagramObject.mrid = lambda self: self.io.mRID

zepben/ewb/services/diagram/translator/diagram_cim2proto.py

@@ -0,0 +1,51 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["diagram_to_pb", "diagram_object_to_pb", "diagram_object_point_to_pb"]
+
+from zepben.protobuf.cim.iec61970.base.diagramlayout.DiagramObjectPoint_pb2 import DiagramObjectPoint as PBDiagramObjectPoint
+from zepben.protobuf.cim.iec61970.base.diagramlayout.DiagramObject_pb2 import DiagramObject as PBDiagramObject
+from zepben.protobuf.cim.iec61970.base.diagramlayout.Diagram_pb2 import Diagram as PBDiagram
+
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.diagram import Diagram
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.diagram_object import DiagramObject
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.diagram_object_point import DiagramObjectPoint
+from zepben.ewb.services.common.translator.base_cim2proto import identified_object_to_pb
+from zepben.ewb.services.common.translator.util import mrid_or_empty
+# noinspection PyProtectedMember
+from zepben.ewb.services.diagram.translator.diagram_enum_mappers import _map_diagram_style, _map_orientation_kind
+
+
+################################
+# IEC61970 Base Diagram Layout #
+################################
+
+def diagram_to_pb(cim: Diagram) -> PBDiagram:
+    return PBDiagram(
+        io=identified_object_to_pb(cim),
+        diagramStyle=_map_diagram_style.to_pb(cim.diagram_style),
+        orientationKind=_map_orientation_kind.to_pb(cim.orientation_kind),
+        diagramObjectMRIDs=[str(io.mrid) for io in cim.diagram_objects]
+    )
+
+
+def diagram_object_to_pb(cim: DiagramObject) -> PBDiagramObject:
+    return PBDiagramObject(
+        io=identified_object_to_pb(cim),
+        diagramMRID=mrid_or_empty(cim.diagram),
+        identifiedObjectMRID=cim.identified_object_mrid,
+        diagramObjectStyle=cim.style,
+        rotation=cim.rotation,
+        diagramObjectPoints=[diagram_object_point_to_pb(io) for io in cim.points]
+    )
+
+
+def diagram_object_point_to_pb(cim: DiagramObjectPoint) -> PBDiagramObjectPoint:
+    return PBDiagramObjectPoint(xPosition=cim.x_position, yPosition=cim.y_position)
+
+
+Diagram.to_pb = diagram_to_pb
+DiagramObject.to_pb = diagram_object_to_pb
+DiagramObjectPoint.to_pb = diagram_object_point_to_pb

zepben/ewb/services/diagram/translator/diagram_enum_mappers.py

@@ -0,0 +1,21 @@
+#  Copyright 2025 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = []
+
+from zepben.protobuf.cim.iec61970.base.diagramlayout.DiagramStyle_pb2 import DiagramStyle as PBDiagramStyle
+from zepben.protobuf.cim.iec61970.base.diagramlayout.OrientationKind_pb2 import OrientationKind as PBOrientationKind
+
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.diagram_style import DiagramStyle
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.orientation_kind import OrientationKind
+# noinspection PyProtectedMember
+from zepben.ewb.services.common.enum_mapper import EnumMapper
+
+#
+# NOTE: These are deliberately excluded from the module export, as they aren't part of the public api.
+#
+
+_map_diagram_style = EnumMapper(DiagramStyle, PBDiagramStyle)
+_map_orientation_kind = EnumMapper(OrientationKind, PBOrientationKind)

zepben/ewb/services/diagram/translator/diagram_proto2cim.py

@@ -0,0 +1,61 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["diagram_object_point_to_cim", "diagram_to_cim", "diagram_object_to_cim"]
+
+from zepben.protobuf.cim.iec61970.base.diagramlayout.DiagramObjectPoint_pb2 import DiagramObjectPoint as PBDiagramObjectPoint
+from zepben.protobuf.cim.iec61970.base.diagramlayout.DiagramObject_pb2 import DiagramObject as PBDiagramObject
+from zepben.protobuf.cim.iec61970.base.diagramlayout.Diagram_pb2 import Diagram as PBDiagram
+
+import zepben.ewb.services.common.resolver as resolver
+from zepben.ewb import identified_object_to_cim, OrientationKind, DiagramStyle
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.diagram import Diagram
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.diagram_object import DiagramObject
+from zepben.ewb.model.cim.iec61970.base.diagramlayout.diagram_object_point import DiagramObjectPoint
+from zepben.ewb.services.diagram.diagrams import DiagramService
+
+
+################################
+# IEC61970 Base Diagram Layout #
+################################
+
+def diagram_to_cim(pb: PBDiagram, service: DiagramService):
+    cim = Diagram(
+        mrid=pb.mrid(),
+        orientation_kind=OrientationKind(pb.orientationKind),
+        diagram_style=DiagramStyle(pb.diagramStyle)
+    )
+
+    for mrid in pb.diagramObjectMRIDs:
+        service.resolve_or_defer_reference(resolver.diagram_objects(cim), mrid)
+
+    identified_object_to_cim(pb.io, cim, service)
+    return cim if service.add(cim) else None
+
+
+def diagram_object_to_cim(pb: PBDiagramObject, service: DiagramService):
+    cim = DiagramObject(
+        mrid=pb.mrid(),
+        identified_object_mrid=pb.identifiedObjectMRID if pb.identifiedObjectMRID else None,
+        style=pb.diagramObjectStyle if pb.diagramObjectStyle else None,
+        rotation=pb.rotation,
+    )
+
+    service.resolve_or_defer_reference(resolver.diagram(cim), pb.diagramMRID)
+    for point in pb.diagramObjectPoints:
+        cim.add_point(diagram_object_point_to_cim(point))
+
+    identified_object_to_cim(pb.io, cim, service)
+    return cim if service.add_diagram_object(cim) else None
+
+
+def diagram_object_point_to_cim(pb: PBDiagramObjectPoint) -> DiagramObjectPoint:
+    # noinspection PyArgumentList
+    return DiagramObjectPoint(pb.xPosition, pb.yPosition)
+
+
+PBDiagram.to_cim = diagram_to_cim
+PBDiagramObject.to_cim = diagram_object_to_cim
+PBDiagramObjectPoint.to_cim = diagram_object_point_to_cim

zepben/ewb/services/measurement/__init__.py

@@ -0,0 +1,4 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.

zepben/ewb/services/measurement/measurements.py

@@ -0,0 +1,35 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+from __future__ import annotations
+
+__all__ = ["MeasurementService"]
+
+from typing import List, Optional, Generator, TYPE_CHECKING
+
+from zepben.ewb.model.cim.iec61970.base.meas.measurement_value import MeasurementValue
+
+if TYPE_CHECKING:
+    from zepben.ewb import IdentifiedObject
+
+
+class MeasurementService:
+
+    def __init__(self):
+        super().__init__()
+        self._measurements: List[MeasurementValue] = []
+
+    def add(self, value: MeasurementValue):
+        self._measurements.append(value)
+
+    def remove(self, value: MeasurementValue):
+        self._measurements.remove(value)
+
+    def len_of(self, t: type = None) -> int:
+        return len([m for m in self._measurements if isinstance(m, t)]) if t is not None else len(self._measurements)
+
+    def objects(self, obj_type: Optional[type] = None, exc_types: Optional[List[type]] = None) -> Generator[IdentifiedObject, None, None]:
+        for m in self._measurements:
+            yield m

zepben/ewb/services/measurement/translator/__init__.py

@@ -0,0 +1,6 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = []

zepben/ewb/services/measurement/translator/measurement_cim2proto.py

@@ -0,0 +1,42 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["analog_value_to_pb", "accumulator_value_to_pb", "discrete_value_to_pb", "measurement_value_to_pb"]
+
+# noinspection PyPackageRequirements,PyUnresolvedReferences
+from google.protobuf.timestamp_pb2 import Timestamp
+from zepben.protobuf.cim.iec61970.base.meas.AccumulatorValue_pb2 import AccumulatorValue as PBAccumulatorValue
+from zepben.protobuf.cim.iec61970.base.meas.AnalogValue_pb2 import AnalogValue as PBAnalogValue
+from zepben.protobuf.cim.iec61970.base.meas.DiscreteValue_pb2 import DiscreteValue as PBDiscreteValue
+from zepben.protobuf.cim.iec61970.base.meas.MeasurementValue_pb2 import MeasurementValue as PBMeasurementValue
+
+from zepben.ewb.model.cim.iec61970.base.meas.accumulator_value import AccumulatorValue
+from zepben.ewb.model.cim.iec61970.base.meas.analog_value import AnalogValue
+from zepben.ewb.model.cim.iec61970.base.meas.discrete_value import DiscreteValue
+from zepben.ewb.model.cim.iec61970.base.meas.measurement_value import MeasurementValue
+
+
+def analog_value_to_pb(cim: AnalogValue) -> PBAnalogValue:
+    return PBAnalogValue(mv=measurement_value_to_pb(cim), analogMRID=cim.analog_mrid, value=cim.value)
+
+
+def accumulator_value_to_pb(cim: AccumulatorValue) -> PBAccumulatorValue:
+    return PBAccumulatorValue(mv=measurement_value_to_pb(cim), accumulatorMRID=cim.accumulator_mrid, value=cim.value)
+
+
+def discrete_value_to_pb(cim: DiscreteValue) -> PBDiscreteValue:
+    return PBDiscreteValue(mv=measurement_value_to_pb(cim), discreteMRID=cim.discrete_mrid, value=cim.value)
+
+
+def measurement_value_to_pb(cim: MeasurementValue) -> PBMeasurementValue:
+    ts = Timestamp()
+    ts.FromDatetime(cim.time_stamp)
+    return PBMeasurementValue(timeStamp=ts)
+
+
+AnalogValue.to_pb = analog_value_to_pb
+AccumulatorValue.to_pb = accumulator_value_to_pb
+DiscreteValue.to_pb = discrete_value_to_pb
+MeasurementValue.to_pb = measurement_value_to_pb

zepben/ewb/services/measurement/translator/measurement_proto2cim.py

@@ -0,0 +1,52 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["measurement_value_to_cim", "analog_value_to_cim", "accumulator_value_to_cim", "discrete_value_to_cim"]
+
+from zepben.protobuf.cim.iec61970.base.meas.AccumulatorValue_pb2 import AccumulatorValue as PBAccumulatorValue
+from zepben.protobuf.cim.iec61970.base.meas.AnalogValue_pb2 import AnalogValue as PBAnalogValue
+from zepben.protobuf.cim.iec61970.base.meas.DiscreteValue_pb2 import DiscreteValue as PBDiscreteValue
+from zepben.protobuf.cim.iec61970.base.meas.MeasurementValue_pb2 import MeasurementValue as PBMeasurementValue
+
+from zepben.ewb.model.cim.iec61970.base.meas.accumulator_value import AccumulatorValue
+from zepben.ewb.model.cim.iec61970.base.meas.analog_value import AnalogValue
+from zepben.ewb.model.cim.iec61970.base.meas.discrete_value import DiscreteValue
+from zepben.ewb.model.cim.iec61970.base.meas.measurement_value import MeasurementValue
+from zepben.ewb.services.measurement.measurements import MeasurementService
+
+
+######################
+# IEC61970 Base Meas #
+######################
+
+def accumulator_value_to_cim(pb: PBAccumulatorValue, service: MeasurementService):
+    # noinspection PyArgumentList
+    cim = AccumulatorValue(accumulator_mrid=pb.accumulatorMRID, value=pb.value)
+    measurement_value_to_cim(pb.mv, cim)
+    service.add(cim)
+
+
+def analog_value_to_cim(pb: PBAnalogValue, service: MeasurementService):
+    # noinspection PyArgumentList
+    cim = AnalogValue(analog_mrid=pb.analogMRID, value=pb.value)
+    measurement_value_to_cim(pb.mv, cim)
+    service.add(cim)
+
+
+def discrete_value_to_cim(pb: PBDiscreteValue, service: MeasurementService):
+    # noinspection PyArgumentList
+    cim = DiscreteValue(discrete_mrid=pb.discreteMRID, value=pb.value)
+    measurement_value_to_cim(pb.mv, cim)
+    service.add(cim)
+
+
+def measurement_value_to_cim(pb: PBMeasurementValue, cim: MeasurementValue):
+    cim.time_stamp = pb.timeStamp.ToDatetime()
+
+
+PBAccumulatorValue.to_cim = accumulator_value_to_cim
+PBAnalogValue.to_cim = analog_value_to_cim
+PBDiscreteValue.to_cim = discrete_value_to_cim
+PBMeasurementValue.to_cim = measurement_value_to_cim

zepben/ewb/services/network/__init__.py

@@ -0,0 +1,4 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.

zepben/ewb/services/network/network_extensions.py

@@ -0,0 +1,119 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["create_ac_line_segment", "create_two_winding_power_transformer", "create_energy_consumer",
+           "create_energy_source", "create_bus", "create_breaker"]
+
+from zepben.ewb.model.cim.iec61970.base.core.conducting_equipment import ConductingEquipment
+from zepben.ewb.model.cim.iec61970.base.core.connectivity_node import ConnectivityNode
+from zepben.ewb.model.cim.iec61970.base.core.phase_code import PhaseCode
+from zepben.ewb.model.cim.iec61970.base.core.terminal import Terminal
+from zepben.ewb.model.cim.iec61970.base.wires.ac_line_segment import AcLineSegment
+from zepben.ewb.model.cim.iec61970.base.wires.breaker import Breaker
+from zepben.ewb.model.cim.iec61970.base.wires.energy_consumer import EnergyConsumer
+from zepben.ewb.model.cim.iec61970.base.wires.energy_source import EnergySource
+from zepben.ewb.model.cim.iec61970.base.wires.junction import Junction
+from zepben.ewb.model.cim.iec61970.base.wires.power_transformer import PowerTransformer
+from zepben.ewb.model.cim.iec61970.base.wires.power_transformer_end import PowerTransformerEnd
+from zepben.ewb.services.network.network_service import NetworkService
+from zepben.ewb.util import CopyableUUID
+
+
+# !! WARNING !! #
+# THIS CODE IS IN ACTIVE DEVELOPMENT, UNSTABLE, AND LIKELY TO HAVE ISSUES. FOR EXPERIMENTATION ONLY.
+
+def create_ac_line_segment(network_service: NetworkService, cn1: ConnectivityNode, cn2: ConnectivityNode,
+                           **kwargs) -> AcLineSegment:
+    acls = AcLineSegment(**kwargs)
+    _create_two_terminal_conducting_equipment(network_service=network_service, ce=acls)
+    _connect_two_terminal_conducting_equipment(network_service=network_service, ce=acls, cn1=cn1, cn2=cn2)
+    return acls
+
+
+def create_two_winding_power_transformer(network_service: NetworkService, cn1: ConnectivityNode, cn2: ConnectivityNode,
+                                         **kwargs) -> PowerTransformer:
+    power_transformer = PowerTransformer(**kwargs)
+    _create_two_terminal_conducting_equipment(network_service=network_service, ce=power_transformer, **kwargs)
+    _connect_two_terminal_conducting_equipment(network_service=network_service, ce=power_transformer, cn1=cn1, cn2=cn2)
+    # TODO: How to associated PowerTransformerEndInfo to a PowerTransformerInfo
+    for i in range(1, 2):
+        end = PowerTransformerEnd(power_transformer=power_transformer)
+        power_transformer.add_end(end)
+        end.terminal = power_transformer.get_terminal_by_sn(i)
+    return power_transformer
+
+
+def create_energy_consumer(net: NetworkService, cn: ConnectivityNode, **kwargs) -> EnergyConsumer:
+    ec = EnergyConsumer(**kwargs)
+    _create_single_terminal_conducting_equipment(network_service=net, ce=ec, **kwargs)
+    _connect_single_terminal_conducting_equipment(network_service=net, ce=ec, cn=cn)
+    return ec
+
+
+def create_energy_source(net: NetworkService, cn: ConnectivityNode, **kwargs) -> EnergySource:
+    es = EnergySource(**kwargs)
+    _create_single_terminal_conducting_equipment(network_service=net, ce=es, cn=cn, **kwargs)
+    _connect_single_terminal_conducting_equipment(network_service=net, ce=es, cn=cn)
+    return es
+
+
+def create_breaker(network_service: NetworkService, cn1: ConnectivityNode, cn2: ConnectivityNode, **kwargs) -> Breaker:
+    ce = Breaker(**kwargs)
+    _create_two_terminal_conducting_equipment(network_service=network_service, ce=ce, **kwargs)
+    _connect_two_terminal_conducting_equipment(network_service=network_service, ce=ce, cn1=cn1, cn2=cn2)
+    return ce
+
+
+def create_bus(network_service: NetworkService, **kwargs) -> Junction:
+    bus = Junction(**kwargs)
+    if 'mrid' not in kwargs:
+        bus.mrid = str(CopyableUUID())
+    network_service.add(bus)
+    _create_terminals(ce=bus, network=network_service)
+    # TODO: Figure out how to add Voltage to Buses - Looks like we need to add topologicalNode to support the
+    #  relationship to BaseVoltage. Meanwhile using Junction.
+    return bus
+
+
+def _create_two_terminal_conducting_equipment(network_service: NetworkService, ce: ConductingEquipment, **kwargs):
+    if 'mrid' not in kwargs:
+        ce.mrid = str(CopyableUUID())
+    network_service.add(ce)
+    _create_terminals(ce=ce, num_terms=2, network=network_service)
+
+
+def _connect_two_terminal_conducting_equipment(network_service: NetworkService, ce: ConductingEquipment,
+                                               cn1: ConnectivityNode, cn2: ConnectivityNode):
+    network_service.connect_by_mrid(ce.get_terminal_by_sn(1), cn1.mrid)
+    network_service.connect_by_mrid(ce.get_terminal_by_sn(2), cn2.mrid)
+
+
+def _create_single_terminal_conducting_equipment(network_service: NetworkService, ce: ConductingEquipment, **kwargs):
+    if 'mrid' not in kwargs:
+        ce.mrid = str(CopyableUUID())
+    network_service.add(ce)
+    _create_terminals(ce=ce, network=network_service)
+
+
+def _connect_single_terminal_conducting_equipment(network_service: NetworkService, ce: ConductingEquipment,
+                                                  cn: ConnectivityNode):
+    network_service.connect_by_mrid(ce.get_terminal_by_sn(1), cn.mrid)
+
+
+def _create_terminals(network: NetworkService, ce: ConductingEquipment, num_terms: int = 1,
+                      phases: PhaseCode = PhaseCode.ABC):
+    for i in range(1, num_terms + 1):
+        terminal: Terminal = Terminal(mrid=f"{ce.mrid}_t{i}", conducting_equipment=ce, phases=phases, sequence_number=i)
+        ce.add_terminal(terminal)
+        network.add(terminal)
+        terminal.conducting_equipment = ce
+
+
+NetworkService.create_bus = create_bus
+NetworkService.create_energy_source = create_energy_source
+NetworkService.create_two_winding_power_transformer = create_two_winding_power_transformer
+NetworkService.create_ac_line_segment = create_ac_line_segment
+NetworkService.create_energy_consumer = create_energy_consumer
+NetworkService.create_breaker = create_breaker