52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
from threading import Thread
|
|
import yaml
|
|
|
|
|
|
with open("mediamtx.yml", "r") as config_file:
|
|
MEDIASERVER_CONFIG = yaml.safe_load(config_file)
|
|
|
|
|
|
def is_alive(subprocess):
|
|
return True if (subprocess and subprocess.poll() is None) else False
|
|
|
|
|
|
class Streamer:
|
|
PROTOCOL = "rtsp"
|
|
REST_PATH = "stream"
|
|
|
|
def __init__(self):
|
|
self.run = False
|
|
self.thread = None
|
|
self.name = None
|
|
|
|
def stream_path(self):
|
|
return f"{MEDIASERVER_CONFIG['rtspAddress']}/{type(self).REST_PATH}/{self.name}"
|
|
|
|
def stream_address(self, host):
|
|
return f"{Streamer.PROTOCOL}://{host}{self.stream_path()}"
|
|
|
|
def is_streaming(self):
|
|
return True if (self.thread and self.thread.is_alive()) else False
|
|
|
|
def start_stream(self):
|
|
if self.is_streaming():
|
|
raise RuntimeError("Stream thread is already running")
|
|
|
|
self.run = True
|
|
self.thread = Thread(target=self._stream_thread, daemon=True, args=())
|
|
self.thread.start()
|
|
|
|
def end_stream(self):
|
|
if self.thread is None:
|
|
raise RuntimeError("No stream thread to terminate")
|
|
|
|
self.run = False
|
|
self.thread.join()
|
|
self.thread = None
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from pprint import pprint
|
|
|
|
pprint(MEDIASERVER_CONFIG)
|