mirror of
https://github.com/onyx-and-iris/obsws-cli.git
synced 2025-05-16 14:30:24 +01:00
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
"""module containing commands for controlling OBS scenes."""
|
|
|
|
from typing import Annotated
|
|
|
|
import obsws_python as obsws
|
|
import typer
|
|
|
|
from .alias import AliasGroup
|
|
from .errors import ObswsCliBadParameter, ObswsCliError
|
|
|
|
app = typer.Typer(cls=AliasGroup)
|
|
|
|
|
|
@app.callback()
|
|
def main():
|
|
"""Control OBS scenes."""
|
|
|
|
|
|
@app.command('list | ls')
|
|
def list(ctx: typer.Context):
|
|
"""List all scenes."""
|
|
resp = ctx.obj['obsws'].get_scene_list()
|
|
scenes = (scene.get('sceneName') for scene in reversed(resp.scenes))
|
|
typer.echo('\n'.join(scenes))
|
|
|
|
|
|
@app.command('current | get')
|
|
def current(
|
|
ctx: typer.Context,
|
|
preview: Annotated[
|
|
bool, typer.Option(help='Get the preview scene instead of the program scene')
|
|
] = False,
|
|
):
|
|
"""Get the current program scene or preview scene."""
|
|
try:
|
|
if preview:
|
|
resp = ctx.obj['obsws'].get_current_preview_scene()
|
|
typer.echo(resp.current_preview_scene_name)
|
|
else:
|
|
resp = ctx.obj['obsws'].get_current_program_scene()
|
|
typer.echo(resp.current_program_scene_name)
|
|
except obsws.error.OBSSDKRequestError as e:
|
|
match e.code:
|
|
case 506:
|
|
raise ObswsCliError(
|
|
'Not in studio mode, cannot get preview scene.'
|
|
) from e
|
|
raise
|
|
|
|
|
|
@app.command('switch | set')
|
|
def switch(
|
|
ctx: typer.Context,
|
|
scene_name: str,
|
|
preview: Annotated[
|
|
bool,
|
|
typer.Option(help='Switch to the preview scene instead of the program scene'),
|
|
] = False,
|
|
):
|
|
"""Switch to a scene."""
|
|
try:
|
|
if preview:
|
|
ctx.obj['obsws'].set_current_preview_scene(scene_name)
|
|
else:
|
|
ctx.obj['obsws'].set_current_program_scene(scene_name)
|
|
except obsws.error.OBSSDKRequestError as e:
|
|
match e.code:
|
|
case 506:
|
|
raise ObswsCliError(
|
|
'Not in studio mode, cannot set preview scene.'
|
|
) from e
|
|
case 600:
|
|
raise ObswsCliBadParameter(f"Scene '{scene_name}' not found.") from e
|
|
raise
|