Add source

shared/database/models.py

@@ -0,0 +1,91 @@
+"""
+ORM models for database
+"""
+from datetime import datetime
+from sqlalchemy import Column, Integer, BigInteger, String, Boolean, DateTime, Text, ForeignKey
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship
+
+Base = declarative_base()
+
+
+class User(Base):
+    """User model"""
+    __tablename__ = "users"
+    
+    user_id = Column(Integer, primary_key=True, unique=True, index=True)
+    username = Column(String(255), nullable=True)
+    first_name = Column(String(255), nullable=True)
+    last_name = Column(String(255), nullable=True)
+    is_admin = Column(Boolean, default=False)
+    is_blocked = Column(Boolean, default=False)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    
+    # Relationships
+    tasks = relationship("Task", back_populates="user")
+    
+    def __repr__(self):
+        return f"<User(user_id={self.user_id}, username={self.username}, is_admin={self.is_admin})>"
+
+
+class Task(Base):
+    """Task model"""
+    __tablename__ = "tasks"
+    
+    id = Column(BigInteger, primary_key=True, index=True)
+    user_id = Column(Integer, ForeignKey("users.user_id"), nullable=False, index=True)  # Index for frequent queries
+    task_type = Column(String(50), nullable=False)  # download, process, etc.
+    status = Column(String(50), default="pending", index=True)  # Index for status filtering
+    url = Column(Text, nullable=True)
+    file_path = Column(String(500), nullable=True)
+    progress = Column(Integer, default=0)  # 0-100
+    error_message = Column(String(1000), nullable=True)  # Limit to 1000 characters
+    created_at = Column(DateTime, default=datetime.utcnow, index=True)  # Index for sorting
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    completed_at = Column(DateTime, nullable=True)
+    
+    # Relationships
+    user = relationship("User", back_populates="tasks")
+    
+    def __repr__(self):
+        return f"<Task(id={self.id}, user_id={self.user_id}, status={self.status})>"
+
+
+class Download(Base):
+    """Download model"""
+    __tablename__ = "downloads"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    task_id = Column(BigInteger, ForeignKey("tasks.id"), nullable=False)
+    url = Column(Text, nullable=False)
+    download_type = Column(String(50), nullable=False)  # direct, ytdlp
+    file_path = Column(String(500), nullable=True)
+    file_size = Column(Integer, nullable=True)
+    duration = Column(Integer, nullable=True)  # Download duration in seconds
+    created_at = Column(DateTime, default=datetime.utcnow)
+    
+    # Relationships
+    task = relationship("Task")
+    
+    def __repr__(self):
+        return f"<Download(id={self.id}, task_id={self.task_id}, download_type={self.download_type})>"
+
+
+class OTPCode(Base):
+    """One-time password code model for web interface authentication"""
+    __tablename__ = "otp_codes"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    user_id = Column(Integer, ForeignKey("users.user_id"), nullable=False, index=True)
+    code = Column(String(6), nullable=False, index=True)  # 6-digit code
+    expires_at = Column(DateTime, nullable=False, index=True)
+    used = Column(Boolean, default=False, index=True)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    
+    # Relationships
+    user = relationship("User")
+    
+    def __repr__(self):
+        return f"<OTPCode(id={self.id}, user_id={self.user_id}, code={self.code[:2]}**, used={self.used})>"
+