Xiaoxia AI
302 lines
8.9 KiB
Python
302 lines
8.9 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass, field
|
|
from datetime import datetime, timezone
|
|
from enum import StrEnum
|
|
from typing import Any
|
|
from uuid import uuid4
|
|
|
|
|
|
class AssetLibraryKind(StrEnum):
|
|
VIDEO = "video"
|
|
VOICE = "voice"
|
|
IMAGE = "image"
|
|
|
|
|
|
class IngestJobStatus(StrEnum):
|
|
PENDING = "pending"
|
|
PROCESSING = "processing"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class User:
|
|
id: str
|
|
email: str
|
|
display_name: str
|
|
username: str = ""
|
|
# 认证相关字段
|
|
password_hash: str = ""
|
|
email_verified: bool = False
|
|
email_verification_token: str | None = None
|
|
password_reset_token: str | None = None
|
|
password_reset_expires_at: datetime | None = None
|
|
last_login_at: datetime | None = None
|
|
last_login_ip: str | None = None
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class Workspace:
|
|
id: str
|
|
name: str
|
|
owner_user_id: str
|
|
# 订阅相关字段
|
|
subscription_plan: str = "free" # free, pro, enterprise
|
|
subscription_status: str = "active" # active, cancelled, expired
|
|
subscription_expires_at: datetime | None = None
|
|
# 配额限制
|
|
max_projects: int = 3 # free: 3, pro: unlimited, enterprise: unlimited
|
|
max_storage_gb: int = 10 # free: 10, pro: 100, enterprise: 1000
|
|
used_storage_gb: float = 0.0
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class WorkspaceMemberRole(StrEnum):
|
|
"""工作空间成员角色"""
|
|
|
|
OWNER = "owner" # 所有者（创建者，唯一）
|
|
ADMIN = "admin" # 管理员（可管理成员和项目）
|
|
MEMBER = "member" # 成员（可创建和编辑项目）
|
|
VIEWER = "viewer" # 查看者（只读）
|
|
|
|
|
|
class InvitationStatus(StrEnum):
|
|
"""邀请状态"""
|
|
|
|
PENDING = "pending" # 待处理
|
|
ACCEPTED = "accepted" # 已接受
|
|
DECLINED = "declined" # 已拒绝
|
|
EXPIRED = "expired" # 已过期
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class WorkspaceMember:
|
|
"""工作空间成员"""
|
|
|
|
id: str
|
|
workspace_id: str
|
|
user_id: str
|
|
role: str # owner, admin, member, viewer
|
|
invited_by: str | None = None # 邀请人 user_id
|
|
joined_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class WorkspaceInvitation:
|
|
"""工作空间邀请"""
|
|
|
|
id: str
|
|
workspace_id: str
|
|
inviter_user_id: str # 邀请人
|
|
invitee_email: str # 被邀请人邮箱
|
|
role: str # 邀请角色
|
|
invitation_token: str # 邀请令牌
|
|
status: str = "pending" # pending, accepted, declined, expired
|
|
expires_at: datetime | None = None # 过期时间
|
|
accepted_at: datetime | None = None # 接受时间
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class Project:
|
|
id: str
|
|
workspace_id: str
|
|
name: str
|
|
description: str = ""
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
@classmethod
|
|
def create(cls, workspace_id: str, name: str, description: str = "") -> "Project":
|
|
clean_name = name.strip()
|
|
if not clean_name:
|
|
raise ValueError("项目名称不能为空")
|
|
return cls(
|
|
id=uuid4().hex,
|
|
workspace_id=workspace_id,
|
|
name=clean_name,
|
|
description=description.strip(),
|
|
)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class AssetLibrary:
|
|
id: str
|
|
workspace_id: str
|
|
project_id: str
|
|
name: str
|
|
kind: AssetLibraryKind
|
|
asset_count: int = 0
|
|
total_size: int = 0
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
updated_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
@classmethod
|
|
def create(
|
|
cls,
|
|
workspace_id: str,
|
|
project_id: str,
|
|
name: str,
|
|
kind: AssetLibraryKind,
|
|
) -> "AssetLibrary":
|
|
clean_name = name.strip()
|
|
if not clean_name:
|
|
raise ValueError("素材库名称不能为空")
|
|
return cls(
|
|
id=uuid4().hex,
|
|
workspace_id=workspace_id,
|
|
project_id=project_id,
|
|
name=clean_name,
|
|
kind=kind,
|
|
asset_count=0,
|
|
total_size=0,
|
|
)
|
|
|
|
|
|
class AssetStatus(StrEnum):
|
|
UPLOADING = "uploading"
|
|
READY = "ready"
|
|
PROCESSING = "processing"
|
|
ERROR = "error"
|
|
|
|
|
|
class ClassificationStatus(StrEnum):
|
|
PENDING = "pending"
|
|
PROCESSING = "processing"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class Asset:
|
|
id: str
|
|
workspace_id: str
|
|
project_id: str
|
|
library_id: str
|
|
name: str
|
|
storage_key: str
|
|
mime_type: str
|
|
file_size: int = 0
|
|
thumbnail_url: str | None = None
|
|
duration: float | None = None
|
|
width: int | None = None
|
|
height: int | None = None
|
|
fps: float | None = None
|
|
codec: str | None = None
|
|
status: AssetStatus = AssetStatus.UPLOADING
|
|
classification_status: ClassificationStatus = ClassificationStatus.PENDING
|
|
quality_score: float | None = None
|
|
uploaded_by_user_id: str = ""
|
|
metadata: dict[str, Any] = field(default_factory=dict)
|
|
tags: list[str] = field(default_factory=list)
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
updated_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
@classmethod
|
|
def create(
|
|
cls,
|
|
workspace_id: str,
|
|
project_id: str,
|
|
library_id: str,
|
|
name: str,
|
|
storage_key: str,
|
|
mime_type: str,
|
|
metadata: dict[str, Any] | None = None,
|
|
*,
|
|
file_size: int = 0,
|
|
thumbnail_url: str | None = None,
|
|
duration: float | None = None,
|
|
width: int | None = None,
|
|
height: int | None = None,
|
|
fps: float | None = None,
|
|
codec: str | None = None,
|
|
status: AssetStatus = AssetStatus.UPLOADING,
|
|
classification_status: ClassificationStatus = ClassificationStatus.PENDING,
|
|
quality_score: float | None = None,
|
|
uploaded_by_user_id: str = "",
|
|
) -> "Asset":
|
|
clean_name = name.strip()
|
|
if not clean_name:
|
|
raise ValueError("素材名称不能为空")
|
|
if not storage_key.strip():
|
|
raise ValueError("storage_key 不能为空")
|
|
if not mime_type.strip():
|
|
raise ValueError("mime_type 不能为空")
|
|
return cls(
|
|
id=uuid4().hex,
|
|
workspace_id=workspace_id,
|
|
project_id=project_id,
|
|
library_id=library_id,
|
|
name=clean_name,
|
|
storage_key=storage_key.strip(),
|
|
mime_type=mime_type.strip(),
|
|
file_size=file_size,
|
|
thumbnail_url=thumbnail_url,
|
|
duration=duration,
|
|
width=width,
|
|
height=height,
|
|
fps=fps,
|
|
codec=codec,
|
|
status=status,
|
|
classification_status=classification_status,
|
|
quality_score=quality_score,
|
|
uploaded_by_user_id=uploaded_by_user_id.strip(),
|
|
metadata=metadata or {},
|
|
tags=[],
|
|
)
|
|
|
|
def add_tag(self, tag: str) -> None:
|
|
"""添加标签。空标签会被忽略，自动去重。"""
|
|
clean_tag = tag.strip()
|
|
if not clean_tag:
|
|
raise ValueError("标签不能为空")
|
|
if clean_tag not in self.tags:
|
|
self.tags.append(clean_tag)
|
|
self.updated_at = datetime.now(timezone.utc)
|
|
|
|
def remove_tag(self, tag: str) -> None:
|
|
"""删除标签。如果标签不存在，不报错（幂等性）。"""
|
|
clean_tag = tag.strip()
|
|
if clean_tag in self.tags:
|
|
self.tags.remove(clean_tag)
|
|
self.updated_at = datetime.now(timezone.utc)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class IngestJob:
|
|
id: str
|
|
workspace_id: str
|
|
project_id: str
|
|
library_id: str
|
|
storage_key: str
|
|
status: IngestJobStatus = IngestJobStatus.PENDING
|
|
error_message: str = ""
|
|
result_asset_id: str = ""
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
updated_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
@classmethod
|
|
def create(
|
|
cls,
|
|
workspace_id: str,
|
|
project_id: str,
|
|
library_id: str,
|
|
storage_key: str,
|
|
) -> "IngestJob":
|
|
if not workspace_id.strip():
|
|
raise ValueError("workspace_id 不能为空")
|
|
if not project_id.strip():
|
|
raise ValueError("project_id 不能为空")
|
|
if not library_id.strip():
|
|
raise ValueError("library_id 不能为空")
|
|
if not storage_key.strip():
|
|
raise ValueError("storage_key 不能为空")
|
|
return cls(
|
|
id=uuid4().hex,
|
|
workspace_id=workspace_id.strip(),
|
|
project_id=project_id.strip(),
|
|
library_id=library_id.strip(),
|
|
storage_key=storage_key.strip(),
|
|
)
|