37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from sqlalchemy import Column, Integer, String, DateTime, JSON, Text, Float, Enum
|
|
from datetime import datetime
|
|
import enum
|
|
from models.subscription_models import Base
|
|
|
|
class VideoTaskStatus(enum.Enum):
|
|
PENDING = "pending"
|
|
PROCESSING = "processing"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
|
|
class VideoGenerationTask(Base):
|
|
"""
|
|
Model for tracking video generation tasks (Video Studio).
|
|
"""
|
|
__tablename__ = "video_generation_tasks"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
task_id = Column(String(36), unique=True, index=True, nullable=False) # UUID
|
|
user_id = Column(String(255), nullable=False, index=True)
|
|
|
|
status = Column(Enum(VideoTaskStatus), default=VideoTaskStatus.PENDING)
|
|
|
|
# Task inputs (stored as JSON)
|
|
request_data = Column(JSON, nullable=True)
|
|
|
|
# Task results
|
|
result = Column(JSON, nullable=True)
|
|
error = Column(Text, nullable=True)
|
|
|
|
# Progress tracking
|
|
progress = Column(Float, default=0.0)
|
|
message = Column(String(255), nullable=True)
|
|
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|