remotivelabs-cli 0.0.36__tar.gz → 0.0.37__tar.gz

{remotivelabs_cli-0.0.36 → remotivelabs_cli-0.0.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: remotivelabs-cli
-Version: 0.0.36
+Version: 0.0.37
 Summary: CLI for operating RemotiveCloud and RemotiveBroker
 Author: Johan Rask
 Author-email: johan.rask@remotivelabs.com

{remotivelabs_cli-0.0.36 → remotivelabs_cli-0.0.37}/cli/broker/lib/broker.py

@@ -14,6 +14,7 @@ from threading import Thread
 from typing import Any, Callable, Dict, Iterable, List, Sequence, Union
 
 import grpc
+import remotivelabs.broker.generated.sync.network_api_pb2 as network_api
 import remotivelabs.broker.generated.sync.traffic_api_pb2 as traffic_api
 import remotivelabs.broker.sync as br
 import remotivelabs.broker.sync.helper as br_helper
@@ -220,6 +221,13 @@ class Broker:
                     self.play(recording_and_namespace)
             callback(offset_length, total_length, get_mode(mode))
 
+    def listen_on_frame_distribution(self, namespace: str, callback: Callable[[Dict[str, Any]], None]) -> None:
+        config = network_api.FramesDistributionConfig(namespace=br.common_pb2.NameSpace(name=namespace))
+        frame_distribution_stream = self.network_stub.SubscribeToFramesDistribution(config)
+        for frame in frame_distribution_stream:
+            f = MessageToDict(frame, including_default_value_fields=True, preserving_proto_field_name=True)
+            callback(f)
+
     def pause(self, namespace: str, path: str, silent: bool = False) -> br.traffic_api_pb2.PlaybackInfos:
         playback_list = [
             {

{remotivelabs_cli-0.0.36 → remotivelabs_cli-0.0.37}/cli/broker/signals.py

@@ -4,6 +4,7 @@ import json
 import numbers
 import os
 import signal as os_signal
+from datetime import datetime
 from pathlib import Path
 from typing import Any, Dict, Iterable, List, TypedDict, Union
 
@@ -64,7 +65,7 @@ def read_scripted_code_file(file_path: Path) -> bytes:
 def subscribe(  # noqa: C901
     url: str = typer.Option(..., help="Broker URL", envvar="REMOTIVE_BROKER_URL"),
     api_key: str = typer.Option(None, help="Cloud Broker API-KEY or access token", envvar="REMOTIVE_BROKER_API_KEY"),
-    signal: List[str] = typer.Option(None, help="Signal names to subscribe to, mandatory when not using script"),
+    signal: List[str] = typer.Option([], help="Signal names to subscribe to, mandatory when not using script"),
     script: Path = typer.Option(
         None,
         exists=True,
@@ -96,7 +97,7 @@ def subscribe(  # noqa: C901
 
     if script is None:
         if len(signal) == 0:
-            ErrorPrinter.print_generic_error("You must use include at least one signal and one namespace or use script when subscribing")
+            ErrorPrinter.print_generic_error("You must use --signal or use --script when subscribing")
             exit(1)
 
     if script is not None:
@@ -194,3 +195,29 @@ def namespaces(
         print(json.dumps(namespaces_json))
     except grpc.RpcError as rpc_error:
         ErrorPrinter.print_grpc_error(rpc_error)
+
+
+@app.command()
+def frame_distribution(
+    url: str = typer.Option(..., help="Broker URL", envvar="REMOTIVE_BROKER_URL"),
+    api_key: str = typer.Option(None, help="Cloud Broker API-KEY or access token", envvar="REMOTIVE_BROKER_API_KEY"),
+    namespace: str = typer.Option(..., help="Namespace"),
+) -> None:
+    """
+    Use this command to get frames currently available on the specified namespace.
+    """
+    try:
+        broker = Broker(url, api_key)
+
+        def on_data(data: Dict[str, Any]) -> None:
+            timestamp: str = datetime.now().strftime("%H:%M:%S")
+            distribution = data["countsByFrameId"]
+            if len(distribution) == 0:
+                ErrorPrinter.print_hint(f"{timestamp} - No frames available")
+            else:
+                for d in distribution:
+                    ErrorPrinter.print_generic_message(f"{timestamp}: {d}")
+
+        broker.listen_on_frame_distribution(namespace, on_data)
+    except grpc.RpcError as rpc_error:
+        ErrorPrinter.print_grpc_error(rpc_error)

{remotivelabs_cli-0.0.36 → remotivelabs_cli-0.0.37}/cli/errors.py

@@ -7,6 +7,8 @@ from rich.console import Console
 err_console = Console(stderr=True)
 
 
+# This should be renamed to Printer since print_generic_message is not an error message.
+# Then we can also add more things here
 class ErrorPrinter:
     @staticmethod
     def print_grpc_error(error: grpc.RpcError) -> None:

{remotivelabs_cli-0.0.36 → remotivelabs_cli-0.0.37}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "remotivelabs-cli"
-version = "0.0.36"
+version = "0.0.37"
 description = "CLI for operating RemotiveCloud and RemotiveBroker"
 authors = ["Johan Rask <johan.rask@remotivelabs.com>"]
 readme = "README.md"