wiliot-certificate 4.4.3__py3-none-any.whl → 4.5.0a1__py3-none-any.whl

gw_certificate/common/analysis_data_bricks.py

@@ -1,60 +0,0 @@
-import logging
-from importlib import reload
-import os
-import inspect
-from appdirs import user_data_dir
-
-from gw_certificate.common.debug import debug_print
-from gw_certificate.common.utils import current_timestamp
-
-
-def db_utils():
-    """
-    Use databricks dbutils from an external package
-    for example: to use dbutils.fs.head() - use db_utils().fs.head()
-    """
-    for frame in inspect.getouterframes(inspect.currentframe()):
-        global_names = set(frame.frame.f_globals)
-        # Use multiple functions to reduce risk of mismatch
-        if all(v in global_names for v in ["dbutils"]):
-            return frame.frame.f_globals["dbutils"]()
-    raise EnvironmentError("Unable to detect dbutils function")
-
-def initialize_logger(working_directory=None):
-    """
-    initializes the logger to print to log and to logfile, which by default is named by the current timestamp (ms)
-    when calling the function.
-    :param working_directory: working directory to save logfile (in case running locally)
-    :type working_directory: str
-    :return: logger fileHandler filename
-    """
-    logging.shutdown()
-    reload(logging)
-    logger = logging.getLogger()
-    for handler in logger.handlers:
-        try:
-            # extract current filename from log
-            filename = handler.baseFilename.split('\\')[-1].split('.')[0]
-            debug_print(f'Logger already initialized! passing logfile {filename}')
-            return filename
-        except Exception:
-            pass
-    logger_filename = int(current_timestamp())
-    if working_directory is None:
-        working_directory = os.path.join(user_data_dir(), 'wiliot', 'deployment_tools')
-    if not os.path.exists(working_directory):
-        os.makedirs(working_directory)
-    logging.basicConfig(level=logging.DEBUG,
-                        format='%(asctime)s | %(levelname)s | %(message)s',
-                        handlers=[
-                            logging.FileHandler(f'{working_directory}/{logger_filename}.log', 'a'),
-                            # logging.handlers.RotatingFileHandler
-                            logging.StreamHandler()
-                        ])
-    # filter stream to show info and up
-    logging.getLogger().handlers[0].setLevel(logging.DEBUG)
-    logging.getLogger().handlers[1].setLevel(logging.INFO)
-    debug_print(f'logger initialized at {logger_filename}', center=True)
-    debug_print(f'logfile located at {working_directory}/{logger_filename}.log')
-    logging.getLogger().setLevel(logging.DEBUG)
-    return logger_filename

gw_certificate/common/debug.py

@@ -1,42 +0,0 @@
-import datetime
-import logging
-import colorama
-import tabulate
-import pprint
-
-
-# Debugging & Printing
-def debug_print(txt, pretty=False, center=False, tab=False, color=None, enable=True):
-    """
-    :type txt: string
-    :param txt: text to print
-    :type pretty: bool
-    :param pretty: pretty print
-    :type center: bool
-    :param center: print with stars (*)
-    :type tab: bool
-    :param tab: tabulate input
-    :type color: str
-    :param color: colorama color code
-    """
-    if enable:
-        if tab:
-            txt = '\n' + tabulate.tabulate(txt, tablefmt='rst')
-        if type(txt) != str:
-            try:
-                txt = str(txt)
-            except TypeError:
-                debug_print('Could not Print!')
-        if center:
-            txt = txt.center(94, '-')
-        if color is not None:
-            txt = eval(f'colorama.Fore.{color}') + txt + colorama.Style.RESET_ALL
-        if pretty:
-            txt = pprint.pformat(txt, sort_dicts=False)
-
-        # Print Text
-        if logging.getLogger().hasHandlers():
-            log = logging.getLogger()
-            log.info(txt)
-        else:
-            print(datetime.datetime.now().strftime("[%d/%m/%Y, %H:%M:%S]: ") + txt)

gw_certificate/common/serialization_formatter.py

@@ -1,93 +0,0 @@
-
-from enum import Enum
-from google.protobuf.json_format import MessageToDict, Parse, ParseError, ParseDict
-import base64
-
-from gw_certificate.interface.mqtt import Serialization
-from gw_certificate.interface.if_defines import LOCATION
-from gw_certificate.common import wltPb_pb2
-from gw_certificate.ag.ut_defines import GW_CONF, ADDITIONAL, VERSION, BLE_VERSION, WIFI_VERSION, LAT, LNG
-from gw_certificate.common.debug import debug_print
-
-# helper defines
-ACL_MODE =                      "mode"
-ACL_BRIDGE_IDS =                "bridgeIds"
-ACL_DENY =                      "deny"
-ACL_ALLOW =                     "allow"
-ACL_DENY_VALUE =                 0
-ACL_ALLOW_VALUE =                1
-
-
-class Configurable(Enum):
-    ACL = "accessControlList"
-
-class SerializationFormatter():
-    def __init__(self, serialization:Serialization):
-        self.serialization = serialization
-    
-    def is_pb(self):
-        return self.serialization == Serialization.PB
-
-    def cfg_param_set(self, received_status_msg: dict, param: Configurable, value):
-        '''
-        Since configuration paramters change between partners, we must use a received_status_msg to keep every parameter as is.
-        In the future we may use a validation schema file or default.
-        '''
-
-        def gatewaystatus_to_downlink(gw_status_dict: dict, param_to_set: Configurable, value_to_set) -> wltPb_pb2.DownlinkMessage:
-            gw_status = ParseDict(gw_status_dict, wltPb_pb2.GatewayStatus())
-            gw_cfg = wltPb_pb2.GatewayConfig()
-
-            if gw_status.HasField(VERSION):
-                gw_cfg.version = gw_status.version
-            if gw_status.HasField('bleSwVersion'):
-                gw_cfg.bleSwVersion = gw_status.bleSwVersion
-            if gw_status.HasField('interfaceSwVersion'):
-                gw_cfg.interfaceSwVersion = gw_status.interfaceSwVersion
-            if gw_status.HasField(LOCATION):
-                gw_cfg.location.lat = gw_status.location.lat
-                gw_cfg.location.lng = gw_status.location.lng
-            
-            for key, val in gw_status.config.items():
-                if key != param_to_set:
-                    gw_cfg.config[key].CopyFrom(val)
-                else:
-                    pb_gcvalue = wltPb_pb2.GatewayConfigValue()
-                    if isinstance(value_to_set, int):
-                        pb_gcvalue.integerValue = value_to_set
-                    elif isinstance(value_to_set, float):
-                        pb_gcvalue.numberValue = value_to_set
-                    elif isinstance(value_to_set, str):
-                        pb_gcvalue.stringValue = value_to_set
-                    elif isinstance(value_to_set, bool):
-                        pb_gcvalue.boolValue = value_to_set
-                    elif isinstance(value_to_set, dict) and key == Configurable.ACL.value:
-                        pb_gcvalue.aclValue.mode_allow = ACL_ALLOW_VALUE if value_to_set.get(ACL_MODE) == ACL_ALLOW else ACL_DENY_VALUE
-                        ids_bytes = [bytes.fromhex(id) for id in value_to_set.get(ACL_BRIDGE_IDS)]
-                        pb_gcvalue.aclValue.ids.extend(ids_bytes)
-                    else:
-                        debug_print(f"Unsupported value type for key '{key}': {type(val)}, protobuf cfg may be invalid")
-
-                    gw_cfg.config[key].CopyFrom(pb_gcvalue)
-
-            return wltPb_pb2.DownlinkMessage(gatewayConfig=gw_cfg)
-            
-        cfg_msg = received_status_msg.copy()
-        if self.serialization == Serialization.PB:
-            # In protobuf, we must convert the GatewayStatus to a DownlinkMessage with oneof=GatewayConfig
-            cfg_msg = gatewaystatus_to_downlink(cfg_msg, param, value)
-            return cfg_msg
-        else:
-            # JSON
-            cfg_msg.setdefault(GW_CONF, {}).setdefault(ADDITIONAL, {})[param] = value
-            debug_print(cfg_msg)
-            return cfg_msg
-    
-    def pb_status_acl_bytes_to_hex_string(self, msg: dict) -> dict:
-        brg_ids_bytes = msg.get('config', {}).get('accessControlList', {}).get('aclValue', {}).get('ids', {})
-        brg_ids_ascii = []
-        for brg_id in brg_ids_bytes:
-            brg_ids_ascii.append(base64.b64decode(brg_id).hex().upper())
-        if len(brg_ids_ascii) > 0:
-            msg['config']['accessControlList']['aclValue']['ids'] = brg_ids_ascii
-        return msg

gw_certificate/common/utils.py

@@ -1,8 +0,0 @@
-
-# External libraries
-import datetime
-
-# Date & Time Related
-def current_timestamp():
-    """returns current timestamp (UTC) in milliseconds"""
-    return datetime.datetime.timestamp(datetime.datetime.now()) * 1000

gw_certificate/common/utils_defines.py

@@ -1,15 +0,0 @@
-# Brownout / OTA Defines
-SEP = "#" * 50
-BO_DICT = {
-    'datapath': {'config': {'pktFilter': 'Disable forwarding'}},
-    'energy2400': {'config': {'energyPattern2400': 'No Energizing'}},
-    'energySub1g': {'config': {'sub1gEnergyPattern': 'No Energizing'}},
-    'calibration': {'config': {'calibPattern': 'Disable calibration beaconing'}}
-}
-BROADCAST_DST_MAC = 'FFFFFFFFFFFF'
-
-colors = ['red', 'blue', 'yellow', 'cyan', 'green', 'brown', 'orange', 'pink', 'purple', 'black']
-
-# Test Tool
-GW_DATA_SRC = 'gwDataSrc'
-GW_DATA_MODE = 'gwDataMode'

gw_certificate/common/wltPb_pb2.py

@@ -1,84 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: wltPb.proto
-# Protobuf Python Version: 6.30.2
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    6,
-    30,
-    2,
-    '',
-    'wltPb.proto'
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0bwltPb.proto\x12\x05wltPb\"$\n\x08Location\x12\x0b\n\x03lat\x18\x01 \x01(\x01\x12\x0b\n\x03lng\x18\x02 \x01(\x01\"j\n\x05Value\x12\x16\n\x0cintegerValue\x18\x01 \x01(\x03H\x00\x12\x15\n\x0bnumberValue\x18\x02 \x01(\x01H\x00\x12\x15\n\x0bstringValue\x18\x03 \x01(\tH\x00\x12\x13\n\tboolValue\x18\x04 \x01(\x08H\x00\x42\x06\n\x04type\"\x89\x02\n\x0bGatewayData\x12\x11\n\tgatewayId\x18\x01 \x01(\t\x12\x11\n\ttimestamp\x18\x02 \x01(\x04\x12*\n\x07packets\x18\x03 \x03(\x0b\x32\x19.wltPb.GatewayData.Packet\x12&\n\x08location\x18\x04 \x01(\x0b\x32\x0f.wltPb.LocationH\x00\x88\x01\x01\x1as\n\x06Packet\x12\x0f\n\x07payload\x18\x01 \x01(\x0c\x12\x11\n\ttimestamp\x18\x02 \x01(\x04\x12\x12\n\nsequenceId\x18\x03 \x01(\r\x12\x11\n\x04rssi\x18\x04 \x01(\rH\x00\x88\x01\x01\x12\x15\n\raliasBridgeId\x18\x05 \x01(\tB\x07\n\x05_rssiB\x0b\n\t_location\"\xc9\x01\n\rUplinkMessage\x12-\n\rgatewayStatus\x18\x01 \x01(\x0b\x32\x14.wltPb.GatewayStatusH\x00\x12)\n\x0bgatewayInfo\x18\x02 \x01(\x0b\x32\x12.wltPb.GatewayInfoH\x00\x12)\n\x0bgatewayLogs\x18\x03 \x01(\x0b\x32\x12.wltPb.GatewayLogsH\x00\x12+\n\x0c\x61\x63tionStatus\x18\x04 \x01(\x0b\x32\x13.wltPb.ActionStatusH\x00\x42\x06\n\x04type\"&\n\x03\x41\x43L\x12\x12\n\nmode_allow\x18\x01 \x01(\x08\x12\x0b\n\x03ids\x18\x02 \x03(\x0c\"\x97\x01\n\x12GatewayConfigValue\x12\x16\n\x0cintegerValue\x18\x01 \x01(\x03H\x00\x12\x15\n\x0bnumberValue\x18\x02 \x01(\x01H\x00\x12\x15\n\x0bstringValue\x18\x03 \x01(\tH\x00\x12\x13\n\tboolValue\x18\x04 \x01(\x08H\x00\x12\x1e\n\x08\x61\x63lValue\x18\x05 \x01(\x0b\x32\n.wltPb.ACLH\x00\x42\x06\n\x04type\"\xe8\x03\n\rGatewayStatus\x12\x11\n\tgatewayId\x18\x01 \x01(\t\x12\x13\n\x0bgatewayType\x18\x02 \x01(\t\x12\x19\n\x11\x64ownlinkSupported\x18\x03 \x01(\x08\x12!\n\x19\x62ridgeOtaUpgradeSupported\x18\x04 \x01(\x08\x12\x12\n\napiVersion\x18\x05 \x01(\r\x12\x14\n\x07version\x18\x06 \x01(\tH\x00\x88\x01\x01\x12\x19\n\x0c\x62leSwVersion\x18\x07 \x01(\tH\x01\x88\x01\x01\x12\x1f\n\x12interfaceSwVersion\x18\x08 \x01(\tH\x02\x88\x01\x01\x12&\n\x08location\x18\t \x01(\x0b\x32\x0f.wltPb.LocationH\x03\x88\x01\x01\x12\x30\n\x06\x63onfig\x18\n \x03(\x0b\x32 .wltPb.GatewayStatus.ConfigEntry\x12\x17\n\nbleAddress\x18\x0b \x01(\tH\x04\x88\x01\x01\x1aH\n\x0b\x43onfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12(\n\x05value\x18\x02 \x01(\x0b\x32\x19.wltPb.GatewayConfigValue:\x02\x38\x01\x42\n\n\x08_versionB\x0f\n\r_bleSwVersionB\x15\n\x13_interfaceSwVersionB\x0b\n\t_locationB\r\n\x0b_bleAddress\"}\n\x0bGatewayInfo\x12\x30\n\x07\x65ntries\x18\x01 \x03(\x0b\x32\x1f.wltPb.GatewayInfo.EntriesEntry\x1a<\n\x0c\x45ntriesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.wltPb.Value:\x02\x38\x01\"\x1b\n\x0bGatewayLogs\x12\x0c\n\x04logs\x18\x01 \x03(\t\"\x92\x01\n\x0c\x41\x63tionStatus\x12\x0e\n\x06\x61\x63tion\x18\x01 \x01(\r\x12\x0e\n\x06status\x18\x02 \x01(\r\x12\x11\n\x04step\x18\x03 \x01(\rH\x00\x88\x01\x01\x12\x15\n\x08progress\x18\x04 \x01(\rH\x01\x88\x01\x01\x12\x15\n\x08\x62ridgeId\x18\x05 \x01(\x0cH\x02\x88\x01\x01\x42\x07\n\x05_stepB\x0b\n\t_progressB\x0b\n\t_bridgeId\"\x8e\x02\n\x0f\x44ownlinkMessage\x12#\n\x08txPacket\x18\x01 \x01(\x0b\x32\x0f.wltPb.TxPacketH\x00\x12-\n\rgatewayAction\x18\x02 \x01(\x0b\x32\x14.wltPb.GatewayActionH\x00\x12-\n\rbridgeUpgrade\x18\x03 \x01(\x0b\x32\x14.wltPb.BridgeUpgradeH\x00\x12-\n\rgatewayConfig\x18\x04 \x01(\x0b\x32\x14.wltPb.GatewayConfigH\x00\x12-\n\rcustomMessage\x18\x06 \x01(\x0b\x32\x14.wltPb.CustomMessageH\x00\x42\x06\n\x04typeJ\x04\x08\x05\x10\x06R\x0c\x63ustomBroker\"F\n\x08TxPacket\x12\x0f\n\x07payload\x18\x01 \x01(\x0c\x12\x12\n\nmaxRetries\x18\x02 \x01(\r\x12\x15\n\rmaxDurationMs\x18\x03 \x01(\r\"\x1f\n\rGatewayAction\x12\x0e\n\x06\x61\x63tion\x18\x01 \x01(\t\"\xa5\x01\n\rBridgeUpgrade\x12\x14\n\x0crebootPacket\x18\x01 \x01(\x0c\x12\x17\n\x0ftxMaxDurationMs\x18\x02 \x01(\r\x12\x14\n\x0ctxMaxRetries\x18\x03 \x01(\r\x12\x10\n\x08\x62ridgeId\x18\x04 \x01(\t\x12\x13\n\x0bversionUuid\x18\x05 \x01(\t\x12\x13\n\x0bupgradeBlSd\x18\x06 \x01(\x08\x12\x13\n\x0bimageDirUrl\x18\x07 \x01(\t\"\xc6\x02\n\rGatewayConfig\x12\x14\n\x07version\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x19\n\x0c\x62leSwVersion\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x1f\n\x12interfaceSwVersion\x18\x03 \x01(\tH\x02\x88\x01\x01\x12&\n\x08location\x18\x04 \x01(\x0b\x32\x0f.wltPb.LocationH\x03\x88\x01\x01\x12\x30\n\x06\x63onfig\x18\x05 \x03(\x0b\x32 .wltPb.GatewayConfig.ConfigEntry\x1aH\n\x0b\x43onfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12(\n\x05value\x18\x02 \x01(\x0b\x32\x19.wltPb.GatewayConfigValue:\x02\x38\x01\x42\n\n\x08_versionB\x0f\n\r_bleSwVersionB\x15\n\x13_interfaceSwVersionB\x0b\n\t_location\"\x81\x01\n\rCustomMessage\x12\x32\n\x07\x65ntries\x18\x01 \x03(\x0b\x32!.wltPb.CustomMessage.EntriesEntry\x1a<\n\x0c\x45ntriesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.wltPb.Value:\x02\x38\x01\x62\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'wltPb_pb2', _globals)
-if not _descriptor._USE_C_DESCRIPTORS:
-  DESCRIPTOR._loaded_options = None
-  _globals['_GATEWAYSTATUS_CONFIGENTRY']._loaded_options = None
-  _globals['_GATEWAYSTATUS_CONFIGENTRY']._serialized_options = b'8\001'
-  _globals['_GATEWAYINFO_ENTRIESENTRY']._loaded_options = None
-  _globals['_GATEWAYINFO_ENTRIESENTRY']._serialized_options = b'8\001'
-  _globals['_GATEWAYCONFIG_CONFIGENTRY']._loaded_options = None
-  _globals['_GATEWAYCONFIG_CONFIGENTRY']._serialized_options = b'8\001'
-  _globals['_CUSTOMMESSAGE_ENTRIESENTRY']._loaded_options = None
-  _globals['_CUSTOMMESSAGE_ENTRIESENTRY']._serialized_options = b'8\001'
-  _globals['_LOCATION']._serialized_start=22
-  _globals['_LOCATION']._serialized_end=58
-  _globals['_VALUE']._serialized_start=60
-  _globals['_VALUE']._serialized_end=166
-  _globals['_GATEWAYDATA']._serialized_start=169
-  _globals['_GATEWAYDATA']._serialized_end=434
-  _globals['_GATEWAYDATA_PACKET']._serialized_start=306
-  _globals['_GATEWAYDATA_PACKET']._serialized_end=421
-  _globals['_UPLINKMESSAGE']._serialized_start=437
-  _globals['_UPLINKMESSAGE']._serialized_end=638
-  _globals['_ACL']._serialized_start=640
-  _globals['_ACL']._serialized_end=678
-  _globals['_GATEWAYCONFIGVALUE']._serialized_start=681
-  _globals['_GATEWAYCONFIGVALUE']._serialized_end=832
-  _globals['_GATEWAYSTATUS']._serialized_start=835
-  _globals['_GATEWAYSTATUS']._serialized_end=1323
-  _globals['_GATEWAYSTATUS_CONFIGENTRY']._serialized_start=1171
-  _globals['_GATEWAYSTATUS_CONFIGENTRY']._serialized_end=1243
-  _globals['_GATEWAYINFO']._serialized_start=1325
-  _globals['_GATEWAYINFO']._serialized_end=1450
-  _globals['_GATEWAYINFO_ENTRIESENTRY']._serialized_start=1390
-  _globals['_GATEWAYINFO_ENTRIESENTRY']._serialized_end=1450
-  _globals['_GATEWAYLOGS']._serialized_start=1452
-  _globals['_GATEWAYLOGS']._serialized_end=1479
-  _globals['_ACTIONSTATUS']._serialized_start=1482
-  _globals['_ACTIONSTATUS']._serialized_end=1628
-  _globals['_DOWNLINKMESSAGE']._serialized_start=1631
-  _globals['_DOWNLINKMESSAGE']._serialized_end=1901
-  _globals['_TXPACKET']._serialized_start=1903
-  _globals['_TXPACKET']._serialized_end=1973
-  _globals['_GATEWAYACTION']._serialized_start=1975
-  _globals['_GATEWAYACTION']._serialized_end=2006
-  _globals['_BRIDGEUPGRADE']._serialized_start=2009
-  _globals['_BRIDGEUPGRADE']._serialized_end=2174
-  _globals['_GATEWAYCONFIG']._serialized_start=2177
-  _globals['_GATEWAYCONFIG']._serialized_end=2503
-  _globals['_GATEWAYCONFIG_CONFIGENTRY']._serialized_start=1171
-  _globals['_GATEWAYCONFIG_CONFIGENTRY']._serialized_end=1243
-  _globals['_CUSTOMMESSAGE']._serialized_start=2506
-  _globals['_CUSTOMMESSAGE']._serialized_end=2635
-  _globals['_CUSTOMMESSAGE_ENTRIESENTRY']._serialized_start=1390
-  _globals['_CUSTOMMESSAGE_ENTRIESENTRY']._serialized_end=1450
-# @@protoc_insertion_point(module_scope)

gw_certificate/gw_certificate.py

@@ -1,154 +0,0 @@
-# External Imports
-import time
-import datetime
-import os
-import webbrowser
-from jinja2 import Environment, FileSystemLoader
-import pkg_resources
-import time
-import importlib.metadata
-from wiliot_core.utils.utils import WiliotDir
-
-# Internal Imports
-from gw_certificate.common.analysis_data_bricks import initialize_logger
-from gw_certificate.common.debug import debug_print
-from gw_certificate.interface.ble_sniffer import BLESniffer
-from gw_certificate.interface.uart_if import UARTError, UARTInterface
-from gw_certificate.interface.ble_simulator import BLESimulator
-from gw_certificate.interface.mqtt import MqttClient
-from gw_certificate.tests import *
-from gw_certificate.interface.uart_ports import get_uart_ports
-from gw_certificate.api_if.gw_capabilities import GWCapabilities, ConfigurationData
-from gw_certificate.tests import TESTS_NO_UART
-
-import gw_certificate.cert_results as cert_results
-
-GW_CERT_VERSION = importlib.metadata.version("wiliot-certificate")
-
-class GWCertificateError(Exception):
-    pass
-
-
-class TemplateEngine:
-    def __init__(self):
-        self.template_dir = pkg_resources.resource_filename(__name__, 'templates')
-        self.env = Environment(loader=FileSystemLoader(self.template_dir))
-
-    def get_template(self, template_name):
-        return self.env.get_template(template_name)
-
-    def render_template(self, template_name, **kwargs):
-        template = self.get_template(template_name)
-        return template.render(**kwargs)
-
-
-class GWCertificate:
-    def __init__(self, gw_id, owner_id, actions:list, tests:list, topic_suffix='', update_fw=False, stress_pps=None,
-                 aggregation_time=0, env='prod'):
-        # Runtime
-        self.env_dirs = WiliotDir()
-        self.current_datetime = datetime.datetime.now()
-        self.duration = None
-        self.certificate_dir = os.path.join(self.env_dirs.get_wiliot_root_app_dir(), 'gw-certificate', self.current_datetime.strftime('%Y%m%d_%H%M%S'))
-        self.env_dirs.create_dir(self.certificate_dir)
-        self.logger_filename = initialize_logger(self.certificate_dir)
-        self.logger_filepath = os.path.join(self.certificate_dir, f'{self.logger_filename}.log')
-        self.mqtt_logger_filepath = os.path.join(self.certificate_dir, f'{self.logger_filename}_mqtt.log')
-        self.sniffer_logger_filepath = os.path.join(self.certificate_dir, f'{self.logger_filename}_sniffer.log')
-        self.result_html_path = os.path.join(self.certificate_dir, f"results_{self.current_datetime.strftime('%Y%m%d_%H%M%S')}.html")
-        self.result_pdf_path = os.path.join(self.certificate_dir, f"results_{self.current_datetime.strftime('%Y%m%d_%H%M%S')}.pdf")
-        self.template_engine = TemplateEngine()
-        self.env = env
-        self.error = ""
-
-        # Version 
-        self.gw_cert_version = GW_CERT_VERSION
-        
-        # GW & specific tests related
-        self.gw_id = gw_id
-        self.owner_id = owner_id
-        self.topic_suffix = topic_suffix
-        self.mqttc = MqttClient(gw_id, owner_id, self.mqtt_logger_filepath, topic_suffix=topic_suffix, broker='eclipse')
-        self.gw_capabilities = GWCapabilities()
-        self.stress_pps = stress_pps
-        self.aggregation_time = aggregation_time
-        self.actions = actions
-        self.cfg_data = ConfigurationData()
-        
-        # UART-related. Require only when running tests that need it
-        self.use_uart = not all(test in TESTS_NO_UART for test in tests)
-        if self.use_uart:
-            self.uart = None
-            self.uart_comports = get_uart_ports()
-            debug_print(f'UART Ports:{self.uart_comports}')
-            if len(self.uart_comports) < 1:
-                self.error = "A Wiliot certification kit must be connected to USB!"
-                raise GWCertificateError(self.error)
-        
-            for port in self.uart_comports:
-                try:
-                    self.uart = UARTInterface(port, update_fw=update_fw)
-                    break
-                except UARTError as e:
-                    debug_print(f'Port: {port} - {e}')
-            if type(self.uart) is not UARTInterface:
-                self.error = "Cannot initialize any port!"
-                raise GWCertificateError(self.error)
-            self.ble_sim = BLESimulator(self.uart)
-            self.sniffer = BLESniffer(self.uart, logger_filepath=self.sniffer_logger_filepath)
-        
-        # Tests
-        self.tests = [t(**self.__dict__) for t in tests]
-        debug_print(f'Running Tests: {self.tests}')
-
-    def runtime(self):
-        datetime.timedelta
-        return datetime.datetime.now() - self.current_datetime
-    
-    def run_tests(self):
-        debug_print("Sleeping 20 seconds after mqtt connect")
-        time.sleep(20)
-
-        # capabilities_received = ConnectionTest in self.tests
-        for test in self.tests:
-            # if capabilities_received:
-            #     if (type(test) == DownlinkTest and self.gw_capabilities.downlinkSupported == False):
-            #         debug_print(f'# Skipping {type(test)} since it is not a supported capability. #')
-            #         continue
-            test.prepare_test()
-            test.run()
-            test.end_test()
-        
-    def create_results_html(self):
-        with open(self.logger_filepath, 'r') as f:
-            log = f.read().split('\n')
-        with open(self.mqtt_logger_filepath, 'r') as f:
-            mqtt_log = f.read().split('\n')
-        if self.use_uart:
-            with open(self.sniffer_logger_filepath, 'r') as f:
-                sniffer_log = f.read().split('\n')
-        else:
-            sniffer_log = []
-
-        
-        html = self.template_engine.render_template('results.html', tests=self.tests,
-                                                    log = log,
-                                                    mqtt_log = mqtt_log,
-                                                    sniffer_log = sniffer_log,
-                                                    gw_id = self.gw_id,
-                                                    version = self.gw_cert_version,
-                                                    datetime = self.current_datetime.strftime('%Y%m%d_%H%M%S'))
-        with open(self.result_html_path, 'w', encoding="utf-8") as f:
-            f.write(html)
-        cert_results.generate_pdf_results_file(self)
-        debug_print("Test Finished. Results HTML Saved: " + self.result_html_path)
-        webbrowser.open('file://' + os.path.realpath(self.result_html_path))
-        webbrowser.open('file://' + os.path.realpath(self.result_pdf_path))
-
-if __name__ == "__main__":
-    from api_secrets import *
-    gw_id = 'gw_id'
-    owner_id = 'owner_id'
-    gwc = GWCertificate(gw_id=gw_id,owner_id=owner_id, tests=[ConnectionTest, DownlinkTest, UplinkTest, StressTest])
-    gwc.run_tests()
-    gwc.create_results_html()

gw_certificate/gw_certificate_cli.py

@@ -1,87 +0,0 @@
-
-import time
-from argparse import ArgumentParser, SUPPRESS
-
-from gw_certificate.common.debug import debug_print
-from gw_certificate.gw_certificate import GWCertificate, GW_CERT_VERSION
-from gw_certificate.tests import TESTS
-from gw_certificate.tests.actions import ACTIONS_STAGES
-from gw_certificate.tests.throughput import STRESS_DEFAULT_PPS, StressTest
-from gw_certificate.tests.registration import REG_CERT_OWNER_ID, RegistrationTest
-from gw_certificate.tests.uplink import UplinkTest
-
-def filter_by_args(args_list, list_to_filter):
-    chosen_list = []
-    for entry in list_to_filter:
-        for arg in args_list:
-            if arg in entry.__name__.lower() and entry not in chosen_list:
-                chosen_list.append(entry)
-    return chosen_list
-
-def filter_tests(tests_names):
-    return filter_by_args(tests_names, TESTS)
-
-def filter_actions(actions_names):
-    return filter_by_args(actions_names, ACTIONS_STAGES)
-
-def main():
-    usage = (
-        "usage: wlt-gw-certificate [-h] -owner OWNER -gw GW\n"
-        f"                          [-tests {{connection, uplink, downlink, actions, stress}}] [-noupdate] [-pps {STRESS_DEFAULT_PPS}]\n"
-        f"                          [-actions {{info, reboot, bridgeota}}] [-agg AGG] [-env {{prod, test, dev}}]"
-        )
-
-    parser = ArgumentParser(prog='wlt-gw-certificate',
-                            description=f'Gateway Certificate v{GW_CERT_VERSION} - CLI Tool to test Wiliot GWs', usage=usage)
-
-    required = parser.add_argument_group('required arguments')
-    required.add_argument('-gw', type=str, help="Gateway ID", required=True)
-    optional = parser.add_argument_group('optional arguments')
-    optional.add_argument('-owner', type=str, help="Owner ID", required=False, default=REG_CERT_OWNER_ID)
-    optional.add_argument('-suffix', type=str, help="Topic suffix", default='', required=False)
-    optional.add_argument('-tests', type=str, choices=['registration', 'connection', 'uplink', 'downlink', 'actions', 'stress'],
-                          help="Tests to run. Registration omitted by default.", required=False, nargs='+',
-                          default=['connection', 'uplink', 'downlink', 'actions', 'stress'])
-    optional.add_argument('-actions', type=str, choices=['info', 'reboot', 'bridgeota'],
-                          help="Action stages to run under ActionsTest", required=False, nargs='+', default=['info', 'reboot', 'bridgeota'])
-    optional.add_argument('-pps', type=int, help='Single packets-per-second rate to simulate in the stress test',
-                          choices=STRESS_DEFAULT_PPS, default=None, required=False)
-    optional.add_argument('-agg', type=int, help='Aggregation time [seconds] the Uplink stages wait before processing results',
-                          default=0, required=False)
-    optional.add_argument('-env', type=str, help='Environment for the RegistrationTest & BridgeOTAStage',
-                          choices=['prod', 'test', 'dev'], default='prod', required=False)
-    optional.add_argument('-update', action='store_true', help=SUPPRESS, default=True, required=False)
-    optional.add_argument('-noupdate', action='store_true', help='Skip the default certification kit firmware update', default=False, required=False)
-    args = parser.parse_args()
-
-    topic_suffix = '' if args.suffix == '' else '-'+args.suffix
-    tests = filter_tests(args.tests)
-    actions = filter_actions(args.actions)
-
-    # Validate args combination before running
-    if args.pps != None and StressTest not in tests:
-        parser.error("Packets per second (-pps) flag can only be used when 'stress' is included in test list (e.g. -tests stress)")
-    if args.agg != 0 and not any(t in tests for t in (UplinkTest, StressTest)):
-        parser.error("Aggregation time (-agg) flag can only be used when 'uplink' or 'stress' are included in test list (e.g. -tests uplink)")
-
-    if RegistrationTest in tests:
-        if not all(test == RegistrationTest for test in tests):
-            parser.error("The registration test must be run on it's own, without any others tests.")
-        if args.owner != REG_CERT_OWNER_ID:
-            parser.error(f"The registration test must be run without the -owner flag (defaults to {REG_CERT_OWNER_ID}).")
-    elif args.owner == REG_CERT_OWNER_ID:
-        parser.error("When running any test other than registration, the gateway must be registered to an owner which should be provided using the '-owner' flag.")
-
-
-    gwc = GWCertificate(gw_id=args.gw, owner_id=args.owner, topic_suffix=topic_suffix, tests=tests, update_fw=(not args.noupdate),
-                        stress_pps=args.pps, aggregation_time=args.agg, env=args.env, actions=actions)
-    debug_print(f"All arguments: {vars(args)}")
-    gwc.run_tests()
-    gwc.create_results_html()
-
-def main_cli():
-    main()
-
-if __name__ == '__main__':
-    main()
-    

gw_certificate/interface/ble_simulator.py

@@ -1,61 +0,0 @@
-import datetime
-import time
-
-from gw_certificate.interface.uart_if import UARTInterface
-from gw_certificate.common.debug import debug_print
-from gw_certificate.interface.if_defines import *
-
-class BLESimulator():
-    def __init__(self, uart:UARTInterface):
-        self.uart = uart
-        self.sim_mode = False
-
-    def set_sim_mode(self, sim_mode):
-        self.uart.flush()
-        mode_dict = {True: 1, False: 0}
-        self.sim_mode = sim_mode
-        self.uart.reset_gw()
-        self.uart.write_ble_command(f"!ble_sim_init {mode_dict[sim_mode]}")
-        if not sim_mode:
-            self.uart.reset_gw()
-        time.sleep(3)
-
-    def send_packet(self, raw_packet, duplicates=DEFAULT_DUPLICATES, output_power=DEFAULT_OUTPUT_POWER, channel=SEND_ALL_ADV_CHANNELS,
-                    delay=DEFAULT_DELAY, print_for_debug=True):
-        assert self.sim_mode is True, 'BLE Sim not initialized!'
-        if len(raw_packet) == 62:
-            # Add ADVA
-            raw_packet = DEFAULT_ADVA + raw_packet
-        if len(raw_packet) != 74:
-            raise ValueError('Raw Packet must be 62/74 chars long!')
-        self.uart.write_ble_command(f"!ble_sim {str(raw_packet)} {str(duplicates)} {str(output_power)} {str(channel)} {str(delay)}", print_for_debug=print_for_debug)
-        if delay > 0:
-            diff = time.perf_counter()
-            time.sleep((delay/1000) * duplicates)
-            diff = time.perf_counter() - diff
-            if print_for_debug:
-                debug_print(f'Desired Delay: {(delay/1000) * duplicates} Actual Delay {diff}')
-
-    
-    def send_data_si_pair(self, data_packet, si_packet, duplicates, output_power=DEFAULT_OUTPUT_POWER, delay=DEFAULT_DELAY, packet_error=None):
-        if packet_error is None:
-            packet_error = [True for i in range (duplicates * 2)]
-        # debug_print(packet_error)
-        # print(f'delay {delay}')
-        packet_to_send = data_packet
-        def switch_packet(packet_to_send):
-            if packet_to_send == data_packet:
-                return si_packet
-            else:
-                return data_packet
-        for dup in range(duplicates * 2):
-            diff = time.perf_counter()
-            if packet_error[dup]:
-                debug_print(f'Sending Packet {dup}')
-                self.send_packet(packet_to_send, duplicates=1, output_power=output_power, channel=SEND_ALL_ADV_CHANNELS, delay=0)
-            else:
-                debug_print(f'Dropping Packet {dup}')
-            time.sleep(delay/1000)
-            diff = time.perf_counter() - diff
-            debug_print(f'Desired Delay: {delay/1000} Actual Delay {diff}')
-            packet_to_send = switch_packet(packet_to_send)