145 lines
6.1 KiB
Python
145 lines
6.1 KiB
Python
"""Initial migration
|
|
|
|
Revision ID: 001
|
|
Revises:
|
|
Create Date: 2023-10-01
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '001'
|
|
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:
|
|
# Create role table
|
|
op.create_table(
|
|
'role',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(length=50), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_role_id'), 'role', ['id'], unique=False)
|
|
op.create_index(op.f('ix_role_name'), 'role', ['name'], unique=True)
|
|
|
|
# Create user table
|
|
op.create_table(
|
|
'user',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('email', sa.String(length=100), nullable=False),
|
|
sa.Column('username', sa.String(length=50), nullable=False),
|
|
sa.Column('first_name', sa.String(length=50), nullable=True),
|
|
sa.Column('last_name', sa.String(length=50), nullable=True),
|
|
sa.Column('hashed_password', sa.String(length=255), nullable=False),
|
|
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.Column('role_id', sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(['role_id'], ['role.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_user_email'), 'user', ['email'], unique=True)
|
|
op.create_index(op.f('ix_user_id'), 'user', ['id'], unique=False)
|
|
op.create_index(op.f('ix_user_username'), 'user', ['username'], unique=True)
|
|
|
|
# Create teacher table
|
|
op.create_table(
|
|
'teacher',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('teacher_id', sa.String(length=20), nullable=False),
|
|
sa.Column('date_of_birth', sa.Date(), nullable=True),
|
|
sa.Column('address', sa.String(length=255), nullable=True),
|
|
sa.Column('phone_number', sa.String(length=20), nullable=True),
|
|
sa.Column('department', sa.String(length=50), nullable=True),
|
|
sa.Column('hire_date', sa.Date(), nullable=True),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(['user_id'], ['user.id'], ),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('user_id')
|
|
)
|
|
op.create_index(op.f('ix_teacher_id'), 'teacher', ['id'], unique=False)
|
|
op.create_index(op.f('ix_teacher_teacher_id'), 'teacher', ['teacher_id'], unique=True)
|
|
|
|
# Create student table
|
|
op.create_table(
|
|
'student',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('student_id', sa.String(length=20), nullable=False),
|
|
sa.Column('date_of_birth', sa.Date(), nullable=True),
|
|
sa.Column('address', sa.String(length=255), nullable=True),
|
|
sa.Column('phone_number', sa.String(length=20), nullable=True),
|
|
sa.Column('grade_level', sa.String(length=20), nullable=True),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(['user_id'], ['user.id'], ),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('user_id')
|
|
)
|
|
op.create_index(op.f('ix_student_id'), 'student', ['id'], unique=False)
|
|
op.create_index(op.f('ix_student_student_id'), 'student', ['student_id'], unique=True)
|
|
|
|
# Create course table
|
|
op.create_table(
|
|
'course',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('course_code', sa.String(length=20), nullable=False),
|
|
sa.Column('title', sa.String(length=100), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('credits', sa.Integer(), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=True),
|
|
sa.Column('teacher_id', sa.Integer(), nullable=True),
|
|
sa.ForeignKeyConstraint(['teacher_id'], ['teacher.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_course_course_code'), 'course', ['course_code'], unique=True)
|
|
op.create_index(op.f('ix_course_id'), 'course', ['id'], unique=False)
|
|
|
|
# Create class_enrollment table
|
|
op.create_table(
|
|
'classenrollment',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('enrollment_date', sa.DateTime(), nullable=True),
|
|
sa.Column('grade', sa.Float(), nullable=True),
|
|
sa.Column('semester', sa.String(length=20), nullable=False),
|
|
sa.Column('academic_year', sa.String(length=10), nullable=False),
|
|
sa.Column('student_id', sa.Integer(), nullable=False),
|
|
sa.Column('course_id', sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(['course_id'], ['course.id'], ),
|
|
sa.ForeignKeyConstraint(['student_id'], ['student.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_classenrollment_id'), 'classenrollment', ['id'], unique=False)
|
|
|
|
|
|
def downgrade() -> None:
|
|
# Drop tables in reverse order of creation
|
|
op.drop_index(op.f('ix_classenrollment_id'), table_name='classenrollment')
|
|
op.drop_table('classenrollment')
|
|
|
|
op.drop_index(op.f('ix_course_id'), table_name='course')
|
|
op.drop_index(op.f('ix_course_course_code'), table_name='course')
|
|
op.drop_table('course')
|
|
|
|
op.drop_index(op.f('ix_student_student_id'), table_name='student')
|
|
op.drop_index(op.f('ix_student_id'), table_name='student')
|
|
op.drop_table('student')
|
|
|
|
op.drop_index(op.f('ix_teacher_teacher_id'), table_name='teacher')
|
|
op.drop_index(op.f('ix_teacher_id'), table_name='teacher')
|
|
op.drop_table('teacher')
|
|
|
|
op.drop_index(op.f('ix_user_username'), table_name='user')
|
|
op.drop_index(op.f('ix_user_id'), table_name='user')
|
|
op.drop_index(op.f('ix_user_email'), table_name='user')
|
|
op.drop_table('user')
|
|
|
|
op.drop_index(op.f('ix_role_name'), table_name='role')
|
|
op.drop_index(op.f('ix_role_id'), table_name='role')
|
|
op.drop_table('role') |