mirror of
https://github.com/onyx-and-iris/slobs-cli.git
synced 2025-06-27 15:20:24 +01:00
onyx-and-iris
f4421b3351
(we dont intend to iterate through them, we just want to raise the first one)
116 lines
2.9 KiB
Python
116 lines
2.9 KiB
Python
"""module for managing the replay buffer in Slobs CLI."""
|
|
|
|
import asyncclick as click
|
|
from anyio import create_task_group
|
|
from pyslobs import StreamingService
|
|
|
|
from .cli import cli
|
|
from .errors import SlobsCliError
|
|
|
|
|
|
@cli.group()
|
|
def stream():
|
|
"""Manage streaming in Slobs CLI."""
|
|
|
|
|
|
@stream.command()
|
|
@click.pass_context
|
|
async def start(ctx: click.Context):
|
|
"""Start the stream."""
|
|
conn = ctx.obj['connection']
|
|
ss = StreamingService(conn)
|
|
|
|
async def _run():
|
|
model = await ss.get_model()
|
|
active = model.streaming_status != 'offline'
|
|
|
|
if active:
|
|
conn.close()
|
|
raise SlobsCliError('Stream is already active.')
|
|
|
|
await ss.toggle_streaming()
|
|
click.echo('Stream 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
|
|
|
|
|
|
@stream.command()
|
|
@click.pass_context
|
|
async def stop(ctx: click.Context):
|
|
"""Stop the stream."""
|
|
conn = ctx.obj['connection']
|
|
ss = StreamingService(conn)
|
|
|
|
async def _run():
|
|
model = await ss.get_model()
|
|
active = model.streaming_status != 'offline'
|
|
|
|
if not active:
|
|
conn.close()
|
|
raise SlobsCliError('Stream is already inactive.')
|
|
|
|
await ss.toggle_streaming()
|
|
click.echo('Stream 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
|
|
|
|
|
|
@stream.command()
|
|
@click.pass_context
|
|
async def status(ctx: click.Context):
|
|
"""Get the current stream status."""
|
|
conn = ctx.obj['connection']
|
|
ss = StreamingService(conn)
|
|
|
|
async def _run():
|
|
model = await ss.get_model()
|
|
active = model.streaming_status != 'offline'
|
|
|
|
if active:
|
|
click.echo('Stream is currently active.')
|
|
else:
|
|
click.echo('Stream is currently inactive.')
|
|
conn.close()
|
|
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
|
|
|
|
@stream.command()
|
|
@click.pass_context
|
|
async def toggle(ctx: click.Context):
|
|
"""Toggle the stream status."""
|
|
conn = ctx.obj['connection']
|
|
ss = StreamingService(conn)
|
|
|
|
async def _run():
|
|
model = await ss.get_model()
|
|
active = model.streaming_status != 'offline'
|
|
|
|
await ss.toggle_streaming()
|
|
if active:
|
|
click.echo('Stream stopped.')
|
|
else:
|
|
click.echo('Stream started.')
|
|
|
|
conn.close()
|
|
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|