OBSPlugin/actions/ToggleRecord/ToggleRecord.py

74 lines
2.6 KiB
Python

# from ...OBSActionBase import OBSActionBase
from plugins.dev_core447_OBSPlugin.OBSActionBase import OBSActionBase
import os
class ToggleRecord(OBSActionBase):
ACTION_NAME = "Toggle Record"
CONTROLS_KEY_IMAGE = True
def __init__(self, deck_controller, page, coords):
super().__init__(deck_controller=deck_controller, page=page, coords=coords)
def on_ready(self):
# Connect to obs if not connected
if not self.PLUGIN_BASE.backend.get_connected(): # self.PLUGIN_BASE.obs.connect_to(host="localhost", port=4444, timeout=3, legacy=False)
self.reconnect_obs()
# Show current rec status
self.show_current_rec_status()
def show_current_rec_status(self, new_paused = False):
if not self.PLUGIN_BASE.backend.get_connected():
self.set_key(media_path=os.path.join(self.PLUGIN_BASE.PATH, "assets", "error.png"))
return
status = self.PLUGIN_BASE.backend.get_record_status()
if status is None:
self.set_key(media_path=os.path.join(self.PLUGIN_BASE.PATH, "assets", "error.png"))
return
if status["paused"]:
self.show_for_state(2)
elif status["active"]:
self.show_for_state(1)
else:
self.show_for_state(0)
def show_for_state(self, state: int):
"""
0: Not Recording
1: Recording
2: Paused
3: Stopping in progress
"""
image = "record_inactive.png"
if state == 0:
self.set_bottom_label(None)
image = "record_inactive.png"
elif state == 1:
self.show_rec_time()
image = "record_active.png"
print("active")
elif state == 2:
self.show_rec_time()
image = "record_resume.png"
self.set_key(media_path=os.path.join(self.PLUGIN_BASE.PATH, "assets", image))
def on_key_down(self):
if not self.PLUGIN_BASE.backend.get_connected():
return
self.PLUGIN_BASE.backend.toggle_record()
def on_tick(self):
self.show_current_rec_status()
def show_rec_time(self):
if not self.PLUGIN_BASE.backend.get_connected():
self.set_key(media_path=os.path.join(self.PLUGIN_BASE.PATH, "assets", "error.png"))
return
status = self.PLUGIN_BASE.backend.get_record_status()
if status is None:
self.set_key(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], font_size=16)