92 lines
3.5 KiB
Python
92 lines
3.5 KiB
Python
"""
|
|
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(BigInteger, 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(BigInteger, 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(BigInteger, 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})>"
|
|
|