119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
import os
|
|
from datetime import datetime
|
|
from hashlib import md5
|
|
from typing import Union, Optional
|
|
|
|
from config import VIDEO_TITLE
|
|
from entity.DanmakuFile import DanmakuFile
|
|
from entity.VideoClip import VideoClip
|
|
from entity.VideoPart import VideoPart
|
|
|
|
|
|
class WorkflowItem(object):
|
|
automatic: bool
|
|
finished: bool
|
|
conflict: bool
|
|
editing: bool
|
|
error: bool
|
|
delay_time: int
|
|
mode: int
|
|
videoParts: list[VideoPart]
|
|
videos: list[VideoClip]
|
|
|
|
def __init__(self):
|
|
self.automatic = False
|
|
self.finished = False
|
|
self.conflict = False
|
|
self.editing = False
|
|
self.error = False
|
|
self.title = VIDEO_TITLE
|
|
self.create_time = datetime.now()
|
|
self.delay_time = 0
|
|
self.mode = WorkflowModeEnum.MANUAL
|
|
self.danmakus = []
|
|
self.subtitleFiles = []
|
|
self.videos = []
|
|
self.videoParts = []
|
|
|
|
@property
|
|
def id(self):
|
|
return md5(self.name.encode("utf-8")).hexdigest()
|
|
|
|
@property
|
|
def name(self):
|
|
return self.title.format(self.create_time.strftime("%Y%m%d"))
|
|
|
|
def mode_set(self, mode: int):
|
|
self.mode = mode
|
|
|
|
def mode_add(self, mode: int):
|
|
self.mode = self.mode | mode
|
|
|
|
def mode_del(self, mode: int):
|
|
self.mode = self.mode ^ mode
|
|
|
|
def mode_has(self, mode: int) -> bool:
|
|
return self.mode & mode == mode
|
|
|
|
def mode_add_merge(self):
|
|
return self.mode_add(WorkflowModeEnum.MERGE)
|
|
|
|
def mode_del_merge(self):
|
|
return self.mode_del(WorkflowModeEnum.MERGE)
|
|
|
|
def mode_has_merge(self) -> bool:
|
|
return self.mode_has(WorkflowModeEnum.MERGE)
|
|
|
|
def mode_add_danmaku(self):
|
|
return self.mode_add(WorkflowModeEnum.DANMAKU)
|
|
|
|
def mode_del_danmaku(self):
|
|
return self.mode_del(WorkflowModeEnum.DANMAKU)
|
|
|
|
def mode_has_danmaku(self) -> bool:
|
|
return self.mode_has(WorkflowModeEnum.DANMAKU)
|
|
|
|
def mode_add_encode(self):
|
|
return self.mode_add(WorkflowModeEnum.ENCODE)
|
|
|
|
def mode_del_encode(self):
|
|
return self.mode_del(WorkflowModeEnum.ENCODE)
|
|
|
|
def mode_has_encode(self) -> bool:
|
|
return self.mode_has(WorkflowModeEnum.ENCODE)
|
|
|
|
def mode_add_split(self):
|
|
return self.mode_add(WorkflowModeEnum.SPLIT)
|
|
|
|
def mode_del_split(self):
|
|
return self.mode_del(WorkflowModeEnum.SPLIT)
|
|
|
|
def mode_has_split(self) -> bool:
|
|
return self.mode_has(WorkflowModeEnum.SPLIT)
|
|
|
|
def add_video(self, video_file: Union[VideoClip, os.PathLike[str], str], base_path: Optional[Union[os.PathLike[str], str]] = None) -> VideoClip:
|
|
if isinstance(video_file, VideoClip):
|
|
self.videos.append(video_file)
|
|
else:
|
|
video_file = VideoClip(video_file, base_path)
|
|
self.videos.append(video_file)
|
|
return video_file
|
|
|
|
def add_danmaku(self, danmaku_file: Union[DanmakuFile, os.PathLike[str], str], base_path: Optional[Union[os.PathLike[str], str]] = None) -> DanmakuFile:
|
|
if isinstance(danmaku_file, DanmakuFile):
|
|
self.danmakus.append(danmaku_file)
|
|
else:
|
|
danmaku_file = DanmakuFile(danmaku_file, base_path)
|
|
self.danmakus.append(danmaku_file)
|
|
return danmaku_file
|
|
|
|
|
|
class WorkflowModeEnum:
|
|
MERGE = 1 << 4
|
|
DANMAKU = 1 << 3
|
|
ENCODE = 1 << 2
|
|
SPLIT = 1 << 1
|
|
DANMAKU_DUAL = DANMAKU | ENCODE | SPLIT
|
|
DANMAKU_SINGLE = DANMAKU | ENCODE | SPLIT
|
|
MANUAL = 0
|