feat: add bilibili cli commands
This commit is contained in:
+106
@@ -9,6 +9,7 @@ from pathlib import Path
|
||||
from typing import Iterable, Sequence
|
||||
|
||||
from conf import BASE_DIR
|
||||
from uploader.bilibili_uploader.runtime import run_biliup_command
|
||||
from uploader.douyin_uploader.main import (
|
||||
DOUYIN_PUBLISH_STRATEGY_IMMEDIATE,
|
||||
DOUYIN_PUBLISH_STRATEGY_SCHEDULED,
|
||||
@@ -81,6 +82,17 @@ class KuaishouNoteUploadRequest:
|
||||
headless: bool = True
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
class BilibiliVideoUploadRequest:
|
||||
account_name: str
|
||||
video_file: Path
|
||||
title: str
|
||||
description: str
|
||||
tid: int
|
||||
tags: list[str]
|
||||
publish_date: datetime | int
|
||||
|
||||
|
||||
def resolve_runtime_home() -> Path:
|
||||
return Path(BASE_DIR)
|
||||
|
||||
@@ -137,6 +149,25 @@ async def check_kuaishou_account(account_name: str) -> bool:
|
||||
return await kuaishou_cookie_auth(str(account_file))
|
||||
|
||||
|
||||
async def login_bilibili_account(account_name: str) -> dict:
|
||||
account_file = resolve_account_file("bilibili", account_name)
|
||||
result = run_biliup_command(["-u", str(account_file), "login"])
|
||||
success = result.returncode == 0
|
||||
return {
|
||||
"success": success,
|
||||
"message": (result.stderr or result.stdout or "").strip() or "Bilibili login completed" if success else (result.stderr or result.stdout or "").strip() or "Bilibili login failed",
|
||||
"account_file": str(account_file),
|
||||
}
|
||||
|
||||
|
||||
async def check_bilibili_account(account_name: str) -> bool:
|
||||
account_file = resolve_account_file("bilibili", account_name)
|
||||
if not account_file.exists():
|
||||
return False
|
||||
result = run_biliup_command(["-u", str(account_file), "renew"])
|
||||
return result.returncode == 0
|
||||
|
||||
|
||||
async def upload_video(request: DouyinVideoUploadRequest) -> Path:
|
||||
account_file = resolve_account_file("douyin", request.account_name)
|
||||
is_ready = await douyin_setup(str(account_file), handle=False)
|
||||
@@ -229,6 +260,36 @@ async def upload_kuaishou_note(request: KuaishouNoteUploadRequest) -> Path:
|
||||
return account_file
|
||||
|
||||
|
||||
async def upload_bilibili_video(request: BilibiliVideoUploadRequest) -> Path:
|
||||
account_file = resolve_account_file("bilibili", request.account_name)
|
||||
if not account_file.exists():
|
||||
raise RuntimeError(
|
||||
f"Bilibili account file is missing: {account_file}. Run `sau bilibili login --account {request.account_name}` first."
|
||||
)
|
||||
|
||||
arguments = [
|
||||
"-u",
|
||||
str(account_file),
|
||||
"upload",
|
||||
str(request.video_file),
|
||||
"--title",
|
||||
request.title,
|
||||
"--desc",
|
||||
request.description,
|
||||
"--tid",
|
||||
str(request.tid),
|
||||
]
|
||||
if request.tags:
|
||||
arguments.extend(["--tag", ",".join(request.tags)])
|
||||
if isinstance(request.publish_date, datetime):
|
||||
arguments.extend(["--dtime", str(int(request.publish_date.timestamp()))])
|
||||
|
||||
result = run_biliup_command(arguments)
|
||||
if result.returncode != 0:
|
||||
raise RuntimeError((result.stderr or result.stdout or "").strip() or "Bilibili upload failed")
|
||||
return account_file
|
||||
|
||||
|
||||
def existing_file_path(value: str) -> Path:
|
||||
path = Path(value)
|
||||
if not path.is_file():
|
||||
@@ -314,6 +375,22 @@ def build_parser() -> argparse.ArgumentParser:
|
||||
kuaishou_upload_note_parser.add_argument("--tags", default="", help="Comma-separated tags, such as tag1,tag2")
|
||||
kuaishou_upload_note_parser.add_argument("--schedule", type=schedule_value, help=f"Schedule time in {schedule_help}")
|
||||
add_runtime_flags(kuaishou_upload_note_parser)
|
||||
|
||||
bilibili_parser = platform_parsers.add_parser("bilibili", help="Bilibili operations")
|
||||
bilibili_actions = bilibili_parser.add_subparsers(dest="action", required=True)
|
||||
|
||||
for action_name in ("login", "check"):
|
||||
action_parser = bilibili_actions.add_parser(action_name, help=f"Bilibili {action_name}")
|
||||
action_parser.add_argument("--account", required=True, help="Bilibili account alias")
|
||||
|
||||
bilibili_upload_video_parser = bilibili_actions.add_parser("upload-video", help="Upload one video to Bilibili")
|
||||
bilibili_upload_video_parser.add_argument("--account", required=True, help="Bilibili account alias")
|
||||
bilibili_upload_video_parser.add_argument("--file", required=True, type=existing_file_path, help="Video file path")
|
||||
bilibili_upload_video_parser.add_argument("--title", required=True, help="Video title")
|
||||
bilibili_upload_video_parser.add_argument("--desc", required=True, help="Video description")
|
||||
bilibili_upload_video_parser.add_argument("--tid", required=True, type=int, help="Bilibili category id")
|
||||
bilibili_upload_video_parser.add_argument("--tags", default="", help="Comma-separated tags, such as tag1,tag2")
|
||||
bilibili_upload_video_parser.add_argument("--schedule", type=schedule_value, help=f"Schedule time in {schedule_help}")
|
||||
return parser
|
||||
|
||||
|
||||
@@ -416,6 +493,35 @@ async def dispatch(args: argparse.Namespace) -> int:
|
||||
|
||||
raise RuntimeError(f"Unsupported Kuaishou action: {args.action}")
|
||||
|
||||
if args.platform == "bilibili":
|
||||
if args.action == "login":
|
||||
result = await login_bilibili_account(args.account)
|
||||
if not result["success"]:
|
||||
raise RuntimeError(result["message"])
|
||||
print(f"Bilibili login flow completed: {result['account_file']}")
|
||||
return 0
|
||||
|
||||
if args.action == "check":
|
||||
is_valid = await check_bilibili_account(args.account)
|
||||
print("valid" if is_valid else "invalid")
|
||||
return 0 if is_valid else 1
|
||||
|
||||
if args.action == "upload-video":
|
||||
request = BilibiliVideoUploadRequest(
|
||||
account_name=args.account,
|
||||
video_file=args.file,
|
||||
title=args.title,
|
||||
description=args.desc,
|
||||
tid=args.tid,
|
||||
tags=parse_tags(args.tags),
|
||||
publish_date=args.schedule or 0,
|
||||
)
|
||||
await upload_bilibili_video(request)
|
||||
print(f"Bilibili video upload submitted: {request.video_file}")
|
||||
return 0
|
||||
|
||||
raise RuntimeError(f"Unsupported Bilibili action: {args.action}")
|
||||
|
||||
raise RuntimeError(f"Unsupported platform: {args.platform}")
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user