remotivelabs-cli 0.0.36__py3-none-any.whl → 0.0.37__py3-none-any.whl

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 = [
             {

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)

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.dist-info → remotivelabs_cli-0.0.37.dist-info}/METADATA

@@ -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.dist-info → remotivelabs_cli-0.0.37.dist-info}/RECORD

@@ -3,13 +3,13 @@ cli/broker/brokers.py,sha256=oUadEL6xQ4bhXucBH-ZjL67VuERf19kn1g240v_lEpg,3197
 cli/broker/export.py,sha256=3sG9i6ZwOQW6snu87NSzOL2_giQTYQMzQlpPg7z8n78,4431
 cli/broker/files.py,sha256=_MVwitQ5Z9-lNDb3biXqnlkKti8rizTEw0nnAViussU,4181
 cli/broker/lib/__about__.py,sha256=xnZ5V6ZcHW9dhWLWdMzVjYJbEnMKpeXm0_S_mbNzypE,141
-cli/broker/lib/broker.py,sha256=_GwgXkILcw5pYhvWEUKotnMztbfSRKTNXAhx1aVv-8s,24523
+cli/broker/lib/broker.py,sha256=HbGo83j6ipaol0uJ93Me78pOPjy7p8PUjYT9xZmRT-w,25086
 cli/broker/license_flows.py,sha256=qJplaeugkUiypFGPdEIl5Asqlf7W3geJ-wU-QbYMP_8,7216
 cli/broker/licenses.py,sha256=Ddl243re8RoeP9CoWWbIzwDePQ9l8r7ixmbd1gqn8f0,3973
 cli/broker/playback.py,sha256=hdDKXGPuIE3gcT-kgQltgn5jsPzK19Yh9hiNcgtkLX0,3992
 cli/broker/record.py,sha256=Oa6hUpS0Dgnt0f6Ig33vl0Jy8wN7wMXfemaxXWjRVoQ,1414
 cli/broker/scripting.py,sha256=8577_C6siOk90s4G1ItIfAoFIUAkS0ItUl5kqR0cD-k,3792
-cli/broker/signals.py,sha256=KflNaVFgsKFi1suec7fSkCfGhcJ0LfPTMRGVQUlsU-0,7058
+cli/broker/signals.py,sha256=BHNWH9o1msPJQgw7bplg7W1JNtB4Us_abQp5H8Y3pRY,8079
 cli/cloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cli/cloud/auth/__init__.py,sha256=MtQ01-n8CgZb9Y_SvxwZUgj44Yo0dFAU3_XwhQiUYtw,54
 cli/cloud/auth/cmd.py,sha256=Lew9gJDC4pU2Hvbrbe4ZPoanphN0CiNwbpLAo9R00gM,1622
@@ -35,7 +35,7 @@ cli/cloud/uri.py,sha256=QZCus--KJQlVwGCOzZqiglvj8VvSRKxfVvN33Pilgyg,3616
 cli/connect/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cli/connect/connect.py,sha256=U--6dtHxUlvE81J37rABFez4TbF7AXWOpZYZnL7sPMY,3994
 cli/connect/protopie/protopie.py,sha256=xEfCgoW4p0vlWnSBvgBeDN9QSlyKcOcHfa6EyeAyVoM,6364
-cli/errors.py,sha256=CXYArw1W82bRFwJkJ3tD-Ek1huKeah502DGMvPxHYFo,1366
+cli/errors.py,sha256=8_BcTPX3mrPFDQJKKBg6ERjs6HSOiewrY86K1Jays74,1495
 cli/remotive.py,sha256=rwCIPFuWn0CNxLU2Sgwug5UVjletC1DTJrxUtGkVGPs,1907
 cli/settings/__init__.py,sha256=4Mj9DFvRBN0LKn6PPLrb-BmuObP707IYeRmg7t7aycg,288
 cli/settings/cmd.py,sha256=sj1bYgMXUAcS4sx_dZjPVCEjoVAR1p_H-KYb7w02wvU,1992
@@ -46,8 +46,8 @@ cli/tools/can/RemotiveLabs.can1.log,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 cli/tools/can/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cli/tools/can/can.py,sha256=8uATViSFlpkdSiIm4fzbuQi1_m7V9Pym-K17TaJQRHU,2262
 cli/tools/tools.py,sha256=0KU-hXR1f9xHP4BOG9A9eXfmICLmNuQCOU8ueF6iGg0,198
-remotivelabs_cli-0.0.36.dist-info/LICENSE,sha256=qDPP_yfuv1fF-u7EfexN-cN3M8aFgGVndGhGLovLKz0,608
-remotivelabs_cli-0.0.36.dist-info/METADATA,sha256=jZI1KYzQA1j9opVPDHcHFi06j-i9MUuY0-ln8pfAqXw,1359
-remotivelabs_cli-0.0.36.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-remotivelabs_cli-0.0.36.dist-info/entry_points.txt,sha256=lvDhPgagLqW_KTnLPCwKSqfYlEp-1uYVosRiPjsVj10,45
-remotivelabs_cli-0.0.36.dist-info/RECORD,,
+remotivelabs_cli-0.0.37.dist-info/LICENSE,sha256=qDPP_yfuv1fF-u7EfexN-cN3M8aFgGVndGhGLovLKz0,608
+remotivelabs_cli-0.0.37.dist-info/METADATA,sha256=kigJThZ9liO7Qm3yg2u5mw8WPQyx9mb-aFiEbVeeb4Y,1359
+remotivelabs_cli-0.0.37.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+remotivelabs_cli-0.0.37.dist-info/entry_points.txt,sha256=lvDhPgagLqW_KTnLPCwKSqfYlEp-1uYVosRiPjsVj10,45
+remotivelabs_cli-0.0.37.dist-info/RECORD,,