mirror of
https://github.com/onyx-and-iris/obsws-cli.git
synced 2025-05-16 14:30:24 +01:00
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
"""module containing commands for manipulating scene collections."""
|
|
|
|
import typer
|
|
|
|
from .alias import AliasGroup
|
|
|
|
app = typer.Typer(cls=AliasGroup)
|
|
|
|
|
|
@app.callback()
|
|
def main():
|
|
"""Control scene collections in OBS."""
|
|
|
|
|
|
@app.command('list | ls')
|
|
def list(ctx: typer.Context):
|
|
"""List all scene collections."""
|
|
resp = ctx.obj['obsws'].get_scene_collection_list()
|
|
typer.echo('\n'.join(resp.scene_collections))
|
|
|
|
|
|
@app.command('current | get')
|
|
def current(ctx: typer.Context):
|
|
"""Get the current scene collection."""
|
|
resp = ctx.obj['obsws'].get_scene_collection_list()
|
|
typer.echo(resp.current_scene_collection_name)
|
|
|
|
|
|
def _scene_collection_in_scene_collections(
|
|
ctx: typer.Context, scene_collection_name: str
|
|
) -> bool:
|
|
"""Check if a scene collection exists."""
|
|
resp = ctx.obj['obsws'].get_scene_collection_list()
|
|
return any(
|
|
collection == scene_collection_name for collection in resp.scene_collections
|
|
)
|
|
|
|
|
|
@app.command('switch | set')
|
|
def switch(ctx: typer.Context, scene_collection_name: str):
|
|
"""Switch to a scene collection."""
|
|
if not _scene_collection_in_scene_collections(ctx, scene_collection_name):
|
|
typer.echo(f"Scene collection '{scene_collection_name}' not found.", err=True)
|
|
raise typer.Exit(code=1)
|
|
|
|
current_scene_collection = (
|
|
ctx.obj['obsws'].get_scene_collection_list().current_scene_collection_name
|
|
)
|
|
if scene_collection_name == current_scene_collection:
|
|
typer.echo(
|
|
f'Scene collection "{scene_collection_name}" is already active.', err=True
|
|
)
|
|
raise typer.Exit(code=1)
|
|
|
|
ctx.obj['obsws'].set_current_scene_collection(scene_collection_name)
|
|
typer.echo(f"Switched to scene collection '{scene_collection_name}'")
|
|
|
|
|
|
@app.command('create | new')
|
|
def create(ctx: typer.Context, scene_collection_name: str):
|
|
"""Create a new scene collection."""
|
|
if _scene_collection_in_scene_collections(ctx, scene_collection_name):
|
|
typer.echo(
|
|
f"Scene collection '{scene_collection_name}' already exists.", err=True
|
|
)
|
|
raise typer.Exit(code=1)
|
|
|
|
ctx.obj['obsws'].create_scene_collection(scene_collection_name)
|
|
typer.echo(f'Created scene collection {scene_collection_name}')
|