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 from GtkHelper.GtkHelper import ComboRow import os import threading # Import gtk modules import gi gi.require_version("Gtk", "4.0") gi.require_version("Adw", "1") from gi.repository import Gtk, Adw class ToggleSceneItemEnabled(OBSActionBase): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.current_state = -1 def on_ready(self): self.current_state = -1 # Connect to obs if not connected if self.plugin_base.backend is not None: if not self.plugin_base.get_connected(): self.reconnect_obs() # Show current scene item status threading.Thread(target=self.show_current_scene_item_status, daemon=True, name="show_current_scene_item_status").start() def show_current_scene_item_status(self, new_paused = False): if self.plugin_base.backend is None: self.current_state = -1 self.show_error() self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return if not self.plugin_base.backend.get_connected(): self.current_state = -1 self.show_error() self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return if not self.get_settings().get("item"): self.current_state = -1 self.show_error() self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return status = self.plugin_base.backend.get_scene_item_enabled(self.get_settings().get("scene"), self.get_settings().get("item")) if status is None: self.current_state = -1 self.show_error() self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return if status["enabled"]: self.show_for_state(1) else: self.show_for_state(0) def show_for_state(self, state: int): """ 0: Item disabled 1: Item enabled """ if state == self.current_state: return self.current_state = state image = "scene_item_disabled.png" if state == 0: image = "scene_item_disabled.png" elif state == 1: image = "scene_item_enabled.png" self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", image), size=0.75) def get_config_rows(self) -> list: super_rows = super().get_config_rows() self.scene_model = Gtk.StringList() self.scene_row = Adw.ComboRow(model=self.scene_model, title=self.plugin_base.lm.get("actions.switch.scene-row.label")) self.item_model = Gtk.StringList() self.item_row = Adw.ComboRow(model=self.item_model, title=self.plugin_base.lm.get("actions.toggle-scene-item-enabled-row.label")) self.connect_signals() self.load_item_model() self.load_configs() super_rows.append(self.scene_row) super_rows.append(self.item_row) return super_rows def connect_signals(self): self.scene_row.connect("notify::selected", self.on_scene_selected) self.item_row.connect("notify::selected", self.on_item_selected) def disconnect_signals(self): try: self.scene_row.disconnect_by_func(self.on_scene_selected) self.item_row.disconnect_by_func(self.on_item_selected) except TypeError as e: pass def load_item_model(self): self.disconnect_signals() # Clear model while self.scene_model.get_n_items() > 0: self.scene_model.remove(0) while self.item_model.get_n_items() > 0: self.item_model.remove(0) # Load model if self.plugin_base.backend.get_connected(): scenes = self.plugin_base.backend.get_scene_names() if scenes is None: return for scene in scenes: self.scene_model.append(scene) # Ensure selection is made if there's only one scene if len(scenes) == 1: self.get_settings()["scene"] = scenes[0] self.scene_row.set_selected(0) self.load_items_for_scene(scenes[0]) self.connect_signals() def load_items_for_scene(self, scene_name): # Clear items model while self.item_model.get_n_items() > 0: self.item_model.remove(0) if self.plugin_base.backend.get_connected(): items = self.plugin_base.backend.get_scene_items(scene_name) if items is None: self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return for item in items: self.item_model.append(item) # Ensure selection is made if there's only one item if len(items) == 1: self.item_row.set_selected(0) def load_configs(self): self.load_selected_device() def load_selected_device(self): self.disconnect_signals() settings = self.get_settings() for i, scene_name in enumerate(self.scene_model): if scene_name.get_string() == settings.get("scene"): self.scene_row.set_selected(i) self.load_items_for_scene(scene_name.get_string()) for j, item_name in enumerate(self.item_model): if item_name.get_string() == settings.get("item"): self.item_row.set_selected(j) break self.connect_signals() return self.scene_row.set_selected(Gtk.INVALID_LIST_POSITION) self.item_row.set_selected(Gtk.INVALID_LIST_POSITION) self.connect_signals() def on_scene_selected(self, *args): settings = self.get_settings() selected_index_scene = self.scene_row.get_selected() if selected_index_scene != Gtk.INVALID_LIST_POSITION: scene_name = self.scene_model[selected_index_scene].get_string() settings["scene"] = scene_name self.set_settings(settings) self.load_items_for_scene(scene_name) def on_item_selected(self, *args): settings = self.get_settings() selected_index_item = self.item_row.get_selected() if selected_index_item != Gtk.INVALID_LIST_POSITION: settings["item"] = self.item_model[selected_index_item].get_string() self.set_settings(settings) def on_key_down(self): if self.plugin_base.backend is None: self.current_state = -1 self.show_error() self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return if not self.plugin_base.backend.get_connected(): self.current_state = -1 self.show_error() self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return scene_name = self.get_settings().get("scene") item_name = self.get_settings().get("item") if scene_name in [None, ""]: self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return if item_name in [None, ""]: self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) return if self.current_state == 0: self.plugin_base.backend.set_scene_item_enabled(scene_name, item_name, True) else: self.plugin_base.backend.set_scene_item_enabled(scene_name, item_name, False) self.on_tick() def on_tick(self): self.show_current_scene_item_status() def reconnect_obs(self): super().reconnect_obs() if hasattr(self, "scene_model") and hasattr(self, "item_model"): self.load_item_model() self.load_configs()