85 lines
4.9 KiB
Python
85 lines
4.9 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import DateTime, Float, ForeignKey, Index, Integer, String, Text, text
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.enums.private_portrait import (
|
|
PrivatePortraitAssetStatus,
|
|
PrivatePortraitAssetType,
|
|
PrivatePortraitLibraryType,
|
|
PrivatePortraitRemoteDeleteStatus,
|
|
)
|
|
from app.models.base import Base, SoftDeleteMixin, TimestampMixin
|
|
|
|
|
|
class PrivatePortraitAsset(Base, TimestampMixin, SoftDeleteMixin):
|
|
"""火山 Asset 本地映射。生成时只信任本地 id，再批量解析成 asset://remote_asset_id。"""
|
|
|
|
__tablename__ = "private_portrait_assets"
|
|
__table_args__ = (
|
|
Index("uq_private_portrait_assets_remote_asset_id", "remote_asset_id", unique=True),
|
|
Index("idx_private_portrait_assets_user_library_status_created", "user_id", "library_type", "status", "created_at"),
|
|
Index("idx_private_portrait_assets_user_status_created", "user_id", "status", "created_at"),
|
|
Index("idx_private_portrait_assets_project_status_created", "project_id", "status", "created_at"),
|
|
Index("idx_private_portrait_assets_group_status_created", "group_id", "status", "created_at"),
|
|
Index("idx_private_portrait_assets_remote_project_name", "remote_project_name"),
|
|
Index("idx_private_portrait_assets_library_type", "library_type"),
|
|
Index("idx_private_portrait_assets_asset_type", "asset_type"),
|
|
Index(
|
|
"idx_private_portrait_assets_next_poll_status",
|
|
"next_poll_at",
|
|
"status",
|
|
postgresql_where=text("deleted_at IS NULL AND next_poll_at IS NOT NULL"),
|
|
),
|
|
Index("idx_private_portrait_assets_remote_delete_status", "remote_delete_status"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(String(32), primary_key=True)
|
|
user_id: Mapped[str] = mapped_column(String(32), ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True)
|
|
project_id: Mapped[str] = mapped_column(String(32), ForeignKey("private_portrait_projects.id", ondelete="CASCADE"), nullable=False, index=True)
|
|
group_id: Mapped[str] = mapped_column(String(32), ForeignKey("private_portrait_asset_groups.id", ondelete="CASCADE"), nullable=False, index=True)
|
|
library_type: Mapped[str] = mapped_column(
|
|
String(32),
|
|
nullable=False,
|
|
default=PrivatePortraitLibraryType.REAL_PERSON.value,
|
|
server_default=PrivatePortraitLibraryType.REAL_PERSON.value,
|
|
index=True,
|
|
)
|
|
remote_group_id: Mapped[str] = mapped_column(String(128), nullable=False, index=True)
|
|
remote_asset_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
remote_project_name: Mapped[str] = mapped_column(String(256), nullable=False, index=True)
|
|
asset_type: Mapped[str] = mapped_column(String(16), nullable=False, default=PrivatePortraitAssetType.IMAGE.value, index=True)
|
|
name: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
source_url: Mapped[str] = mapped_column(Text, nullable=False)
|
|
preview_url: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
remote_url: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
remote_url_expired_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
video_duration: Mapped[float | None] = mapped_column(Float, nullable=True, comment="视频素材时长，秒")
|
|
video_cover_url: Mapped[str | None] = mapped_column(Text, nullable=True, comment="视频素材封面预览地址")
|
|
file_size: Mapped[int | None] = mapped_column(Integer, nullable=True, comment="素材文件大小，字节")
|
|
mime_type: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
status: Mapped[str] = mapped_column(
|
|
String(32),
|
|
nullable=False,
|
|
default=PrivatePortraitAssetStatus.CREATING.value,
|
|
server_default=PrivatePortraitAssetStatus.CREATING.value,
|
|
index=True,
|
|
)
|
|
moderation_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
last_poll_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
next_poll_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)
|
|
poll_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0, server_default="0")
|
|
remote_delete_status: Mapped[str] = mapped_column(
|
|
String(32),
|
|
nullable=False,
|
|
default=PrivatePortraitRemoteDeleteStatus.NONE.value,
|
|
server_default=PrivatePortraitRemoteDeleteStatus.NONE.value,
|
|
index=True,
|
|
)
|
|
remote_deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
remote_delete_error: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
raw_response_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|