obs.py

from fastapi import logger as _logger
from obswebsocket import obsws
from obswebsocket import requests
from obswebsocket.exceptions import ConnectionFailure
from websocket._exceptions import WebSocketConnectionClosedException

from settings import OBS_Settings

logger = _logger.logger
logger.setLevel(20)


class OBS:
    def __init__(self):
        self.enabled = OBS_Settings.enabled
        self.__connected = False
        if self.enabled:
            self.client = obsws(OBS_Settings.ip, OBS_Settings.ws_port, OBS_Settings.secret)

    @property
    def connected(self) -> bool:
        if not self.__connected:
            try:
                self.client.connect()
                self.__connected = True
                logger.info("OBS Connected")
            except ConnectionFailure:
                self.__connected = False
                logger.warning("OBS failed to connect.")
            except Exception as e:
                print(type(e), e)

        return self.__connected

    def send(self, enabled: bool, scene: str, source: str) -> None:
        """
        Sends a command to OBS to show or hide the source if connected
        Returns None
        """

        if self.connected:
            try:
                self.client.call(requests.SetSceneItemRender(enabled, source, scene))
            except WebSocketConnectionClosedException:
                self.client.disconnect()
                self.__connected = False
            except Exception as e:
                print(type(e), e)