Files
xiaoxia-saas/packages/application/duplication.py
T
xiaoxia 52ff2f80ad
CI/CD Pipeline / Validate Code Quality And Tests (push) Has been cancelled
CI/CD Pipeline / Frontend Lint (push) Has been cancelled
Deploy / Deploy Staging (push) Has been cancelled
Deploy / Build Production Runtime Images (push) Has been cancelled
Deploy / Deploy Production (push) Has been cancelled
Deploy / Production Browser E2E (push) Has been cancelled
style: apply black formatting to pass CI validation (#126)
2026-06-30 17:23:08 +08:00

92 lines
3.1 KiB
Python

"""查重应用层用例。"""
from __future__ import annotations
import logging
from dataclasses import dataclass
from packages.domain.duplication import DuplicationRecord
from packages.ports.duplication_repository import DuplicationRecordRepository
logger = logging.getLogger(__name__)
@dataclass
class UploadForDuplicationCommand:
user_id: str
filename: str
file_size: int
storage_key: str
duration_seconds: float = 0.0
class UploadForDuplicationUseCase:
"""上传视频进行查重 — 创建查重记录并返回。"""
def __init__(self, duplication_repository: DuplicationRecordRepository):
self.duplication_repository = duplication_repository
def execute(self, command: UploadForDuplicationCommand) -> DuplicationRecord:
record = DuplicationRecord.create(
user_id=command.user_id,
filename=command.filename,
file_size=command.file_size,
storage_key=command.storage_key,
duration_seconds=command.duration_seconds,
)
record = self.duplication_repository.create(record)
logger.info("Duplication record created: %s for user %s", record.id, record.user_id)
return record
class ListDuplicationRecordsUseCase:
"""获取用户查重记录列表。"""
def __init__(self, duplication_repository: DuplicationRecordRepository):
self.duplication_repository = duplication_repository
def execute(self, user_id: str, *, offset: int = 0, limit: int = 50) -> list[DuplicationRecord]:
if not user_id.strip():
raise ValueError("user_id 不能为空")
return self.duplication_repository.list_by_user(user_id.strip(), offset=offset, limit=limit)
class GetDuplicationDetailUseCase:
"""获取查重记录详情(含重复片段)。"""
def __init__(self, duplication_repository: DuplicationRecordRepository):
self.duplication_repository = duplication_repository
def execute(self, record_id: str) -> DuplicationRecord | None:
return self.duplication_repository.get(record_id)
class DeleteDuplicationRecordUseCase:
"""删除查重记录。"""
def __init__(self, duplication_repository: DuplicationRecordRepository):
self.duplication_repository = duplication_repository
def execute(self, record_id: str) -> bool:
return self.duplication_repository.delete(record_id)
class RetryDuplicationUseCase:
"""重新提交查重 — 将记录状态重置为 pending。"""
def __init__(self, duplication_repository: DuplicationRecordRepository):
self.duplication_repository = duplication_repository
def execute(self, record_id: str) -> DuplicationRecord | None:
record = self.duplication_repository.get(record_id)
if record is None:
return None
record.status = "pending"
record.error_message = ""
record.duplicate_rate = None
record.duplicate_count = 0
record.segments = []
record = self.duplication_repository.update(record)
logger.info("Duplication record %s reset to pending for retry", record_id)
return record