diff --git a/src/telemetry/backend/service/HelperMethods.py b/src/telemetry/backend/service/HelperMethods.py index db56c9a76c2eefa1fc929ae73d05ca531ac21415..332b3f10d7ea9ec357897dcce62e856545d7f37c 100644 --- a/src/telemetry/backend/service/HelperMethods.py +++ b/src/telemetry/backend/service/HelperMethods.py @@ -19,12 +19,13 @@ from .collector_api.DriverInstanceCache import get_driver from common.proto.kpi_manager_pb2 import KpiId from common.tools.context_queries.Device import get_device from common.tools.context_queries.EndPoint import get_endpoint_names +from typing import List, Tuple, Optional LOGGER = logging.getLogger(__name__) def get_subscription_parameters( kpi_id : str, kpi_manager_client, context_client, duration, interval - ) -> list[tuple] | None: + ) -> Optional[List[Tuple]]: """ Method to get subscription parameters based on KPI ID. Returns a list of tuples with subscription parameters. @@ -95,7 +96,7 @@ def get_subscription_parameters( def get_collector_by_kpi_id(kpi_id: str, kpi_manager_client, context_client, driver_instance_cache - ) -> _Collector | None: + ) -> Optional[_Collector]: """ Method to get a collector instance based on KPI ID. Preconditions: diff --git a/src/telemetry/backend/service/collectors/__init__.py b/src/telemetry/backend/service/collectors/__init__.py index 1e07dc43fab0d52b3c6584eb168d0f7e3f991b29..b2391d6234b67595446b84cbcab80471cb35f72d 100644 --- a/src/telemetry/backend/service/collectors/__init__.py +++ b/src/telemetry/backend/service/collectors/__init__.py @@ -49,7 +49,7 @@ if LOAD_ALL_DEVICE_DRIVERS: ) if LOAD_ALL_DEVICE_DRIVERS: - from ...collectors.intcollector.INTCollector import INTCollector # pylint: disable=wrong-import-position + from .intcollector.INTCollector import INTCollector # pylint: disable=wrong-import-position COLLECTORS.append( (INTCollector, [ { @@ -58,4 +58,3 @@ if LOAD_ALL_DEVICE_DRIVERS: } ]) ) - diff --git a/src/telemetry/backend/service/collectors/gnmi_oc/SubscriptionNew.py b/src/telemetry/backend/service/collectors/gnmi_oc/SubscriptionNew.py index e34b2e4729e673faa8992b87dee67b1dfce8c86c..cbf0ebf424b6e6a9e4310d838250c9586b8459f3 100644 --- a/src/telemetry/backend/service/collectors/gnmi_oc/SubscriptionNew.py +++ b/src/telemetry/backend/service/collectors/gnmi_oc/SubscriptionNew.py @@ -41,7 +41,7 @@ class Subscription: metric_queue: Queue, mode: str = "stream", sample_interval_ns: int = 10_000_000_000, - heartbeat_interval_ns: int | None = None, # ← NEW + heartbeat_interval_ns: Optional[int] = None, # ← NEW encoding: str = "json_ietf", on_update: Optional[Callable[[dict], None]] = None, ) -> None: @@ -82,7 +82,7 @@ class Subscription: path_list: List[str], mode: str, sample_interval_ns: int, - heartbeat_interval_ns: int | None, + heartbeat_interval_ns: Optional[int], encoding: str, on_update: Optional[Callable[[dict], None]], ) -> None: # pragma: no cover diff --git a/src/telemetry/backend/collectors/intcollector/INTCollector.py b/src/telemetry/backend/service/collectors/intcollector/INTCollector.py similarity index 99% rename from src/telemetry/backend/collectors/intcollector/INTCollector.py rename to src/telemetry/backend/service/collectors/intcollector/INTCollector.py index 9d89827f4eecbbb80a092763488f87296acffb5a..5931f33f34b1d097c435c0a8483379026e819213 100644 --- a/src/telemetry/backend/collectors/intcollector/INTCollector.py +++ b/src/telemetry/backend/service/collectors/intcollector/INTCollector.py @@ -20,7 +20,7 @@ from apscheduler.schedulers.background import BackgroundScheduler from apscheduler.jobstores.memory import MemoryJobStore from apscheduler.executors.pool import ThreadPoolExecutor from datetime import datetime -from telemetry.backend.collector_api._Collector import _Collector +from telemetry.backend.service.collector_api._Collector import _Collector from scapy.all import * import struct diff --git a/src/telemetry/backend/collectors/intcollector/INTCollectorCommon.py b/src/telemetry/backend/service/collectors/intcollector/INTCollectorCommon.py similarity index 100% rename from src/telemetry/backend/collectors/intcollector/INTCollectorCommon.py rename to src/telemetry/backend/service/collectors/intcollector/INTCollectorCommon.py diff --git a/src/telemetry/backend/collectors/intcollector/__init__.py b/src/telemetry/backend/service/collectors/intcollector/__init__.py similarity index 100% rename from src/telemetry/backend/collectors/intcollector/__init__.py rename to src/telemetry/backend/service/collectors/intcollector/__init__.py