XiguaLiveDanmakuHelper/liveDownloader.py

import sys
import time
from datetime import datetime
import dotenv
import m3u8
import queue
import threading

from api import XiGuaLiveApi
from bilibili import *

q = queue.Queue()
base_uri = ""
isUpload = False
env = dotenv.main.DotEnv(".env")
uq = queue.Queue()
d = datetime.strftime(datetime.now(),"%Y_%m_%d")

class downloader(XiGuaLiveApi):
    files = []
    playlist: str = None

    def updRoomInfo(self):
        super(downloader, self).updRoomInfo()
        self.updPlayList()

    def updPlayList(self):
        if "playInfo" not in self._rawRoomInfo or "Main" not in self._rawRoomInfo["playInfo"]:
            if self.playlist is None:
                self.apiChangedError("无法获取直播链接")
                self.playlist = False
        else:
            self.playlist = self._rawRoomInfo["playInfo"]["Main"]["1"]["Url"]["HlsUrl"]

    def onLike(self, user):
        pass
    def onAd(self, i):
        pass
    def onChat(self, chat):
        pass
    def onEnter(self, msg):
        pass
    def onJoin(self, user):
        pass
    def onLeave(self, json):
        self.updRoomInfo()
    def onMessage(self, msg):
        pass
    def onPresent(self, gift):
        pass
    def onPresentEnd(self, gift):
        pass
    def onSubscribe(self, user):
        pass
    def preDownload(self):
        global base_uri
        if self.playlist:
            try:
                p = m3u8.load(self.playlist)
            except:
                self.updRoomInfo()
                return
            base_uri = p.base_uri
            for i in p.files:
                if i not in self.files:
                    self.files.append(i)
                    print("{} : Add Sequence {}".format(datetime.strftime(datetime.now(), "%y%m%d %H%M"),
                                                         len(self.files)))
                    q.put(i)
        self.genNewName()
    def genNewName(self):
        if len(self.files) > 1000:
            q.put(True)
            self.files.clear()


def download(path=datetime.strftime(datetime.now(),"%Y%m%d_%H%M.ts")):
    print("{} : Download Daemon Starting".format(datetime.strftime(datetime.now(), "%y%m%d %H%M")))
    n = False
    i = q.get()
    while True:
        if isinstance(i, bool):
            print("{} : Download Daemon Receive Command {}".format(datetime.strftime(datetime.now(), "%y%m%d %H%M"), i))
            break
        print("{} : Download {}".format(datetime.strftime(datetime.now(), "%y%m%d %H%M"), i))
        try:
            _p = requests.get("{}{}".format(base_uri,i))
        except:
            continue
        f = open(path, "ab")
        f.write(_p.content)
        f.close()
        n=True
        i = q.get()
    if n:
        uq.put(path)
    print("{} : Download Daemon Quiting".format(datetime.strftime(datetime.now(), "%y%m%d %H%M")))


def upload(date = datetime.strftime(datetime.now(), "%Y_%m_%d")):
    print("{} : Upload Daemon Starting".format(datetime.strftime(datetime.now(), "%y%m%d %H%M")))
    i = uq.get()
    while True:
        if isinstance(i, bool):
            if i is True:
                b.finishUpload("【永恒de草薙直播录播】直播于 {} 自动投递实际测试".format(date),
                                 17, ["永恒de草薙", "三国", "三国战记", "自动投递", "直播", "录播"],
                                 "自动投递实际测试\n原主播：永恒de草薙\n直播时间：晚上6点多到白天6点左右",
                                 source= "https://live.ixigua.com/userlive/97621754276", no_reprint= 0)
            print("{} : Upload Daemon Receive Command {}".format(datetime.strftime(datetime.now(), "%y%m%d %H%M"), i))
            break
        print("{} : Upload {}".format(datetime.strftime(datetime.now(), "%y%m%d %H%M"), i))
        try:
            b.preUpload(VideoPart(i, i))
        except:
            continue
        i = uq.get()
    print("{} : Upload Daemon Quiting".format(datetime.strftime(datetime.now(), "%y%m%d %H%M")))


b = Bilibili()
b.login(env.get("b_u"), env.get("b_p"))

if __name__ == "__main__":
    room = 97621754276  # 永恒
    # room = 75366565294
    # room = 83940182312 #Dae
    # room = 5947850784 #⑦
    # room = 58649240617 #戏
    if len(sys.argv) > 1:
        try:
            room = int(sys.argv[1])
        except ValueError:
            pass
    print("西瓜直播录播助手 by JerryYan")
    api = downloader(room)
    print("进入", api.roomLiver, "的直播间")
    if not api.isValidRoom:
        input("房间不存在")
        sys.exit()
    print("=" * 30)
    _preT = datetime.strftime(datetime.now(), "%Y%m%d_%H%M.ts")
    t = threading.Thread(target=download, args=(_preT,))
    ut = threading.Thread(target=upload, args=(d,))
    while True:
        if api.isLive:
            if d is None:
                d = datetime.strftime(datetime.now(), "%Y_%m_%d")
            if not t.is_alive():
                _preT = datetime.strftime(datetime.now(), "%Y%m%d_%H%M.ts")
                t = threading.Thread(target=download, args=(_preT,))
                t.setDaemon(True)
                t.start()
            if not ut.is_alive():
                ut = threading.Thread(target=upload, args=(d,))
                ut.setDaemon(True)
                ut.start()
            api.preDownload()
            isUpload = True
            time.sleep(3)
        else:
            q.put(False)
            if isUpload:
                print("自动投稿中，请稍后")
                uq.put(True)
            else:
                pass
                # print("主播未开播，等待1分钟后重试")
            time.sleep(60)
            d=None
            api.updRoomInfo()