mirror of
https://github.com/onyx-and-iris/slobs-cli.git
synced 2025-06-27 15:20:24 +01:00
onyx-and-iris
d33c209d7c
add exception group handling for all commands that may raise exceptions add comments to for-else blocks
137 lines
3.5 KiB
Python
137 lines
3.5 KiB
Python
import asyncclick as click
|
|
from anyio import create_task_group
|
|
from pyslobs import TransitionsService
|
|
|
|
from .cli import cli
|
|
from .errors import SlobsCliError
|
|
|
|
|
|
@cli.group()
|
|
def studiomode():
|
|
"""Studio mode management commands."""
|
|
|
|
|
|
@studiomode.command()
|
|
@click.pass_context
|
|
async def enable(ctx: click.Context):
|
|
"""Enable studio mode."""
|
|
|
|
conn = ctx.obj["connection"]
|
|
ts = TransitionsService(conn)
|
|
|
|
async def _run():
|
|
model = await ts.get_model()
|
|
if model.studio_mode:
|
|
conn.close()
|
|
raise SlobsCliError("Studio mode is already enabled.")
|
|
|
|
await ts.enable_studio_mode()
|
|
click.echo("Studio mode enabled successfully.")
|
|
conn.close()
|
|
|
|
try:
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
except* SlobsCliError as excgroup:
|
|
for e in excgroup.exceptions:
|
|
raise e
|
|
|
|
|
|
@studiomode.command()
|
|
@click.pass_context
|
|
async def disable(ctx: click.Context):
|
|
"""Disable studio mode."""
|
|
|
|
conn = ctx.obj["connection"]
|
|
ts = TransitionsService(conn)
|
|
|
|
async def _run():
|
|
model = await ts.get_model()
|
|
if not model.studio_mode:
|
|
conn.close()
|
|
raise SlobsCliError("Studio mode is already disabled.")
|
|
|
|
await ts.disable_studio_mode()
|
|
click.echo("Studio mode disabled successfully.")
|
|
conn.close()
|
|
|
|
try:
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
except* SlobsCliError as excgroup:
|
|
for e in excgroup.exceptions:
|
|
raise e
|
|
|
|
|
|
@studiomode.command()
|
|
@click.pass_context
|
|
async def status(ctx: click.Context):
|
|
"""Check the status of studio mode."""
|
|
|
|
conn = ctx.obj["connection"]
|
|
ts = TransitionsService(conn)
|
|
|
|
async def _run():
|
|
model = await ts.get_model()
|
|
if model.studio_mode:
|
|
click.echo("Studio mode is currently enabled.")
|
|
else:
|
|
click.echo("Studio mode is currently disabled.")
|
|
conn.close()
|
|
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
|
|
|
|
@studiomode.command()
|
|
@click.pass_context
|
|
async def toggle(ctx: click.Context):
|
|
"""Toggle studio mode."""
|
|
|
|
conn = ctx.obj["connection"]
|
|
ts = TransitionsService(conn)
|
|
|
|
async def _run():
|
|
model = await ts.get_model()
|
|
if model.studio_mode:
|
|
await ts.disable_studio_mode()
|
|
click.echo("Studio mode disabled successfully.")
|
|
else:
|
|
await ts.enable_studio_mode()
|
|
click.echo("Studio mode enabled successfully.")
|
|
conn.close()
|
|
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
|
|
|
|
@studiomode.command()
|
|
@click.pass_context
|
|
async def force_transition(ctx: click.Context):
|
|
"""Force a transition in studio mode."""
|
|
|
|
conn = ctx.obj["connection"]
|
|
ts = TransitionsService(conn)
|
|
|
|
async def _run():
|
|
model = await ts.get_model()
|
|
if not model.studio_mode:
|
|
conn.close()
|
|
raise SlobsCliError("Studio mode is not enabled.")
|
|
|
|
await ts.execute_studio_mode_transition()
|
|
click.echo("Forced studio mode transition.")
|
|
conn.close()
|
|
|
|
try:
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
except* SlobsCliError as excgroup:
|
|
for e in excgroup.exceptions:
|
|
raise e
|