94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
|
# 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
|
||
|
|
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])
|