mirror of
https://github.com/onyx-and-iris/slobs-cli.git
synced 2025-06-27 15:20:24 +01:00
onyx-and-iris
c02ffac403
allow rich to handle all console output. util.check_mark is now used to pass back colourless check/cross marks if NO_COLOR is set or --style/SLOBS_STYLE was not set.
120 lines
3.0 KiB
Python
120 lines
3.0 KiB
Python
"""module for managing recording commands in Slobs CLI."""
|
|
|
|
import asyncclick as click
|
|
from anyio import create_task_group
|
|
from pyslobs import StreamingService
|
|
|
|
from . import console
|
|
from .cli import cli
|
|
from .errors import SlobsCliError
|
|
|
|
|
|
@cli.group()
|
|
def record():
|
|
"""Manage recording in Slobs CLI."""
|
|
|
|
|
|
@record.command()
|
|
@click.pass_context
|
|
async def start(ctx: click.Context):
|
|
"""Start recording."""
|
|
conn = ctx.obj['connection']
|
|
ss = StreamingService(conn)
|
|
|
|
async def _run():
|
|
model = await ss.get_model()
|
|
active = model.recording_status != 'offline'
|
|
|
|
if active:
|
|
conn.close()
|
|
raise SlobsCliError('Recording is already active.')
|
|
|
|
await ss.toggle_recording()
|
|
console.out.print('Recording started.')
|
|
|
|
conn.close()
|
|
|
|
try:
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
except* SlobsCliError as excgroup:
|
|
raisable = next(iter(excgroup.exceptions))
|
|
raise raisable
|
|
|
|
|
|
@record.command()
|
|
@click.pass_context
|
|
async def stop(ctx: click.Context):
|
|
"""Stop recording."""
|
|
conn = ctx.obj['connection']
|
|
ss = StreamingService(conn)
|
|
|
|
async def _run():
|
|
model = await ss.get_model()
|
|
active = model.recording_status != 'offline'
|
|
|
|
if not active:
|
|
conn.close()
|
|
raise SlobsCliError('Recording is already inactive.')
|
|
|
|
await ss.toggle_recording()
|
|
console.out.print('Recording stopped.')
|
|
|
|
conn.close()
|
|
|
|
try:
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
except* SlobsCliError as excgroup:
|
|
raisable = next(iter(excgroup.exceptions))
|
|
raise raisable
|
|
|
|
|
|
@record.command()
|
|
@click.pass_context
|
|
async def status(ctx: click.Context):
|
|
"""Get recording status."""
|
|
conn = ctx.obj['connection']
|
|
ss = StreamingService(conn)
|
|
|
|
async def _run():
|
|
model = await ss.get_model()
|
|
active = model.recording_status != 'offline'
|
|
|
|
if active:
|
|
console.out.print('Recording is currently active.')
|
|
else:
|
|
console.out.print('Recording is currently inactive.')
|
|
|
|
conn.close()
|
|
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
|
|
|
|
@record.command()
|
|
@click.pass_context
|
|
async def toggle(ctx: click.Context):
|
|
"""Toggle recording status."""
|
|
conn = ctx.obj['connection']
|
|
ss = StreamingService(conn)
|
|
|
|
async def _run():
|
|
model = await ss.get_model()
|
|
active = model.recording_status != 'offline'
|
|
|
|
await ss.toggle_recording()
|
|
if active:
|
|
console.out.print('Recording stopped.')
|
|
else:
|
|
console.out.print('Recording started.')
|
|
|
|
conn.close()
|
|
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|