b876f13fac
- FastAPI backend with JWT auth, roles (admin/moderator/user) - PDF upload (up to 500MB) with streaming, PyMuPDF text extraction - ChromaDB vectorization per page with metadata - LiteLLM AI question extraction from PDF (not generation) - Image extraction from PDF pages, graceful fallback - Quiz modes: timed (countdown timer) + learning (answers shown inline) - Page-by-page question navigation with dot navigator - TTS endpoint using LiteLLM (Google Vertex / OpenAI voices) - Admin dashboard: AI model management per task, user role management - Moderator role: upload PDFs, create sections, generate quizzes - Spaced repetition reminders via SMTP email (SM-2 intervals) - APScheduler daily reminder jobs - Celery + Redis for background PDF processing - React frontend with all pages - Docker Compose deployment (nginx + backend + celery + redis) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
142 lines
6.5 KiB
Python
142 lines
6.5 KiB
Python
"""initial tables
|
|
|
|
Revision ID: 97ef9b957a02
|
|
Revises:
|
|
Create Date: 2026-03-30 18:20:24.251948
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '97ef9b957a02'
|
|
down_revision: Union[str, None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('users',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('email', sa.String(), nullable=False),
|
|
sa.Column('hashed_password', sa.String(), nullable=False),
|
|
sa.Column('name', sa.String(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_users_email'), 'users', ['email'], unique=True)
|
|
op.create_index(op.f('ix_users_id'), 'users', ['id'], unique=False)
|
|
op.create_table('pdf_documents',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('filename', sa.String(), nullable=False),
|
|
sa.Column('original_filename', sa.String(), nullable=False),
|
|
sa.Column('total_pages', sa.Integer(), nullable=True),
|
|
sa.Column('status', sa.String(), nullable=True),
|
|
sa.Column('error_message', sa.String(), nullable=True),
|
|
sa.Column('uploaded_at', sa.DateTime(), nullable=True),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_pdf_documents_id'), 'pdf_documents', ['id'], unique=False)
|
|
op.create_table('sections',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('document_id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(), nullable=False),
|
|
sa.Column('start_page', sa.Integer(), nullable=False),
|
|
sa.Column('end_page', sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(['document_id'], ['pdf_documents.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_sections_id'), 'sections', ['id'], unique=False)
|
|
op.create_table('quizzes',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('section_id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('title', sa.String(), nullable=False),
|
|
sa.Column('questions_count', sa.Integer(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.ForeignKeyConstraint(['section_id'], ['sections.id'], ),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_quizzes_id'), 'quizzes', ['id'], unique=False)
|
|
op.create_table('questions',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('quiz_id', sa.Integer(), nullable=False),
|
|
sa.Column('question_text', sa.Text(), nullable=False),
|
|
sa.Column('question_type', sa.String(), nullable=False),
|
|
sa.Column('options', sa.JSON(), nullable=True),
|
|
sa.Column('correct_answer', sa.String(), nullable=False),
|
|
sa.Column('explanation', sa.Text(), nullable=True),
|
|
sa.Column('page_reference', sa.Integer(), nullable=True),
|
|
sa.ForeignKeyConstraint(['quiz_id'], ['quizzes.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_questions_id'), 'questions', ['id'], unique=False)
|
|
op.create_table('quiz_attempts',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('quiz_id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('score', sa.Integer(), nullable=True),
|
|
sa.Column('total_questions', sa.Integer(), nullable=True),
|
|
sa.Column('started_at', sa.DateTime(), nullable=True),
|
|
sa.Column('completed_at', sa.DateTime(), nullable=True),
|
|
sa.ForeignKeyConstraint(['quiz_id'], ['quizzes.id'], ),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_quiz_attempts_id'), 'quiz_attempts', ['id'], unique=False)
|
|
op.create_table('reminder_schedules',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('quiz_id', sa.Integer(), nullable=False),
|
|
sa.Column('next_reminder_at', sa.DateTime(), nullable=False),
|
|
sa.Column('interval_days', sa.Integer(), nullable=True),
|
|
sa.Column('performance_score', sa.Float(), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=True),
|
|
sa.ForeignKeyConstraint(['quiz_id'], ['quizzes.id'], ),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_reminder_schedules_id'), 'reminder_schedules', ['id'], unique=False)
|
|
op.create_table('attempt_answers',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('attempt_id', sa.Integer(), nullable=False),
|
|
sa.Column('question_id', sa.Integer(), nullable=False),
|
|
sa.Column('user_answer', sa.String(), nullable=False),
|
|
sa.Column('is_correct', sa.Boolean(), nullable=True),
|
|
sa.ForeignKeyConstraint(['attempt_id'], ['quiz_attempts.id'], ),
|
|
sa.ForeignKeyConstraint(['question_id'], ['questions.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_attempt_answers_id'), 'attempt_answers', ['id'], unique=False)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_index(op.f('ix_attempt_answers_id'), table_name='attempt_answers')
|
|
op.drop_table('attempt_answers')
|
|
op.drop_index(op.f('ix_reminder_schedules_id'), table_name='reminder_schedules')
|
|
op.drop_table('reminder_schedules')
|
|
op.drop_index(op.f('ix_quiz_attempts_id'), table_name='quiz_attempts')
|
|
op.drop_table('quiz_attempts')
|
|
op.drop_index(op.f('ix_questions_id'), table_name='questions')
|
|
op.drop_table('questions')
|
|
op.drop_index(op.f('ix_quizzes_id'), table_name='quizzes')
|
|
op.drop_table('quizzes')
|
|
op.drop_index(op.f('ix_sections_id'), table_name='sections')
|
|
op.drop_table('sections')
|
|
op.drop_index(op.f('ix_pdf_documents_id'), table_name='pdf_documents')
|
|
op.drop_table('pdf_documents')
|
|
op.drop_index(op.f('ix_users_id'), table_name='users')
|
|
op.drop_index(op.f('ix_users_email'), table_name='users')
|
|
op.drop_table('users')
|
|
# ### end Alembic commands ###
|