remotivelabs-cli 0.0.12__tar.gz → 0.0.13__tar.gz

{remotivelabs_cli-0.0.12 → remotivelabs_cli-0.0.13}/PKG-INFO

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

{remotivelabs_cli-0.0.12 → remotivelabs_cli-0.0.13}/cli/broker/lib/broker.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import binascii
+import datetime
 import ntpath
 import os
 import posixpath
@@ -8,6 +9,7 @@ import queue
 import signal as os_signal
 import tempfile
 import time
+import typing
 import zipfile
 from dataclasses import dataclass
 from threading import Thread
@@ -181,6 +183,27 @@ class Broker:
             self.__check_playbackmode_result(status)
         return status
 
+    def listen_on_playback(self, callback: Callable[[str], None]):
+        def get_mode(mode: int):
+            if mode == 0:
+                return "playing"
+            if mode == 1:
+                return "paused"
+            if mode == 2:
+                return "stopped"
+
+        sub = self.traffic_stub.PlayTrafficStatus(br.common_pb2.Empty())
+        for playback_state in sub:
+            p = typing.cast(br.traffic_api_pb2.PlaybackInfos, playback_state)
+            offset_time = int(p.playbackInfo[0].playbackMode.offsetTime / 1000000)
+            start_time = p.playbackInfo[0].playbackMode.startTime
+            end_time = p.playbackInfo[0].playbackMode.endTime
+            mode = p.playbackInfo[0].playbackMode.mode
+
+            length = int((end_time - start_time) / 1000000)
+
+            callback(f"{(datetime.timedelta(seconds=offset_time))} / {(datetime.timedelta(seconds=length))} ({get_mode(mode)})")
+
     def pause(self, namespace: str, path: str, silent: bool = False):
         playback_list = [
             {

{remotivelabs_cli-0.0.12 → remotivelabs_cli-0.0.13}/cli/cloud/recordings.py

@@ -11,7 +11,7 @@ from pathlib import Path
 import grpc
 import requests
 import typer
-from rich.progress import track
+from rich.progress import Progress, SpinnerColumn, TextColumn, track
 
 from cli.errors import ErrorPrinter
 
@@ -383,6 +383,18 @@ def pause(
     _do_change_playback_mode("pause", recording_session, broker, project)
 
 
+@app.command(name="playback-status")
+def playback_status(
+    recording_session: str = typer.Argument(..., help="Recording session id", envvar="REMOTIVE_CLOUD_RECORDING_SESSION"),
+    broker: str = typer.Option(None, help="Broker to use"),
+    project: str = typer.Option(..., help="Project ID", envvar="REMOTIVE_CLOUD_PROJECT"),
+):
+    """
+    Display status and progress of the recording beeing played
+    """
+    _do_change_playback_mode("status", recording_session, broker, project)
+
+
 @app.command()
 def seek(
     recording_session: str = typer.Argument(..., help="Recording session id", envvar="REMOTIVE_CLOUD_RECORDING_SESSION"),
@@ -453,6 +465,12 @@ def _do_change_playback_mode(mode: str, recording_session: str, broker: str, pro
         broker.seek(files, int(seconds * 1000000), True)
     elif mode == "stop":
         broker.seek(files, 0, True)
+    elif mode == "status":
+        p = Progress(SpinnerColumn(), TextColumn("[progress.description]{task.description}"), transient=True)
+        t = p.add_task("label", total=1)
+
+        with p:
+            broker.listen_on_playback(lambda c: p.update(t, description=str(c)))
     else:
         raise Exception(f"Illegal command {mode}")
 

{remotivelabs_cli-0.0.12 → remotivelabs_cli-0.0.13}/pyproject.toml

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