zepben.ewb 1.0.0b7__py3-none-any.whl → 1.0.0b9__py3-none-any.whl

zepben/ewb/services/common/translator/base_cim2proto.py

@@ -5,6 +5,9 @@
 
 __all__ = ["identified_object_to_pb", "document_to_pb", "organisation_role_to_pb", "organisation_to_pb"]
 
+import inspect
+from typing import ParamSpec, TypeVar, Callable
+
 # noinspection PyPackageRequirements,PyUnresolvedReferences
 from google.protobuf.timestamp_pb2 import Timestamp as PBTimestamp
 from zepben.protobuf.cim.iec61968.common.Document_pb2 import Document as PBDocument
@@ -23,10 +26,23 @@ from zepben.ewb.model.cim.iec61970.base.core.name_type import NameType
 from zepben.ewb.services.common.translator.util import mrid_or_empty
 
 
+P = ParamSpec("P")
+R = TypeVar("R")
+
+
+def bind_to_pb(func: Callable[P, R]) -> Callable[P, R]:
+    """
+    Get the object described in the type hint of the first argument of the function we are wrapping
+    set that object's `to_pb` function to be the function we are wrapping
+    """
+    inspect.get_annotations(func, eval_str=True)[func.__code__.co_varnames[0]].to_pb = func
+    return func
+
 ###################
 # IEC61968 Common #
 ###################
 
+@bind_to_pb
 def document_to_pb(cim: Document) -> PBDocument:
     timestamp = None
     if cim.created_date_time:
@@ -44,10 +60,12 @@ def document_to_pb(cim: Document) -> PBDocument:
     )
 
 
+@bind_to_pb
 def organisation_to_pb(cim: Organisation) -> PBOrganisation:
     return PBOrganisation(io=identified_object_to_pb(cim))
 
 
+@bind_to_pb
 def organisation_role_to_pb(cim: OrganisationRole) -> PBOrganisationRole:
     return PBOrganisationRole(
         io=identified_object_to_pb(cim),
@@ -55,15 +73,11 @@ def organisation_role_to_pb(cim: OrganisationRole) -> PBOrganisationRole:
     )
 
 
-Document.to_pb = document_to_pb
-Organisation.to_pb = organisation_to_pb
-OrganisationRole.to_pb = organisation_role_to_pb
-
-
 ######################
 # IEC61970 Base Core #
 ######################
 
+@bind_to_pb
 def identified_object_to_pb(cim: IdentifiedObject) -> PBIdentifiedObject:
     return PBIdentifiedObject(
         mRID=str(cim.mrid),
@@ -73,6 +87,7 @@ def identified_object_to_pb(cim: IdentifiedObject) -> PBIdentifiedObject:
     )
 
 
+@bind_to_pb
 def name_to_pb(cim: Name) -> PBName:
     return PBName(
         name=cim.name,
@@ -80,13 +95,9 @@ def name_to_pb(cim: Name) -> PBName:
     )
 
 
+@bind_to_pb
 def name_type_to_pb(cim: NameType) -> PBNameType:
     return PBNameType(
         name=cim.name,
         description=cim.description
     )
-
-
-IdentifiedObject.to_pb = identified_object_to_pb
-Name.to_pb = name_to_pb
-NameType.to_pb = name_type_to_pb

zepben/ewb/services/common/translator/base_proto2cim.py

@@ -3,11 +3,18 @@
 #  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__ = ["identified_object_to_cim", "document_to_cim", "organisation_to_cim", "organisation_role_to_cim", "BaseProtoToCim"]
+from __future__ import annotations
 
+__all__ = ["identified_object_to_cim", "document_to_cim", "organisation_to_cim", "organisation_role_to_cim",
+           "BaseProtoToCim", "add_to_network_or_none", "bind_to_cim"]
+
+import functools
+import inspect
 from abc import ABCMeta
-from typing import Optional
+from typing import Optional, Callable, TypeVar
 
+from google.protobuf.message import Message
+from typing_extensions import ParamSpec
 # noinspection PyPackageRequirements
 from zepben.protobuf.cim.iec61968.common.Document_pb2 import Document as PBDocument
 from zepben.protobuf.cim.iec61968.common.OrganisationRole_pb2 import OrganisationRole as PBOrganisationRole
@@ -23,10 +30,36 @@ from zepben.ewb.services.common import resolver
 from zepben.ewb.services.common.base_service import BaseService
 
 
+TProtoToCimFunc = Callable[[Message, BaseService], Optional[IdentifiedObject]]
+P = ParamSpec("P")
+R = TypeVar("R")
+
+
+def add_to_network_or_none(func: TProtoToCimFunc) -> TProtoToCimFunc:
+    """
+    This should wrap any leaf class of the hierarchy, for example, If you're porting over ewb-sdk-jvm
+    changes, any of the classes that get used in a `network.add(Class)`
+    """
+    @functools.wraps(func)
+    def wrapper(pb: Message, service: BaseService) -> Optional[IdentifiedObject]:
+        return cim if service.add(cim := func(pb, service)) else None
+    return wrapper
+
+
+def bind_to_cim(func: Callable[P, R]) -> Callable[P, R]:
+    """
+    Get the object described in the type hint of the first argument of the function we are wrapping
+    set that object's `to_cim` function to be the function we are wrapping
+    """
+    inspect.get_annotations(func, eval_str=True)[func.__code__.co_varnames[0]].to_cim = func
+    return func
+
+
 ###################
 # IEC61968 Common #
 ###################
 
+@bind_to_cim
 def document_to_cim(pb: PBDocument, cim: Document, service: BaseService):
     cim.title = pb.title
     cim.created_date_time = pb.createdDateTime.ToDatetime() if pb.HasField("createdDateTime") else None
@@ -38,28 +71,27 @@ def document_to_cim(pb: PBDocument, cim: Document, service: BaseService):
     identified_object_to_cim(pb.io, cim, service)
 
 
+@bind_to_cim
+@add_to_network_or_none
 def organisation_to_cim(pb: PBOrganisation, service: BaseService) -> Optional[Organisation]:
     cim = Organisation()
 
     identified_object_to_cim(pb.io, cim, service)
-    return cim if service.add(cim) else None
+    return cim
 
 
+@bind_to_cim
 def organisation_role_to_cim(pb: PBOrganisationRole, cim: OrganisationRole, service: BaseService):
     service.resolve_or_defer_reference(resolver.organisation(cim), pb.organisationMRID)
 
     identified_object_to_cim(pb.io, cim, service)
 
 
-PBDocument.to_cim = document_to_cim
-PBOrganisation.to_cim = organisation_to_cim
-PBOrganisationRole.to_cim = organisation_role_to_cim
-
-
 ######################
 # IEC61970 Base Core #
 ######################
 
+@bind_to_cim
 def identified_object_to_cim(pb: PBIdentifiedObject, cim: IdentifiedObject, service: BaseService):
     cim.mrid = pb.mRID
     cim.name = pb.name
@@ -67,6 +99,7 @@ def identified_object_to_cim(pb: PBIdentifiedObject, cim: IdentifiedObject, serv
     [cim.add_name(name_to_cim(name, cim, service).type, name.name) for name in pb.names]
 
 
+@bind_to_cim
 def name_to_cim(pb: PBName, io: IdentifiedObject, service: BaseService):
     try:
         nt = service.get_name_type(pb.type)
@@ -78,6 +111,7 @@ def name_to_cim(pb: PBName, io: IdentifiedObject, service: BaseService):
     return nt.get_or_add_name(pb.name, io)
 
 
+@bind_to_cim
 def name_type_to_cim(pb: PBNameType, service: BaseService):
     try:
         nt = service.get_name_type(pb.name)
@@ -90,11 +124,6 @@ def name_type_to_cim(pb: PBNameType, service: BaseService):
     return nt
 
 
-PBIdentifiedObject.to_cim = identified_object_to_cim
-PBName.to_cim = name_to_cim
-PBNameType.to_cim = name_type_to_cim
-
-
 @dataclass(slots=True)
 class BaseProtoToCim(object, metaclass=ABCMeta):
     service: BaseService

zepben/ewb/services/common/translator/util.py

@@ -49,7 +49,7 @@ def long_or_none(value: int) -> Optional[int]:
 
 
 def str_or_none(value: str) -> Optional[str]:
-    return value if value else None
+    return value or None
 
 
 def from_nullable_int(value: Optional[int]) -> int:
@@ -69,10 +69,7 @@ def from_nullable_long(value: Optional[int]) -> int:
 
 
 def nullable_bool_settings(flag_name: str, value: Optional[bool]) -> Dict:
-    settings = {}
     if value is None:
-        settings[f"{flag_name}Null"] = NullValue.NULL_VALUE
+        return {f'{flag_name}Null': NullValue.NULL_VALUE}
     else:
-        settings[f"{flag_name}Set"] = value
-
-    return settings
+        return {f'{flag_name}Set': value}

zepben/ewb/services/network/tracing/networktrace/network_trace.py

@@ -413,9 +413,7 @@ class NetworkTrace(Traversal[NetworkTraceStep[T], 'NetworkTrace[T]'], Generic[T]
         return self._tracker.visit(terminal, phases)
 
 
-def default_condition_step_type(step_type: CanActionItem):
-    if step_type is None:
-        return False
+def default_condition_step_type(step_type: CanActionItem) -> NetworkTraceStep.Type:
     if step_type == NetworkTraceActionType.ALL_STEPS:
         return NetworkTraceStep.Type.ALL
     elif step_type == NetworkTraceActionType.FIRST_STEP_ON_EQUIPMENT:
@@ -432,19 +430,4 @@ def compute_data_with_action_type(compute_data: ComputeData[T], action_type: Can
                 current_step.data if next_path.traced_internally else compute_data.compute_next(current_step, current_context, next_path)
             )
         )
-    raise Exception(f'{action_type.__class__}: step doesnt match expected types')
-
-
-def with_paths_with_action_type(self, action_type: NetworkTraceActionType) -> ComputeDataWithPaths[T]:
-    if action_type == NetworkTraceActionType.ALL_STEPS:
-        return self
-    elif action_type == NetworkTraceActionType.FIRST_STEP_ON_EQUIPMENT:
-        return ComputeDataWithPaths(
-            lambda current_step, current_context, next_path, next_paths: (
-                current_step.data if next_path.traced_internally else self.compute_next(current_step, current_context, next_path, next_paths)
-            )
-        )
-    raise Exception('step doesnt match expected types')
-
-
-ComputeDataWithPaths[T].with_action_type = with_paths_with_action_type
+    raise Exception(f'{action_type.__name__}: step doesnt match expected types')

zepben/ewb/services/network/tracing/networktrace/network_trace_action_type.py

@@ -2,14 +2,14 @@
 #  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 typing import Callable, Set, Any
+from typing import Callable, FrozenSet, Any
 from enum import Enum
 
 from zepben.ewb import Terminal, SinglePhaseKind
 from zepben.ewb.services.network.tracing.networktrace.network_trace_step import NetworkTraceStep
 from zepben.ewb.services.network.tracing.traversal.step_context import StepContext
 
-HasTracked = Callable[[Terminal, Set[SinglePhaseKind]], bool]
+HasTracked = Callable[[Terminal, FrozenSet[SinglePhaseKind]], bool]
 CanActionItem = Callable[[NetworkTraceStep[Any], StepContext, HasTracked], bool]
 
 

zepben/ewb/services/network/tracing/networktrace/network_trace_step.py

@@ -95,7 +95,10 @@ class NetworkTraceStep(Generic[T]):
         def next_num_equipment_steps(self, current_num: int) -> int:
             return current_num + 1 if self.traced_externally else current_num
 
-    Type = Enum('Type', ('ALL', 'INTERNAL', 'EXTERNAL'))
+    class Type(Enum):
+        ALL = 'ALL'
+        INTERNAL = 'INTERNAL'
+        EXTERNAL = 'EXTERNAL'
 
     def __init__(self, path: Path, num_terminal_steps: int, num_equipment_steps: int, data: T):
         self.path = path