83 lines
4.3 KiB
Python
83 lines
4.3 KiB
Python
"""Add subscription model
|
|
|
|
Revision ID: 9283c3f52a76
|
|
Revises: f9cf173f48f9
|
|
Create Date: 2026-04-06 14:26:41.926020
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
from app.core.migration_helpers import drop_foreign_key_on
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision: str = '9283c3f52a76'
|
|
down_revision: Union[str, Sequence[str], None] = 'f9cf173f48f9'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
op.create_table('subscription_plans',
|
|
sa.Column('id', sa.UUID(), nullable=False),
|
|
sa.Column('name', sa.String(), nullable=False),
|
|
sa.Column('description', sa.String(), nullable=True),
|
|
sa.Column('price', sa.Numeric(precision=10, scale=2), nullable=True),
|
|
sa.Column('is_public', sa.Boolean(), nullable=True),
|
|
sa.Column('status', sa.String(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_subscription_plans_id'), 'subscription_plans', ['id'], unique=False)
|
|
op.create_index(op.f('ix_subscription_plans_name'), 'subscription_plans', ['name'], unique=True)
|
|
op.create_table('plan_accesses',
|
|
sa.Column('id', sa.UUID(), nullable=False),
|
|
sa.Column('plan_id', sa.UUID(), nullable=False),
|
|
sa.Column('access_id', sa.UUID(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=True),
|
|
sa.ForeignKeyConstraint(['access_id'], ['accesses.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['plan_id'], ['subscription_plans.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('plan_id', 'access_id', name='uq_plan_access')
|
|
)
|
|
op.create_index(op.f('ix_plan_accesses_access_id'), 'plan_accesses', ['access_id'], unique=False)
|
|
op.create_index(op.f('ix_plan_accesses_id'), 'plan_accesses', ['id'], unique=False)
|
|
op.create_index(op.f('ix_plan_accesses_plan_id'), 'plan_accesses', ['plan_id'], unique=False)
|
|
op.create_table('plan_module_accesses',
|
|
sa.Column('id', sa.UUID(), nullable=False),
|
|
sa.Column('plan_id', sa.UUID(), nullable=False),
|
|
sa.Column('module_access_id', sa.UUID(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=True),
|
|
sa.ForeignKeyConstraint(['module_access_id'], ['module_accesses.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['plan_id'], ['subscription_plans.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('plan_id', 'module_access_id', name='uq_plan_module_access')
|
|
)
|
|
op.create_index(op.f('ix_plan_module_accesses_id'), 'plan_module_accesses', ['id'], unique=False)
|
|
op.create_index(op.f('ix_plan_module_accesses_module_access_id'), 'plan_module_accesses', ['module_access_id'], unique=False)
|
|
op.create_index(op.f('ix_plan_module_accesses_plan_id'), 'plan_module_accesses', ['plan_id'], unique=False)
|
|
op.add_column('tenants', sa.Column('plan_id', sa.UUID(), nullable=True))
|
|
op.create_index(op.f('ix_tenants_plan_id'), 'tenants', ['plan_id'], unique=False)
|
|
op.create_foreign_key(None, 'tenants', 'subscription_plans', ['plan_id'], ['id'])
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
drop_foreign_key_on("tenants", "plan_id")
|
|
op.drop_index(op.f('ix_tenants_plan_id'), table_name='tenants')
|
|
op.drop_column('tenants', 'plan_id')
|
|
op.drop_index(op.f('ix_plan_module_accesses_plan_id'), table_name='plan_module_accesses')
|
|
op.drop_index(op.f('ix_plan_module_accesses_module_access_id'), table_name='plan_module_accesses')
|
|
op.drop_index(op.f('ix_plan_module_accesses_id'), table_name='plan_module_accesses')
|
|
op.drop_table('plan_module_accesses')
|
|
op.drop_index(op.f('ix_plan_accesses_plan_id'), table_name='plan_accesses')
|
|
op.drop_index(op.f('ix_plan_accesses_id'), table_name='plan_accesses')
|
|
op.drop_index(op.f('ix_plan_accesses_access_id'), table_name='plan_accesses')
|
|
op.drop_table('plan_accesses')
|
|
op.drop_index(op.f('ix_subscription_plans_name'), table_name='subscription_plans')
|
|
op.drop_index(op.f('ix_subscription_plans_id'), table_name='subscription_plans')
|
|
op.drop_table('subscription_plans')
|