mirror of
https://github.com/onyx-and-iris/slobs-cli.git
synced 2025-07-21 20:31:47 +00:00
149 lines
4.7 KiB
Python
149 lines
4.7 KiB
Python
"""module for managing scenes in Slobs CLI."""
|
|
|
|
import asyncclick as click
|
|
from anyio import create_task_group
|
|
from pyslobs import ScenesService, TransitionsService
|
|
from terminaltables3 import AsciiTable
|
|
|
|
from .cli import cli
|
|
from .errors import SlobsCliError
|
|
|
|
|
|
@cli.group()
|
|
def scene():
|
|
"""Manage scenes in Slobs CLI."""
|
|
|
|
|
|
@scene.command()
|
|
@click.option('--id', is_flag=True, help='Include scene IDs in the output.')
|
|
@click.pass_context
|
|
async def list(ctx: click.Context, id: bool = False):
|
|
"""List all available scenes."""
|
|
conn = ctx.obj['connection']
|
|
ss = ScenesService(conn)
|
|
|
|
async def _run():
|
|
scenes = await ss.get_scenes()
|
|
if not scenes:
|
|
click.echo('No scenes found.')
|
|
conn.close()
|
|
return
|
|
|
|
active_scene = await ss.active_scene()
|
|
|
|
table_data = [
|
|
['Scene Name', 'ID', 'Active'] if id else ['Scene Name', 'Active']
|
|
]
|
|
for scene in scenes:
|
|
if scene.id == active_scene.id:
|
|
to_append = [click.style(scene.name, fg='green')]
|
|
else:
|
|
to_append = [click.style(scene.name, fg='blue')]
|
|
if id:
|
|
to_append.append(scene.id)
|
|
to_append.append('✅' if scene.id == active_scene.id else '')
|
|
|
|
table_data.append(to_append)
|
|
|
|
table = AsciiTable(table_data)
|
|
table.justify_columns = {
|
|
0: 'left',
|
|
1: 'left' if id else 'center',
|
|
2: 'center' if id else None,
|
|
}
|
|
click.echo(table.table)
|
|
|
|
conn.close()
|
|
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
|
|
|
|
@scene.command()
|
|
@click.option('--id', is_flag=True, help='Include scene IDs in the output.')
|
|
@click.pass_context
|
|
async def current(ctx: click.Context, id: bool = False):
|
|
"""Show the currently active scene."""
|
|
conn = ctx.obj['connection']
|
|
ss = ScenesService(conn)
|
|
|
|
async def _run():
|
|
active_scene = await ss.active_scene()
|
|
click.echo(
|
|
f'Current active scene: {click.style(active_scene.name, fg="green")} '
|
|
f'{f"(ID: {active_scene.id})" if id else ""}'
|
|
)
|
|
conn.close()
|
|
|
|
async with create_task_group() as tg:
|
|
tg.start_soon(conn.background_processing)
|
|
tg.start_soon(_run)
|
|
|
|
|
|
@scene.command()
|
|
@click.option('--id', is_flag=True, help='Include scene IDs in the output.')
|
|
@click.argument('scene_name', type=str)
|
|
@click.option(
|
|
'--preview',
|
|
is_flag=True,
|
|
help='Switch the preview scene only.',
|
|
)
|
|
@click.pass_context
|
|
async def switch(
|
|
ctx: click.Context, scene_name: str, preview: bool = False, id: bool = False
|
|
):
|
|
"""Switch to a scene by its name."""
|
|
conn = ctx.obj['connection']
|
|
ss = ScenesService(conn)
|
|
ts = TransitionsService(conn)
|
|
|
|
async def _run():
|
|
scenes = await ss.get_scenes()
|
|
for scene in scenes:
|
|
if scene.name == scene_name:
|
|
model = await ts.get_model()
|
|
|
|
if model.studio_mode:
|
|
await ss.make_scene_active(scene.id)
|
|
if preview:
|
|
click.echo(
|
|
f'Switched to preview scene: {click.style(scene.name, fg="blue")} '
|
|
f'{f"(ID: {scene.id})." if id else ""}'
|
|
)
|
|
else:
|
|
click.echo(
|
|
f'Switched to scene: {click.style(scene.name, fg="blue")} '
|
|
f'{f"(ID: {scene.id})." if id else ""}'
|
|
)
|
|
await ts.execute_studio_mode_transition()
|
|
click.echo(
|
|
'Executed studio mode transition to make the scene active.'
|
|
)
|
|
else:
|
|
if preview:
|
|
conn.close()
|
|
raise SlobsCliError(
|
|
'Cannot switch the preview scene in non-studio mode.'
|
|
)
|
|
|
|
await ss.make_scene_active(scene.id)
|
|
click.echo(
|
|
f'Switched to scene: {click.style(scene.name, fg="blue")} '
|
|
f'{f"(ID: {scene.id})." if id else ""}'
|
|
)
|
|
|
|
conn.close()
|
|
break
|
|
else: # If no scene by the given name was found
|
|
conn.close()
|
|
raise SlobsCliError(f"Scene '{scene_name}' not found.")
|
|
|
|
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
|