wiliot-certificate 1.3.0a1__py3-none-any.whl → 1.4.0a1__py3-none-any.whl

brg_certificate/wltPb_pb2.pyi

@@ -0,0 +1,227 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class Location(_message.Message):
+    __slots__ = ("lat", "lng")
+    LAT_FIELD_NUMBER: _ClassVar[int]
+    LNG_FIELD_NUMBER: _ClassVar[int]
+    lat: float
+    lng: float
+    def __init__(self, lat: _Optional[float] = ..., lng: _Optional[float] = ...) -> None: ...
+
+class Value(_message.Message):
+    __slots__ = ("integerValue", "numberValue", "stringValue", "boolValue")
+    INTEGERVALUE_FIELD_NUMBER: _ClassVar[int]
+    NUMBERVALUE_FIELD_NUMBER: _ClassVar[int]
+    STRINGVALUE_FIELD_NUMBER: _ClassVar[int]
+    BOOLVALUE_FIELD_NUMBER: _ClassVar[int]
+    integerValue: int
+    numberValue: float
+    stringValue: str
+    boolValue: bool
+    def __init__(self, integerValue: _Optional[int] = ..., numberValue: _Optional[float] = ..., stringValue: _Optional[str] = ..., boolValue: bool = ...) -> None: ...
+
+class GatewayData(_message.Message):
+    __slots__ = ("gatewayId", "timestamp", "packets", "location")
+    class Packet(_message.Message):
+        __slots__ = ("payload", "timestamp", "sequenceId", "rssi", "aliasBridgeId")
+        PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+        TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+        SEQUENCEID_FIELD_NUMBER: _ClassVar[int]
+        RSSI_FIELD_NUMBER: _ClassVar[int]
+        ALIASBRIDGEID_FIELD_NUMBER: _ClassVar[int]
+        payload: bytes
+        timestamp: int
+        sequenceId: int
+        rssi: int
+        aliasBridgeId: str
+        def __init__(self, payload: _Optional[bytes] = ..., timestamp: _Optional[int] = ..., sequenceId: _Optional[int] = ..., rssi: _Optional[int] = ..., aliasBridgeId: _Optional[str] = ...) -> None: ...
+    GATEWAYID_FIELD_NUMBER: _ClassVar[int]
+    TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+    PACKETS_FIELD_NUMBER: _ClassVar[int]
+    LOCATION_FIELD_NUMBER: _ClassVar[int]
+    gatewayId: str
+    timestamp: int
+    packets: _containers.RepeatedCompositeFieldContainer[GatewayData.Packet]
+    location: Location
+    def __init__(self, gatewayId: _Optional[str] = ..., timestamp: _Optional[int] = ..., packets: _Optional[_Iterable[_Union[GatewayData.Packet, _Mapping]]] = ..., location: _Optional[_Union[Location, _Mapping]] = ...) -> None: ...
+
+class UplinkMessage(_message.Message):
+    __slots__ = ("gatewayStatus", "gatewayInfo", "gatewayLogs", "actionStatus")
+    GATEWAYSTATUS_FIELD_NUMBER: _ClassVar[int]
+    GATEWAYINFO_FIELD_NUMBER: _ClassVar[int]
+    GATEWAYLOGS_FIELD_NUMBER: _ClassVar[int]
+    ACTIONSTATUS_FIELD_NUMBER: _ClassVar[int]
+    gatewayStatus: GatewayStatus
+    gatewayInfo: GatewayInfo
+    gatewayLogs: GatewayLogs
+    actionStatus: ActionStatus
+    def __init__(self, gatewayStatus: _Optional[_Union[GatewayStatus, _Mapping]] = ..., gatewayInfo: _Optional[_Union[GatewayInfo, _Mapping]] = ..., gatewayLogs: _Optional[_Union[GatewayLogs, _Mapping]] = ..., actionStatus: _Optional[_Union[ActionStatus, _Mapping]] = ...) -> None: ...
+
+class GatewayStatus(_message.Message):
+    __slots__ = ("gatewayId", "gatewayType", "downlinkSupported", "bridgeOtaUpgradeSupported", "apiVersion", "version", "bleSwVersion", "interfaceSwVersion", "location", "config", "bleAddress")
+    class ConfigEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: Value
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+    GATEWAYID_FIELD_NUMBER: _ClassVar[int]
+    GATEWAYTYPE_FIELD_NUMBER: _ClassVar[int]
+    DOWNLINKSUPPORTED_FIELD_NUMBER: _ClassVar[int]
+    BRIDGEOTAUPGRADESUPPORTED_FIELD_NUMBER: _ClassVar[int]
+    APIVERSION_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    BLESWVERSION_FIELD_NUMBER: _ClassVar[int]
+    INTERFACESWVERSION_FIELD_NUMBER: _ClassVar[int]
+    LOCATION_FIELD_NUMBER: _ClassVar[int]
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    BLEADDRESS_FIELD_NUMBER: _ClassVar[int]
+    gatewayId: str
+    gatewayType: str
+    downlinkSupported: bool
+    bridgeOtaUpgradeSupported: bool
+    apiVersion: int
+    version: str
+    bleSwVersion: str
+    interfaceSwVersion: str
+    location: Location
+    config: _containers.MessageMap[str, Value]
+    bleAddress: str
+    def __init__(self, gatewayId: _Optional[str] = ..., gatewayType: _Optional[str] = ..., downlinkSupported: bool = ..., bridgeOtaUpgradeSupported: bool = ..., apiVersion: _Optional[int] = ..., version: _Optional[str] = ..., bleSwVersion: _Optional[str] = ..., interfaceSwVersion: _Optional[str] = ..., location: _Optional[_Union[Location, _Mapping]] = ..., config: _Optional[_Mapping[str, Value]] = ..., bleAddress: _Optional[str] = ...) -> None: ...
+
+class GatewayInfo(_message.Message):
+    __slots__ = ("entries",)
+    class EntriesEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: Value
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+    ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    entries: _containers.MessageMap[str, Value]
+    def __init__(self, entries: _Optional[_Mapping[str, Value]] = ...) -> None: ...
+
+class GatewayLogs(_message.Message):
+    __slots__ = ("logs",)
+    LOGS_FIELD_NUMBER: _ClassVar[int]
+    logs: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, logs: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class ActionStatus(_message.Message):
+    __slots__ = ("action", "status")
+    ACTION_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    action: int
+    status: int
+    def __init__(self, action: _Optional[int] = ..., status: _Optional[int] = ...) -> None: ...
+
+class DownlinkMessage(_message.Message):
+    __slots__ = ("txPacket", "gatewayAction", "bridgeUpgrade", "gatewayConfig", "customBroker", "customMessage")
+    TXPACKET_FIELD_NUMBER: _ClassVar[int]
+    GATEWAYACTION_FIELD_NUMBER: _ClassVar[int]
+    BRIDGEUPGRADE_FIELD_NUMBER: _ClassVar[int]
+    GATEWAYCONFIG_FIELD_NUMBER: _ClassVar[int]
+    CUSTOMBROKER_FIELD_NUMBER: _ClassVar[int]
+    CUSTOMMESSAGE_FIELD_NUMBER: _ClassVar[int]
+    txPacket: TxPacket
+    gatewayAction: GatewayAction
+    bridgeUpgrade: BridgeUpgrade
+    gatewayConfig: GatewayConfig
+    customBroker: CustomBroker
+    customMessage: CustomMessage
+    def __init__(self, txPacket: _Optional[_Union[TxPacket, _Mapping]] = ..., gatewayAction: _Optional[_Union[GatewayAction, _Mapping]] = ..., bridgeUpgrade: _Optional[_Union[BridgeUpgrade, _Mapping]] = ..., gatewayConfig: _Optional[_Union[GatewayConfig, _Mapping]] = ..., customBroker: _Optional[_Union[CustomBroker, _Mapping]] = ..., customMessage: _Optional[_Union[CustomMessage, _Mapping]] = ...) -> None: ...
+
+class TxPacket(_message.Message):
+    __slots__ = ("payload", "maxRetries", "maxDurationMs")
+    PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    MAXRETRIES_FIELD_NUMBER: _ClassVar[int]
+    MAXDURATIONMS_FIELD_NUMBER: _ClassVar[int]
+    payload: bytes
+    maxRetries: int
+    maxDurationMs: int
+    def __init__(self, payload: _Optional[bytes] = ..., maxRetries: _Optional[int] = ..., maxDurationMs: _Optional[int] = ...) -> None: ...
+
+class GatewayAction(_message.Message):
+    __slots__ = ("action",)
+    ACTION_FIELD_NUMBER: _ClassVar[int]
+    action: str
+    def __init__(self, action: _Optional[str] = ...) -> None: ...
+
+class BridgeUpgrade(_message.Message):
+    __slots__ = ("rebootPacket", "txMaxDurationMs", "txMaxRetries", "bridgeId", "versionUuid", "upgradeBlSd", "imageDirUrl")
+    REBOOTPACKET_FIELD_NUMBER: _ClassVar[int]
+    TXMAXDURATIONMS_FIELD_NUMBER: _ClassVar[int]
+    TXMAXRETRIES_FIELD_NUMBER: _ClassVar[int]
+    BRIDGEID_FIELD_NUMBER: _ClassVar[int]
+    VERSIONUUID_FIELD_NUMBER: _ClassVar[int]
+    UPGRADEBLSD_FIELD_NUMBER: _ClassVar[int]
+    IMAGEDIRURL_FIELD_NUMBER: _ClassVar[int]
+    rebootPacket: bytes
+    txMaxDurationMs: int
+    txMaxRetries: int
+    bridgeId: str
+    versionUuid: str
+    upgradeBlSd: bool
+    imageDirUrl: str
+    def __init__(self, rebootPacket: _Optional[bytes] = ..., txMaxDurationMs: _Optional[int] = ..., txMaxRetries: _Optional[int] = ..., bridgeId: _Optional[str] = ..., versionUuid: _Optional[str] = ..., upgradeBlSd: bool = ..., imageDirUrl: _Optional[str] = ...) -> None: ...
+
+class GatewayConfig(_message.Message):
+    __slots__ = ("version", "bleSwVersion", "interfaceSwVersion", "location", "config")
+    class ConfigEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: Value
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    BLESWVERSION_FIELD_NUMBER: _ClassVar[int]
+    INTERFACESWVERSION_FIELD_NUMBER: _ClassVar[int]
+    LOCATION_FIELD_NUMBER: _ClassVar[int]
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    version: str
+    bleSwVersion: str
+    interfaceSwVersion: str
+    location: Location
+    config: _containers.MessageMap[str, Value]
+    def __init__(self, version: _Optional[str] = ..., bleSwVersion: _Optional[str] = ..., interfaceSwVersion: _Optional[str] = ..., location: _Optional[_Union[Location, _Mapping]] = ..., config: _Optional[_Mapping[str, Value]] = ...) -> None: ...
+
+class CustomBroker(_message.Message):
+    __slots__ = ("customBroker", "port", "brokerUrl", "username", "password", "updateTopic", "statusTopic", "dataTopic")
+    CUSTOMBROKER_FIELD_NUMBER: _ClassVar[int]
+    PORT_FIELD_NUMBER: _ClassVar[int]
+    BROKERURL_FIELD_NUMBER: _ClassVar[int]
+    USERNAME_FIELD_NUMBER: _ClassVar[int]
+    PASSWORD_FIELD_NUMBER: _ClassVar[int]
+    UPDATETOPIC_FIELD_NUMBER: _ClassVar[int]
+    STATUSTOPIC_FIELD_NUMBER: _ClassVar[int]
+    DATATOPIC_FIELD_NUMBER: _ClassVar[int]
+    customBroker: bool
+    port: int
+    brokerUrl: str
+    username: str
+    password: str
+    updateTopic: str
+    statusTopic: str
+    dataTopic: str
+    def __init__(self, customBroker: bool = ..., port: _Optional[int] = ..., brokerUrl: _Optional[str] = ..., username: _Optional[str] = ..., password: _Optional[str] = ..., updateTopic: _Optional[str] = ..., statusTopic: _Optional[str] = ..., dataTopic: _Optional[str] = ...) -> None: ...
+
+class CustomMessage(_message.Message):
+    __slots__ = ("entries",)
+    class EntriesEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: Value
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+    ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    entries: _containers.MessageMap[str, Value]
+    def __init__(self, entries: _Optional[_Mapping[str, Value]] = ...) -> None: ...

brg_certificate/wlt_types.py

@@ -0,0 +1,114 @@
+import brg_certificate.ag.wlt_types_ag as ag
+
+def eval_pkt(string):
+    try:
+        return eval(f'ag.{string}')
+    except:
+        return None
+
+def eval_one_param(string):
+    string = string.strip("' ")
+    if string in ag.__dict__:
+        return eval(f'ag.{string}')
+    try:
+        return eval(string)
+    except:
+        return string
+
+def eval_param(string):
+    if "," in string:
+        return [eval_one_param(x) for x in string.strip("[]").split(",")]
+    else:
+        return eval_one_param(string)
+
+class WltPkt():
+    supported_pkt_types = ag.WLT_PKT_TYPES
+
+    def __init__(self, raw='', hdr=None, data_hdr=None, generic=None, pkt=None):
+        self.hdr = ag.Hdr() if hdr is None else hdr
+        self.data_hdr = ag.DataHdr() if data_hdr is None else data_hdr
+        self.generic = generic
+        self.pkt = pkt
+        if raw:
+            self.set(raw)
+
+    def __repr__(self) -> str:
+        if isinstance(self.pkt, (ag.UnifiedEchoPktV0, ag.UnifiedEchoPktV1, ag.UnifiedEchoExtPkt)):
+            return f"{self.data_hdr}\n{self.pkt}"
+        return f"{self.hdr}\n{self.pkt}"
+
+    def __eq__(self, other):
+        if isinstance(other, WltPkt):
+            return (
+                (self.hdr == other.hdr or
+                self.data_hdr == other.data_hdr) and
+                self.generic == other.generic and
+                self.pkt == other.pkt
+            )
+        return False
+
+    def dump(self):
+        if self.pkt:
+            if isinstance(self.pkt, (ag.UnifiedEchoPktV0, ag.UnifiedEchoPktV1, ag.UnifiedEchoExtPkt)):
+                return self.data_hdr.dump() + self.pkt.dump()
+            else:
+                return self.hdr.dump() + self.pkt.dump()
+        return self.data_hdr.dump() + self.generic.dump()
+
+    def set(self, string):
+
+        self.hdr.set(string[0:14])
+        self.data_hdr.set(string[0:14])
+        if self.hdr.group_id == ag.GROUP_ID_BRG2GW or self.hdr.group_id == ag.GROUP_ID_GW2BRG:
+            # GROUP_ID_BRG2GW & GROUP_ID_GW2BRG
+            self.generic = eval_pkt(f'GenericV{ag.API_VERSION_LATEST}()')
+            self.generic.set(string[14:62])
+
+            # MEL modules
+            if self.generic.module_type:
+                self.pkt = eval_pkt(f'{ag.MODULES_DICT[self.generic.module_type]}{self.generic.api_version}()')
+            elif self.generic.module_type == ag.MODULE_GLOBAL:
+                # Action pkts
+                if self.generic.msg_type == ag.BRG_MGMT_MSG_TYPE_ACTION:
+                    pkt = eval_pkt(f'ActionGenericV{self.generic.api_version}("{string[14:62]}")')
+                    if self.generic.api_version >= ag.API_VERSION_V8:
+                        pkt = eval_pkt(f'{ag.ACTIONS_DICT[pkt.action_id]}{self.generic.api_version}()')
+                    self.pkt = pkt
+                # OLD global config (GW2BRG & BRG2GW)
+                elif self.hdr.group_id == ag.GROUP_ID_GW2BRG and self.generic.msg_type == ag.BRG_MGMT_MSG_TYPE_CFG_SET:
+                    self.pkt = eval_pkt(f'Gw2BrgCfgV8()') # no api_version field in Gw2BrgCfg pkts - default parse as api version 8 (Latest Gw2BrgCfg)
+                elif self.hdr.group_id == ag.GROUP_ID_BRG2GW:
+                    if self.generic.msg_type == ag.BRG_MGMT_MSG_TYPE_CFG_SET or self.generic.msg_type == ag.BRG_MGMT_MSG_TYPE_CFG_INFO:
+                        self.pkt = eval_pkt(f'Brg2GwCfgV{self.generic.api_version}()')
+                    elif self.generic.msg_type == ag.BRG_MGMT_MSG_TYPE_HB:
+                        self.pkt = eval_pkt(f'Brg2GwHbV{self.generic.api_version}()')
+                    elif self.generic.msg_type == ag.BRG_MGMT_MSG_TYPE_HB_SLEEP:
+                        self.pkt = eval_pkt(f'Brg2GwHbSleepV{self.generic.api_version}()')
+        # Unified pkt
+        elif self.data_hdr.group_id_major in ag.UNIFIED_GROUP_ID_LIST:
+            if self.data_hdr.group_id_major == ag.GROUP_ID_UNIFIED_PKT_V0:
+                self.pkt = ag.UnifiedEchoPktV0()
+            elif self.data_hdr.group_id_major == ag.GROUP_ID_UNIFIED_PKT_V1 or self.data_hdr.group_id_major == ag.GROUP_ID_BLE5_PKT0: 
+                self.pkt = ag.UnifiedEchoPktV1()
+            elif self.data_hdr.group_id_major == ag.GROUP_ID_BLE5_EXTENDED: 
+                self.pkt = ag.UnifiedEchoExtPkt()
+            else:
+                print(f"Unidentified unified group id found: {hex(self.data_hdr.group_id_major)}")
+        # SideInfo pkts
+        elif self.hdr.group_id == ag.GROUP_ID_SIDE_INFO_SENSOR:
+            self.pkt = ag.SideInfoSensor()
+        elif self.hdr.group_id == ag.GROUP_ID_SIDE_INFO:
+            self.pkt = ag.SideInfo()
+        else:
+            print(f"Unidentified sensor group id found: {hex(self.data_hdr.group_id_major)}")
+
+        if self.pkt:
+            if self.data_hdr.group_id_major == ag.GROUP_ID_BLE5_EXTENDED:
+                self.pkt.set(string[14:84])
+            else:
+                self.pkt.set(string[14:62])
+        # Signal Indicator pkt
+        elif self.hdr.group_id == ag.GROUP_ID_SIGNAL_INDICATOR:
+            self.pkt = eval_pkt(f'SignalIndicatorDataV{ag.SIGNAL_INDICATOR_PACKET_VERSION_LATEST}')(string[8:62])
+        elif self.hdr.uuid_lsb == ag.HDR_DEFAULT_BRG_SENSOR_UUID_LSB and self.hdr.uuid_msb == ag.HDR_DEFAULT_BRG_SENSOR_UUID_MSB:
+            self.pkt = ag.SensorData(string)