144 lines
5.1 KiB
Python
144 lines
5.1 KiB
Python
import threading
|
||
from datetime import datetime
|
||
|
||
from . import LOGGER
|
||
from .bilibiliupload import core, VideoPart
|
||
|
||
|
||
IS_LIVING = threading.Event()
|
||
IS_ENCODING = threading.Event()
|
||
IS_UPLOADING = threading.Event()
|
||
|
||
|
||
class Bilibili:
|
||
def __init__(self):
|
||
self.access_token = ""
|
||
self.session_id = ""
|
||
self.user_id = ""
|
||
self.expires = 0
|
||
self.login_time = None
|
||
self.parts = []
|
||
|
||
def login(self):
|
||
with open("access_token", "r") as f:
|
||
self.access_token = f.read(64).strip()
|
||
self.session_id, self.user_id, self.expires = core.login_by_access_token(self.access_token)
|
||
self.login_time = datetime.now()
|
||
LOGGER.info("B站登录，UID【{}】，过期时间【{}】".format(self.user_id, self.expires))
|
||
|
||
def upload(self,
|
||
parts,
|
||
title,
|
||
tid,
|
||
tag,
|
||
desc,
|
||
source='',
|
||
cover='',
|
||
no_reprint=1,
|
||
):
|
||
"""
|
||
|
||
:param parts: e.g. VideoPart('part path', 'part title', 'part desc'), or [VideoPart(...), VideoPart(...)]
|
||
:type parts: VideoPart or list<VideoPart>
|
||
:param title: video's title
|
||
:type title: str
|
||
:param tid: video type, see: https://member.bilibili.com/x/web/archive/pre
|
||
or https://github.com/uupers/BiliSpider/wiki/%E8%A7%86%E9%A2%91%E5%88%86%E5%8C%BA%E5%AF%B9%E5%BA%94%E8%A1%A8
|
||
:type tid: int
|
||
:param tag: video's tag
|
||
:type tag: list<str>
|
||
:param desc: video's description
|
||
:type desc: str
|
||
:param source: (optional) 转载地址
|
||
:type source: str
|
||
:param cover: (optional) cover's URL, use method *cover_up* to get
|
||
:type cover: str
|
||
:param no_reprint: (optional) 0=可以转载, 1=禁止转载(default)
|
||
:type no_reprint: int
|
||
"""
|
||
self.pre_upload(parts)
|
||
self.finish_upload(title, tid, tag, desc, source, cover, no_reprint)
|
||
self.clear()
|
||
|
||
def pre_upload(self, parts: "VideoPart", max_retry=5):
|
||
"""
|
||
:param max_retry:
|
||
:param parts: e.g. VideoPart('part path', 'part title', 'part desc'), or [VideoPart(...), VideoPart(...)]
|
||
:type parts: VideoPart or list<VideoPart>
|
||
"""
|
||
if not isinstance(parts, list):
|
||
parts = [parts]
|
||
|
||
IS_UPLOADING.set()
|
||
for part in parts:
|
||
if isinstance(part, str):
|
||
part = VideoPart(part)
|
||
LOGGER.info("Start Uploading >{}<".format(part.path))
|
||
status = core.upload_video_part(self.access_token, self.session_id, self.user_id, part, max_retry)
|
||
if status:
|
||
# 上传完毕
|
||
LOGGER.info("Upload >{}< Finished；【{}】".format(part.path, part.server_file_name))
|
||
self.parts.append(part)
|
||
else:
|
||
LOGGER.warn("Upload >{}< Failed".format(part.path))
|
||
IS_UPLOADING.clear()
|
||
|
||
def finish_upload(self,
|
||
title,
|
||
tid,
|
||
tag,
|
||
desc,
|
||
source='',
|
||
cover='',
|
||
no_reprint=1,
|
||
):
|
||
"""
|
||
:param title: video's title
|
||
:type title: str
|
||
:param tid: video type, see: https://member.bilibili.com/x/web/archive/pre
|
||
or https://github.com/uupers/BiliSpider/wiki/%E8%A7%86%E9%A2%91%E5%88%86%E5%8C%BA%E5%AF%B9%E5%BA%94%E8%A1%A8
|
||
:type tid: int
|
||
:param tag: video's tag
|
||
:type tag: str
|
||
:param desc: video's description
|
||
:type desc: str
|
||
:param source: (optional) 转载地址
|
||
:type source: str
|
||
:param cover: (optional) cover's URL, use method *cover_up* to get
|
||
:type cover: str
|
||
:param no_reprint: (optional) 0=可以转载, 1=禁止转载(default)
|
||
:type no_reprint: int
|
||
:param copyright: (optional) 0=转载的, 1=自制的(default)
|
||
:type copyright: int
|
||
"""
|
||
if len(self.parts) == 0:
|
||
return
|
||
if IS_ENCODING.is_set():
|
||
LOGGER.info("[{}]仍在压制，取消发布".format(title))
|
||
return
|
||
if IS_LIVING.is_set():
|
||
LOGGER.info("[{}]仍在直播，取消发布".format(title))
|
||
return
|
||
if IS_UPLOADING.is_set():
|
||
LOGGER.info("[{}]仍在上传，取消发布".format(title))
|
||
return
|
||
LOGGER.info("[{}]投稿中，请稍后".format(title))
|
||
copyright = 2 if source else 1
|
||
try:
|
||
avid, bvid = core.upload(self.access_token, self.session_id, self.user_id, self.parts, copyright,
|
||
title=title, tid=tid, tag=tag, desc=desc, source=source, cover=cover, no_reprint=no_reprint)
|
||
LOGGER.info("[{}]投稿成功；AVID【{}】，BVID【{}】".format(title, avid, bvid))
|
||
self.clear()
|
||
except Exception as e:
|
||
LOGGER.error("[{}]投稿失败".format(title), exc_info=e)
|
||
|
||
|
||
def reloadFromPrevious(self):
|
||
...
|
||
|
||
def clear(self):
|
||
self.parts = []
|
||
|
||
|
||
INSTANCE = Bilibili()
|