anti-plagiarism/services/api/app/models/admin.py

"""Модели для админ-панели: источники парсинга, отстойник работ, сессии админа."""

from datetime import datetime

from sqlalchemy import JSON, ForeignKey, String, func
from sqlalchemy.orm import Mapped, mapped_column

from app.database import Base


class ParseSource(Base):
    """Источник парсинга для наполнения базы документов.

    Админ задаёт тип источника и параметры запроса; задача index.run_parser
    запускает парсинг в фоне и наполняет базу.
    """

    __tablename__ = "parse_sources"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    # Тип парсера: openalex / cyberleninka / arxiv
    source_type: Mapped[str] = mapped_column(String(50), nullable=False, index=True)
    # Человекочитаемое имя
    name: Mapped[str] = mapped_column(String(255), nullable=False)
    query: Mapped[str | None] = mapped_column(nullable=True)
    lang: Mapped[str | None] = mapped_column(String(10), nullable=True)
    year_from: Mapped[int | None] = mapped_column(nullable=True)
    year_to: Mapped[int | None] = mapped_column(nullable=True)
    limit: Mapped[int] = mapped_column(default=1000)
    enabled: Mapped[bool] = mapped_column(default=True)
    # Статус последнего запуска: idle / running / done / error
    last_status: Mapped[str] = mapped_column(String(20), default="idle")
    last_error: Mapped[str | None] = mapped_column(nullable=True)
    last_run_at: Mapped[datetime | None] = mapped_column(nullable=True)
    docs_added: Mapped[int] = mapped_column(default=0)
    created_at: Mapped[datetime] = mapped_column(server_default=func.now())

    def __repr__(self) -> str:
        return f"<ParseSource id={self.id} type={self.source_type!r} name={self.name!r}>"


class StagedWork(Base):
    """Отстойник: проверенная пользовательская работа, ожидающая решения админа.

    После одобрения работа добавляется в базу документов (index.add_document),
    после отклонения — помечается rejected.
    """

    __tablename__ = "staged_works"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    user_id: Mapped[int | None] = mapped_column(ForeignKey("users.id"), nullable=True, index=True)
    # Внутренний UUID задачи проверки (Task.id)
    task_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    filename: Mapped[str | None] = mapped_column(nullable=True)
    # Ключ исходного файла и извлечённого текста в MinIO
    minio_key: Mapped[str | None] = mapped_column(nullable=True)
    text_key: Mapped[str | None] = mapped_column(nullable=True)
    title: Mapped[str | None] = mapped_column(nullable=True)
    authors: Mapped[list] = mapped_column(JSON, default=list)
    year: Mapped[int | None] = mapped_column(nullable=True)
    lang: Mapped[str | None] = mapped_column(String(10), nullable=True)
    word_count: Mapped[int | None] = mapped_column(nullable=True)
    # pending / approved / rejected
    status: Mapped[str] = mapped_column(String(20), default="pending", index=True)
    # ID документа в базе после одобрения
    document_id: Mapped[int | None] = mapped_column(nullable=True)
    reviewed_by: Mapped[int | None] = mapped_column(nullable=True)
    reviewed_at: Mapped[datetime | None] = mapped_column(nullable=True)
    created_at: Mapped[datetime] = mapped_column(server_default=func.now(), index=True)

    def __repr__(self) -> str:
        return f"<StagedWork id={self.id} status={self.status!r} title={(self.title or '')[:30]!r}>"


class AdminSession(Base):
    """Секретный код доступа к админке (часть ссылки /admin/<code>).

    Доступ проверяется по двум факторам: JWT админа (is_admin) И валидный код.
    """

    __tablename__ = "admin_sessions"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False, index=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True, nullable=False)
    created_at: Mapped[datetime] = mapped_column(server_default=func.now())
    expires_at: Mapped[datetime] = mapped_column(nullable=False)
    last_used_at: Mapped[datetime | None] = mapped_column(nullable=True)

    def __repr__(self) -> str:
        return f"<AdminSession id={self.id} user_id={self.user_id}>"