You've already forked my-video-workflow
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
import threading
|
|
from queue import SimpleQueue as Queue
|
|
from time import sleep
|
|
|
|
from model.Workflow import Workflow
|
|
from workflow.video import quick_split_video
|
|
from workflow.worker import do_workflow
|
|
|
|
from . import LOGGER
|
|
|
|
|
|
IS_LIVING = threading.Event()
|
|
IS_ENCODING = threading.Event()
|
|
IS_UPLOADING = threading.Event()
|
|
ENCODING_QUEUE: "Queue[Workflow]" = Queue()
|
|
|
|
|
|
class Bilibili(threading.Thread):
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self.parts = []
|
|
|
|
def run(self) -> None:
|
|
while True:
|
|
if ENCODING_QUEUE.empty():
|
|
sleep(5)
|
|
if len(self.parts) > 0 and not IS_UPLOADING.is_set():
|
|
self.do_upload()
|
|
else:
|
|
workflow_item = ENCODING_QUEUE.get()
|
|
LOGGER.info("收到工作流请求：ID:【{}】".format(workflow_item.id))
|
|
for video_clip in workflow_item.video_clips:
|
|
IS_ENCODING.set()
|
|
try:
|
|
LOGGER.info("工作流视频：ID：【{}】，路径：【{}】".format(video_clip.id, video_clip.full_path))
|
|
if len(video_clip.danmaku_clips) < 1:
|
|
_parts = quick_split_video(video_clip.full_path)
|
|
else:
|
|
_parts = do_workflow(video_clip.full_path, video_clip.danmaku_clips[0].full_path)
|
|
LOGGER.info("工作流视频压制完成：结果：【{}】".format(_parts))
|
|
for _part in _parts:
|
|
self.parts.append(_part)
|
|
except:
|
|
LOGGER.error("压制异常！工作流视频：ID：【{}】，路径：【{}】".format(video_clip.id, video_clip.full_path))
|
|
finally:
|
|
IS_ENCODING.clear()
|
|
|
|
def do_upload(self):
|
|
LOGGER.info("尝试投稿：内容【{}】".format(self.parts))
|
|
self.clear()
|
|
|
|
def clear(self):
|
|
self.parts = []
|
|
|
|
|
|
|
|
INSTANCE = Bilibili()
|