# from ...OBSActionbase import OBSActionBase import threading from plugins.com_core447_OBSPlugin.OBSActionBase import OBSActionBase from src.backend.DeckManagement.DeckController import DeckController from src.backend.PageManagement.Page import Page from src.backend.PluginManager.PluginBase import PluginBase import os class ToggleStream(OBSActionBase): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.current_state = -1 def on_ready(self): self.current_state = -1 if self.plugin_base.backend is None: if not self.plugin_base.get_connected(): self.reconnect_obs() threading.Thread(target=self.show_current_stream_status, daemon=True, name="show_current_stream_status").start() def show_current_stream_status(self, new_paused = False): if self.plugin_base.backend is None: self.current_state = -1 self.show_error() return if not self.plugin_base.backend.get_connected(): self.current_state = -1 self.show_error() return status = self.plugin_base.backend.get_stream_status() if status is None: self.current_state = -1 self.show_error() return self.hide_error() if status["active"]: if status["reconnecting"]: self.show_for_state(2) else: self.show_for_state(1) else: self.show_for_state(0) def show_for_state(self, state: int): """ 0: Not Streaming 1: Streaming Connected 2: Streaming Reconnecting """ if state in [1, 2]: self.show_stream_time() if state == self.current_state: return self.current_state = state image = "stream_inactive.png" if state == 0: self.set_bottom_label(None) elif state == 1: image = "stream_active.png" elif state == 2: image = "stream_reconnecting.png" self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", image)) def on_key_down(self): if self.plugin_base.backend is None: self.current_state = -1 self.show_error() return if not self.plugin_base.backend.get_connected(): self.current_state = -1 self.show_error() return self.plugin_base.backend.toggle_stream() self.on_tick() def on_tick(self): self.show_current_stream_status() def show_stream_time(self): if not self.plugin_base.backend.get_connected(): self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return status = self.plugin_base.backend.get_stream_status() if status is None: self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return if not status["active"]: self.set_bottom_label(None) return self.set_bottom_label(status["timecode"][:-4])