87360051fa
SQLAlchemy's 'metadata' attribute is reserved for model metadata, causing
service startup crash. Changed to 'extra_meta' with Column('metadata', JSON)
mapping to preserve database column name.
Affected models (7):
- AssetLibraryModel, AssetModel, AssetClassificationModel
- GenerationTaskModel, DuplicationRecordModel
- TitleLibraryModel, VoiceLibraryModel
Updated all references across:
- packages/adapters/sqlalchemy_impl/{title_library,voice_library,asset}_repository.py
- apps/worker/worker_app/{core/asset_usage.py,tasks/voice_extraction.py}
- apps/api/app/api/routes/{asset_diagnosis.py,assets.py}
257 lines
12 KiB
Python
Executable File
257 lines
12 KiB
Python
Executable File
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import Boolean, Column, DateTime, Float, Integer, String, Text, JSON, UniqueConstraint, create_engine
|
|
from sqlalchemy.orm import declarative_base
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
class UserModel(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(String(36), primary_key=True)
|
|
email = Column(String(255), nullable=False, unique=True, index=True)
|
|
username = Column(String(100), nullable=True, unique=True, index=True)
|
|
display_name = Column(String(255), nullable=False)
|
|
password_hash = Column(String(255), nullable=False)
|
|
email_verified = Column(Boolean, nullable=False, default=False)
|
|
email_verification_token = Column(String(255), nullable=True)
|
|
password_reset_token = Column(String(255), nullable=True)
|
|
password_reset_expires_at = Column(DateTime, nullable=True)
|
|
last_login_at = Column(DateTime, nullable=True)
|
|
last_login_ip = Column(String(50), nullable=True)
|
|
# 订阅相关字段 (移到 User 级别)
|
|
subscription_plan = Column(String(20), nullable=False, default="free")
|
|
subscription_status = Column(String(20), nullable=False, default="active")
|
|
subscription_expires_at = Column(DateTime, nullable=True)
|
|
# 配额限制 (移到 User 级别)
|
|
max_projects = Column(Integer, nullable=False, default=3)
|
|
max_storage_gb = Column(Integer, nullable=False, default=10)
|
|
used_storage_gb = Column(Integer, nullable=False, default=0)
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class ProjectModel(Base):
|
|
__tablename__ = "projects"
|
|
|
|
id = Column(String(32), primary_key=True)
|
|
owner_user_id = Column(String(32), nullable=False, index=True)
|
|
name = Column(String(100), nullable=False)
|
|
description = Column(Text, nullable=False, default="")
|
|
shared_users = Column(JSON, nullable=False, default=list) # 被共享的用户 ID 列表
|
|
extra_meta = Column('metadata', JSON, nullable=False, default=dict)
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class AssetLibraryModel(Base):
|
|
__tablename__ = "asset_libraries"
|
|
|
|
id = Column(String(36), primary_key=True)
|
|
project_id = Column(String(36), nullable=True, index=True)
|
|
name = Column(String(255), nullable=False)
|
|
kind = Column(String(20), nullable=False, index=True)
|
|
asset_count = Column(Float, nullable=False, default=0)
|
|
total_size = Column(Float, nullable=False, default=0)
|
|
extra_meta = Column('metadata', JSON, nullable=False, default=dict)
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class AssetModel(Base):
|
|
__tablename__ = "assets"
|
|
|
|
id = Column(String(36), primary_key=True)
|
|
project_id = Column(String(36), nullable=False, index=True)
|
|
asset_library_id = Column(String(36), nullable=False, index=True)
|
|
name = Column(String(500), nullable=False)
|
|
file_type = Column(String(20), nullable=False, index=True)
|
|
# storage_key: OSS 对象键(相对路径),用于内部存储和操作
|
|
storage_key = Column(String(255), nullable=False)
|
|
# file_size: 文件大小(字节),使用 Integer 类型以确保精确性
|
|
file_size = Column(Integer, nullable=False)
|
|
# file_url: 完整可访问的 URL,用于客户端直接访问文件
|
|
file_url = Column(String(1000), nullable=False)
|
|
thumbnail_url = Column(String(1000), nullable=True)
|
|
duration = Column(Float, nullable=True)
|
|
width = Column(Float, nullable=True)
|
|
height = Column(Float, nullable=True)
|
|
fps = Column(Float, nullable=True)
|
|
codec = Column(String(50), nullable=True)
|
|
status = Column(String(20), nullable=False, default="uploading", index=True)
|
|
classification_status = Column(String(20), nullable=False, default="pending", index=True)
|
|
classification_result = Column(Text, nullable=True)
|
|
quality_score = Column(Float, nullable=True)
|
|
uploaded_by_user_id = Column(String(36), nullable=False)
|
|
extra_meta = Column('metadata', JSON, nullable=False, default=dict)
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc), index=True)
|
|
updated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
|
|
class EditTemplateModel(Base):
|
|
__tablename__ = "edit_templates"
|
|
|
|
id = Column(String(32), primary_key=True)
|
|
project_id = Column(String(32), nullable=False, index=True)
|
|
name = Column(String(120), nullable=False)
|
|
description = Column(Text, nullable=False, default="")
|
|
target_duration = Column(Float, nullable=False, default=30)
|
|
clip_count = Column(Integer, nullable=False, default=3)
|
|
is_active = Column(Boolean, nullable=False, default=True)
|
|
created_by_user_id = Column(String(32), nullable=False, default="")
|
|
extra_meta = Column('metadata', JSON, nullable=False, default=dict)
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
|
|
|
|
|
|
class IngestJobModel(Base):
|
|
__tablename__ = "ingest_jobs"
|
|
|
|
id = Column(String(32), primary_key=True)
|
|
project_id = Column(String(32), nullable=False, index=True)
|
|
library_id = Column(String(32), nullable=False, index=True)
|
|
storage_key = Column(String(255), nullable=False)
|
|
status = Column(String(20), nullable=False, default="pending")
|
|
error_message = Column(Text, nullable=False, default="")
|
|
result_asset_id = Column(String(32), nullable=False, default="")
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class ClassificationJobModel(Base):
|
|
__tablename__ = "classification_jobs"
|
|
|
|
id = Column(String(32), primary_key=True)
|
|
project_id = Column(String(32), nullable=False, index=True)
|
|
asset_id = Column(String(32), nullable=False, index=True)
|
|
status = Column(String(20), nullable=False, default="pending")
|
|
classification = Column(String(50), nullable=False, default="")
|
|
confidence = Column(Float, nullable=False, default=0.0)
|
|
error_message = Column(Text, nullable=False, default="")
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class GenerationTaskModel(Base):
|
|
__tablename__ = "generation_tasks"
|
|
|
|
id = Column(String(32), primary_key=True)
|
|
project_id = Column(String(32), nullable=False, index=True)
|
|
strategy_id = Column(String(32), nullable=False, default="")
|
|
asset_library_id = Column(String(32), nullable=False, index=True)
|
|
voice_library_id = Column(String(32), nullable=False, default="")
|
|
editing_mode = Column(String(20), nullable=False, default="one_take", index=True) # 剪辑模式: one_take, pip, voice_over, voice_pip
|
|
status = Column(String(20), nullable=False, default="pending", index=True)
|
|
progress = Column(Float, nullable=False, default=0.0)
|
|
result_count = Column(Float, nullable=False, default=0)
|
|
error_message = Column(Text, nullable=False, default="")
|
|
started_at = Column(DateTime, nullable=True)
|
|
completed_at = Column(DateTime, nullable=True)
|
|
created_by_user_id = Column(String(32), nullable=False, default="")
|
|
extra_meta = Column('metadata', JSON, nullable=False, default=dict)
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class GeneratedVideoModel(Base):
|
|
__tablename__ = "generated_videos"
|
|
|
|
id = Column(String(32), primary_key=True)
|
|
project_id = Column(String(32), nullable=False, index=True)
|
|
generation_task_id = Column(String(32), nullable=False, index=True)
|
|
name = Column(String(255), nullable=False)
|
|
# file_url: 完整可访问的 URL,用于客户端直接访问视频
|
|
file_url = Column(String(1000), nullable=False)
|
|
# file_size: 文件大小(字节),使用 Integer 类型以确保精确性
|
|
file_size = Column(Integer, nullable=False)
|
|
duration = Column(Float, nullable=False)
|
|
thumbnail_url = Column(String(1000), nullable=True)
|
|
width = Column(Float, nullable=False)
|
|
height = Column(Float, nullable=False)
|
|
fps = Column(Float, nullable=False)
|
|
status = Column(String(20), nullable=False, default="completed", index=True)
|
|
review_status = Column(String(20), nullable=False, default="pending_review", index=True)
|
|
generation_params = Column(Text, nullable=False, default="{}")
|
|
generated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, nullable=True)
|
|
video_fingerprint = Column(Text, nullable=True)
|
|
is_duplicate = Column(Boolean, nullable=False, default=False)
|
|
duplicate_of = Column(String(32), nullable=True)
|
|
|
|
|
|
|
|
|
|
|
|
class TitleLibraryModel(Base):
|
|
__tablename__ = "title_libraries"
|
|
|
|
id = Column(String(36), primary_key=True)
|
|
user_id = Column(String(36), nullable=False, index=True)
|
|
name = Column(String(255), nullable=False)
|
|
description = Column(Text, nullable=False, default="")
|
|
category = Column(String(50), nullable=False, default="default", index=True)
|
|
text = Column(String(500), nullable=False)
|
|
tags = Column(JSON, nullable=False, default=list)
|
|
usage_count = Column(Integer, nullable=False, default=0)
|
|
is_active = Column(Boolean, nullable=False, default=True, index=True)
|
|
extra_meta = Column('metadata', JSON, nullable=False, default=dict)
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class VoiceLibraryModel(Base):
|
|
__tablename__ = "voice_libraries"
|
|
|
|
id = Column(String(36), primary_key=True)
|
|
user_id = Column(String(36), nullable=False, index=True)
|
|
project_id = Column(String(36), nullable=True, index=True)
|
|
name = Column(String(255), nullable=False)
|
|
text = Column(Text, nullable=False, default="")
|
|
voice_provider = Column(String(50), nullable=False, default="")
|
|
voice_id = Column(String(100), nullable=False, default="")
|
|
voice_name = Column(String(100), nullable=False, default="")
|
|
audio_url = Column(String(1000), nullable=False, default="")
|
|
duration = Column(Float, nullable=False, default=0)
|
|
file_size = Column(Integer, nullable=False, default=0)
|
|
status = Column(String(20), nullable=False, default="completed", index=True)
|
|
tags = Column(JSON, nullable=False, default=list)
|
|
extra_meta = Column('metadata', JSON, nullable=False, default=dict)
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class DuplicationRecordModel(Base):
|
|
__tablename__ = "duplication_records"
|
|
|
|
id = Column(String(36), primary_key=True)
|
|
user_id = Column(String(36), nullable=False, index=True)
|
|
filename = Column(String(500), nullable=False)
|
|
file_size = Column(Integer, nullable=False)
|
|
storage_key = Column(String(500), nullable=False)
|
|
duration_seconds = Column(Float, nullable=False, default=0)
|
|
status = Column(String(20), nullable=False, default="pending", index=True)
|
|
duplicate_rate = Column(Float, nullable=True)
|
|
duplicate_count = Column(Integer, nullable=False, default=0)
|
|
video_fingerprint = Column(Text, nullable=True)
|
|
error_message = Column(Text, nullable=False, default="")
|
|
created_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class DuplicationSegmentModel(Base):
|
|
__tablename__ = "duplication_segments"
|
|
|
|
id = Column(String(36), primary_key=True)
|
|
record_id = Column(String(36), nullable=False, index=True)
|
|
source_start = Column(Float, nullable=False)
|
|
source_end = Column(Float, nullable=False)
|
|
matched_video_id = Column(String(36), nullable=False)
|
|
matched_video_name = Column(String(500), nullable=False, default="")
|
|
matched_start = Column(Float, nullable=False)
|
|
matched_end = Column(Float, nullable=False)
|
|
similarity = Column(Float, nullable=False)
|
|
|