XiguaLiveDanmakuHelper/liveDownloader.py

import time
from datetime import datetime
import threading
import Common
import os
import requests

session = requests.session()


def download():
    while Common.api.isLive and not Common.forceNotDownload:
        if not Common.streamUrl:
            Common.appendError("Download with No StreamUrl Specific")
            break
        path = datetime.strftime(datetime.now(), "%Y%m%d_%H%M.flv")
        base_path = Common.config["path"]
        if not os.path.isdir(base_path):
            os.makedirs(base_path)
        try:
            p = session.get(Common.streamUrl, stream=True, timeout=3)
            p.raise_for_status()
        except Exception as e:
            Common.appendError("Download >{}< with Exception [{}]".format(path,e.__str__()))
            break
        Common.api.initSave(os.path.join(base_path, path)+".xml")
        Common.appendDownloadStatus("Download >{}< Start".format(path))
        f = open(os.path.join(base_path, path), "wb")
        _size = 0
        try:
            for T in p.iter_content(chunk_size=Common.config["c_s"]):
                if Common.forceNotDownload:
                    Common.modifyLastDownloadStatus("Force Stop Download".format(path))
                    return
                f.write(T)
                _size += len(T)
                Common.modifyLastDownloadStatus(
                    "Downloading >{}< @ {:.2f}%".format(path, 100.0 * _size / Common.config["p_s"]))
                if _size > Common.config["p_s"] and not Common.config["dlO"]:
                    Common.modifyLastDownloadStatus("Download >{}< Exceed MaxSize".format(path))
                    break
            Common.modifyLastDownloadStatus("Download >{}< Finished".format(path))
        except Exception as e:
            Common.appendError("Download >{}< With Exception {}".format(path, e.__str__()))
            Common.api.updRoomInfo(True)
        finally:
            f.close()
        if os.path.getsize(path) < Common.config["i_s"]:
            Common.modifyLastDownloadStatus("Downloaded File >{}< is too small, will ignore it".format(path))
        else:
            Common.encodeQueue.put(path)
        Common.doUpdatePlaylist()
    Common.api.updRoomInfo(True)


def encode():
    Common.appendEncodeStatus("Encode Daemon Starting")
    while True:
        i = Common.encodeQueue.get()
        Common.encodeVideo(i)


def upload():
    date = datetime.strftime(datetime.now(), Common.config["sdf"])
    Common.appendUploadStatus("Upload Daemon Starting")
    i = Common.uploadQueue.get()
    while True:
        if i is True:
            Common.publishVideo(date)
            break
        try:
            Common.uploadVideo(i)
        except Exception as e:
            Common.appendError(e.__str__())
            continue
        finally:
            time.sleep(90)
        i = Common.uploadQueue.get()
    Common.appendUploadStatus("Upload Daemon Quiting")


t = threading.Thread(target=download, args=())
ut = threading.Thread(target=upload, args=())
et = threading.Thread(target=encode, args=())


def awakeEncode():
    global et
    if et.is_alive():
        return True
    et = threading.Thread(target=encode, args=())
    et.setDaemon(True)
    et.start()
    return False


def awakeDownload():
    global t
    if t.is_alive():
        return True
    t = threading.Thread(target=download, args=())
    t.setDaemon(True)
    t.start()
    Common.api.updRoomInfo()
    return False


def awakeUpload():
    global ut
    if ut.is_alive():
        return True
    ut = threading.Thread(target=upload, args=())
    ut.setDaemon(True)
    ut.start()
    return False


def run():
    Common.refreshDownloader()
    if not Common.api.isValidUser:
        Common.appendError("[{}]用户未找到".format(Common.api.name))
        return
    while True:
        if Common.api.isLive and not Common.forceNotBroadcasting:
            if not Common.forceNotDownload:
                awakeDownload()
            if not Common.forceNotUpload:
                awakeUpload()
            if not Common.forceNotEncode:
                awakeEncode()
            try:
                Common.api.getDanmaku()
            except Exception as e:
                Common.appendError(e.__str__())
            finally:
                time.sleep(1)
        else:
            try:
                Common.api.updRoomInfo()
            except Exception as e:
                Common.appendError(e.__str__())
                Common.refreshDownloader()
            if not Common.api.broadcaster:
                Common.refreshDownloader()
            if Common.forceStartEncodeThread:
                awakeEncode()
                Common.forceStartEncodeThread = False
            if Common.forceStartUploadThread:
                awakeUpload()
                Common.forceStartUploadThread = False
            if Common.doDelay():
                Common.uploadQueue.put(True)
            time.sleep(5)