Upload New File

website/utils/stream.py

@@ -0,0 +1,71 @@
+import os
+import re
+import time
+from website.constants import SEGMENTS_DIR, SEGMENTS_M3U8, SEGMENT_INIT, STREAM_PIDFILE, STREAM_START, STREAM_TITLE
+
+RE_SEGMENT = re.compile(r'stream(?P<number>\d+).m4s')
+
+def _segment_number(fn):
+    if fn == SEGMENT_INIT: return None
+    return int(RE_SEGMENT.fullmatch(fn).group('number'))
+
+def _is_segment(fn):
+    return bool(RE_SEGMENT.fullmatch(fn))
+
+def current_segment():
+    try:
+        files = os.listdir(SEGMENTS_DIR)
+    except FileNotFoundError:
+        return None
+
+    try:
+        m3u8 = open(SEGMENTS_M3U8).read()
+    except FileNotFoundError:
+        return None
+
+    files = filter(lambda fn: fn in m3u8, files)
+
+    try:
+        last_segment = max(filter(_is_segment, files), key=_segment_number)
+        return _segment_number(last_segment)
+    except ValueError:
+        return None
+
+def is_online():
+    # If the pidfile doesn't exist, return False
+    try:
+        pid = open(STREAM_PIDFILE).read()
+        pid = int(pid)
+    except (FileNotFoundError, ValueError):
+        return False
+
+    # If the process ID doesn't exist, return False
+    try:
+        os.kill(pid, 0)
+    except OSError:
+        return False
+
+    # Otherwise return True
+    return True
+
+def get_title():
+    try:
+        return open(STREAM_TITLE).read().strip()
+    except FileNotFoundError:
+        return ''
+
+def get_start(absolute=True, relative=False):
+    try:
+        start = open(STREAM_START).read()
+        start = int(start)
+    except (FileNotFoundError, ValueError):
+        start = None
+
+    diff = None if start == None else int(time.time()) - start
+
+    if absolute and relative:
+        return start, diff
+    elif absolute:
+        return start
+    elif relative:
+        return diff