Improve show_current_rec_status

This commit is contained in:
Core447 2024-03-27 09:58:41 +01:00
parent 59f5ea0fec
commit 234c34e0da
No known key found for this signature in database
GPG Key ID: 5B97C0BD62F98516
1 changed files with 16 additions and 2 deletions

View File

@ -13,6 +13,7 @@ class ToggleRecord(OBSActionBase):
self.current_state = -1 self.current_state = -1
def on_ready(self): def on_ready(self):
self.current_state = -1
# Connect to obs if not connected # Connect to obs if not connected
if self.plugin_base.backend is not None: if self.plugin_base.backend is not None:
if not self.plugin_base.backend.get_connected(): # self.plugin_base.obs.connect_to(host="localhost", port=4444, timeout=3, legacy=False) if not self.plugin_base.backend.get_connected(): # self.plugin_base.obs.connect_to(host="localhost", port=4444, timeout=3, legacy=False)
@ -22,12 +23,18 @@ class ToggleRecord(OBSActionBase):
self.show_current_rec_status() self.show_current_rec_status()
def show_current_rec_status(self, new_paused = False): def show_current_rec_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(): if not self.plugin_base.backend.get_connected():
self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) self.current_state = -1
self.show_error()
return return
status = self.plugin_base.backend.get_record_status() status = self.plugin_base.backend.get_record_status()
if status is None: if status is None:
self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", "error.png")) self.current_state = -1
self.show_error()
return return
if status["paused"]: if status["paused"]:
self.show_for_state(2) self.show_for_state(2)
@ -61,11 +68,18 @@ class ToggleRecord(OBSActionBase):
self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", image)) self.set_media(media_path=os.path.join(self.plugin_base.PATH, "assets", image))
def on_key_down(self): 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(): if not self.plugin_base.backend.get_connected():
self.current_state = -1
self.show_error()
return return
self.plugin_base.backend.toggle_record() self.plugin_base.backend.toggle_record()
def on_tick(self): def on_tick(self):
return
self.show_current_rec_status() self.show_current_rec_status()
def show_rec_time(self): def show_rec_time(self):