94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
import multiprocessing
|
|
import engine
|
|
import common
|
|
from engine import *
|
|
from engine.downloader import download, check
|
|
from common import logger
|
|
from common.event import Event
|
|
from engine.plugins.base_adapter import UploadBase
|
|
from engine.uploader import upload
|
|
|
|
# 初始化事件管理器
|
|
event_manager = common.event.EventManager(context)
|
|
|
|
|
|
@event_manager.register(DOWNLOAD, block=True)
|
|
def process(name, url):
|
|
try:
|
|
data = {"url": url, "date": common.time_now()}
|
|
p = multiprocessing.Process(target=download, args=(name, url))
|
|
p.start()
|
|
p.join()
|
|
# download(name, url)
|
|
upload("bili_web", name, data)
|
|
finally:
|
|
return Event(BE_MODIFIED, args=(url,))
|
|
|
|
|
|
@event_manager.register(UPLOAD, block=True)
|
|
def process_upload(name, url):
|
|
try:
|
|
data = {"url": url, "date": common.time_now()}
|
|
upload("bili_web", name, data)
|
|
finally:
|
|
return Event(BE_MODIFIED, args=(url,))
|
|
|
|
|
|
@event_manager.server()
|
|
class KernelFunc:
|
|
def __init__(self, urls, url_status: dict):
|
|
self.urls = urls
|
|
self.url_status = url_status
|
|
self.__raw_streamer_status = url_status.copy()
|
|
|
|
@event_manager.register(CHECK, block=True)
|
|
def batch_check(self):
|
|
live = check(self.urls, "batch")
|
|
return Event(CHECK_UPLOAD, args=(live,)), Event(TO_MODIFY, args=(live,))
|
|
|
|
@event_manager.register(CHECK, block=True)
|
|
def singleton_check(self):
|
|
live = check(self.urls, "single")
|
|
return Event(TO_MODIFY, args=(live,))
|
|
|
|
@event_manager.register(TO_MODIFY)
|
|
def modify(self, live_m):
|
|
if not live_m:
|
|
return logger.debug('无人直播')
|
|
live_d = {}
|
|
for live in live_m:
|
|
if self.url_status[live] == 1:
|
|
logger.debug('已开播正在下载')
|
|
else:
|
|
name = inverted_index[live]
|
|
logger.debug(f'{name}刚刚开播，去下载')
|
|
event_manager.send_event(Event(DOWNLOAD, args=(name, live)))
|
|
|
|
live_d[live] = 1
|
|
self.url_status.update(live_d)
|
|
# self.url_status = {**self.__raw_streamer_status, **live_d}
|
|
|
|
def free(self, list_url):
|
|
status_num = list(map(lambda x: self.url_status.get(x), list_url))
|
|
# if 1 in status_num or 2 in status_num:
|
|
# return False
|
|
# else:
|
|
# return True
|
|
return not (1 in status_num or 2 in status_num)
|
|
|
|
@event_manager.register(CHECK_UPLOAD)
|
|
def free_upload(self, _urls):
|
|
logger.debug(_urls)
|
|
for title, v in engine.streamer_url.items():
|
|
url = v[0]
|
|
if self.free(v) and UploadBase.filter_file(title):
|
|
event_manager.send_event(Event(UPLOAD, args=(title, url)))
|
|
self.url_status[url] = 2
|
|
|
|
@event_manager.register(BE_MODIFIED)
|
|
def revise(self, url):
|
|
if url:
|
|
# 更新字典
|
|
# url_status = {**url_status, **{url: 0}}
|
|
self.url_status.update({url: 0})
|