models.py

from flask_sqlalchemy import SQLAlchemy
from datetime import datetime

db = SQLAlchemy()

class File(db.Model):
    __tablename__ = 'files'
    id = db.Column(db.Integer, primary_key=True)
    filename = db.Column(db.String(512), nullable=False)
    file_id = db.Column(db.String(255), nullable=False)
    folder = db.Column(db.String(512), nullable=False, index=True)
    size = db.Column(db.BigInteger) # 使用 BigInteger 支援大檔案
    mime_type = db.Column(db.String(100))
    upload_date = db.Column(db.DateTime, default=datetime.utcnow)
    thumbnail_file_id = db.Column(db.String(255))
    is_deleted = db.Column(db.Boolean, nullable=False, default=False, index=True)
    cover_file_id = db.Column(db.String(255))
    message_link = db.Column(db.String(1024))

class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(100), unique=True, nullable=False)
    password = db.Column(db.String(255), nullable=False)
    paths = db.relationship('UserPath', backref='user', cascade="all, delete-orphan")

class UserPath(db.Model):
    __tablename__ = 'user_paths'
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
    path = db.Column(db.String(512), nullable=False)

class UploadTask(db.Model):
    __tablename__ = 'upload_tasks'
    id = db.Column(db.Integer, primary_key=True)
    filename = db.Column(db.String(512), nullable=False)
    temp_file_path = db.Column(db.String(1024), nullable=False)
    target_folder = db.Column(db.String(512), nullable=False)
    file_size = db.Column(db.BigInteger) # 使用 BigInteger
    mime_type = db.Column(db.String(100))
    thumbnail_path = db.Column(db.String(1024))
    api_used = db.Column(db.String(50), nullable=False)
    status = db.Column(db.String(50), nullable=False, default='queued', index=True)
    priority = db.Column(db.Integer, default=0)
    error_message = db.Column(db.String(2048))
    uploaded_bytes = db.Column(db.BigInteger, default=0) # 已上傳字節
    speed = db.Column(db.String(50)) # 上傳速度
    telegram_file_id = db.Column(db.String(255))
    telegram_message_link = db.Column(db.String(1024))
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    def to_dict(self):
        return {
            'id': self.id,
            'filename': self.filename,
            'temp_file_path': self.temp_file_path,
            'target_folder': self.target_folder,
            'file_size': self.file_size,
            'mime_type': self.mime_type,
            'thumbnail_path': self.thumbnail_path,
            'api_used': self.api_used,
            'status': self.status,
            'priority': self.priority,
            'error_message': self.error_message,
            'uploaded_bytes': self.uploaded_bytes,
            'speed': self.speed,
            'telegram_file_id': self.telegram_file_id,
            'telegram_message_link': self.telegram_message_link,
            'created_at': self.created_at.isoformat() if self.created_at else None,
            'updated_at': self.updated_at.isoformat() if self.updated_at else None,
        }