init biliup-next
This commit is contained in:
43
src/biliup_next/modules/publish/service.py
Normal file
43
src/biliup_next/modules/publish/service.py
Normal file
@ -0,0 +1,43 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from pathlib import Path
|
||||
|
||||
from biliup_next.core.models import Artifact, PublishRecord, utc_now_iso
|
||||
from biliup_next.core.registry import Registry
|
||||
from biliup_next.infra.task_repository import TaskRepository
|
||||
|
||||
|
||||
class PublishService:
|
||||
def __init__(self, registry: Registry, repo: TaskRepository):
|
||||
self.registry = registry
|
||||
self.repo = repo
|
||||
|
||||
def run(self, task_id: str, settings: dict[str, object]) -> PublishRecord:
|
||||
task = self.repo.get_task(task_id)
|
||||
if task is None:
|
||||
raise RuntimeError(f"task not found: {task_id}")
|
||||
artifacts = self.repo.list_artifacts(task_id)
|
||||
clip_videos = [a for a in artifacts if a.artifact_type == "clip_video"]
|
||||
provider = self.registry.get("publish_provider", str(settings.get("provider", "biliup_cli")))
|
||||
started_at = utc_now_iso()
|
||||
self.repo.update_step_status(task_id, "publish", "running", started_at=started_at)
|
||||
record = provider.publish(task, clip_videos, settings)
|
||||
self.repo.add_publish_record(record)
|
||||
if record.bvid:
|
||||
session_dir = Path(str(settings.get("session_dir", "session"))) / task.title
|
||||
bvid_path = str((session_dir / "bvid.txt").resolve())
|
||||
self.repo.add_artifact(
|
||||
Artifact(
|
||||
id=None,
|
||||
task_id=task_id,
|
||||
artifact_type="publish_bvid",
|
||||
path=bvid_path,
|
||||
metadata_json=json.dumps({}),
|
||||
created_at=utc_now_iso(),
|
||||
)
|
||||
)
|
||||
finished_at = utc_now_iso()
|
||||
self.repo.update_step_status(task_id, "publish", "succeeded", finished_at=finished_at)
|
||||
self.repo.update_task_status(task_id, "published", finished_at)
|
||||
return record
|
||||
Reference in New Issue
Block a user