115 lines
5.3 KiB
Python
115 lines
5.3 KiB
Python
"""create_gdpr_requests_table
|
|
|
|
Revision ID: d709b14aa24a
|
|
Revises: b1c4d7c154ec
|
|
Create Date: 2025-12-06 01:10:15.233886
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import mysql
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = 'd709b14aa24a'
|
|
down_revision = 'b1c4d7c154ec'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Check if table exists before creating
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
tables = inspector.get_table_names()
|
|
|
|
if 'gdpr_requests' not in tables:
|
|
# Create GDPR requests table
|
|
op.create_table(
|
|
'gdpr_requests',
|
|
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column('request_type', sa.Enum('data_export', 'data_deletion', 'data_rectification', 'consent_withdrawal', name='gdprrequesttype'), nullable=False),
|
|
sa.Column('status', sa.Enum('pending', 'processing', 'completed', 'rejected', 'cancelled', name='gdprrequeststatus'), nullable=False, server_default='pending'),
|
|
sa.Column('user_id', sa.Integer(), nullable=True),
|
|
sa.Column('user_email', sa.String(length=255), nullable=False),
|
|
sa.Column('is_anonymous', sa.Boolean(), nullable=False, server_default='0'),
|
|
sa.Column('request_data', sa.JSON(), nullable=True),
|
|
sa.Column('verification_token', sa.String(length=255), nullable=True),
|
|
sa.Column('verified_at', sa.DateTime(), nullable=True),
|
|
sa.Column('processed_by', sa.Integer(), nullable=True),
|
|
sa.Column('processed_at', sa.DateTime(), nullable=True),
|
|
sa.Column('processing_notes', sa.Text(), nullable=True),
|
|
sa.Column('export_file_path', sa.String(length=500), nullable=True),
|
|
sa.Column('deletion_log', sa.JSON(), nullable=True),
|
|
sa.Column('ip_address', sa.String(length=45), nullable=True),
|
|
sa.Column('user_agent', sa.String(length=255), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP')),
|
|
sa.Column('expires_at', sa.DateTime(), nullable=True),
|
|
sa.ForeignKeyConstraint(['processed_by'], ['users.id'], ),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
|
|
# Create indexes
|
|
op.create_index(op.f('ix_gdpr_requests_id'), 'gdpr_requests', ['id'], unique=False)
|
|
op.create_index(op.f('ix_gdpr_requests_request_type'), 'gdpr_requests', ['request_type'], unique=False)
|
|
op.create_index(op.f('ix_gdpr_requests_status'), 'gdpr_requests', ['status'], unique=False)
|
|
op.create_index(op.f('ix_gdpr_requests_user_id'), 'gdpr_requests', ['user_id'], unique=False)
|
|
op.create_index(op.f('ix_gdpr_requests_is_anonymous'), 'gdpr_requests', ['is_anonymous'], unique=False)
|
|
op.create_index(op.f('ix_gdpr_requests_created_at'), 'gdpr_requests', ['created_at'], unique=False)
|
|
op.create_index(op.f('ix_gdpr_requests_verification_token'), 'gdpr_requests', ['verification_token'], unique=True)
|
|
else:
|
|
# Table already exists, ensure it has the is_anonymous column if missing
|
|
columns = [col['name'] for col in inspector.get_columns('gdpr_requests')]
|
|
if 'is_anonymous' not in columns:
|
|
op.add_column('gdpr_requests', sa.Column('is_anonymous', sa.Boolean(), nullable=False, server_default='0'))
|
|
op.create_index(op.f('ix_gdpr_requests_is_anonymous'), 'gdpr_requests', ['is_anonymous'], unique=False)
|
|
|
|
# Ensure user_id is nullable
|
|
# Note: This might fail if there are existing non-null values, but for a new table it should be fine
|
|
try:
|
|
op.alter_column('gdpr_requests', 'user_id', existing_type=sa.Integer(), nullable=True)
|
|
except Exception:
|
|
pass # Column might already be nullable
|
|
|
|
|
|
def downgrade() -> None:
|
|
# Drop indexes
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
tables = inspector.get_table_names()
|
|
|
|
if 'gdpr_requests' in tables:
|
|
try:
|
|
op.drop_index(op.f('ix_gdpr_requests_verification_token'), table_name='gdpr_requests')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_index(op.f('ix_gdpr_requests_created_at'), table_name='gdpr_requests')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_index(op.f('ix_gdpr_requests_is_anonymous'), table_name='gdpr_requests')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_index(op.f('ix_gdpr_requests_user_id'), table_name='gdpr_requests')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_index(op.f('ix_gdpr_requests_status'), table_name='gdpr_requests')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_index(op.f('ix_gdpr_requests_request_type'), table_name='gdpr_requests')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_index(op.f('ix_gdpr_requests_id'), table_name='gdpr_requests')
|
|
except Exception:
|
|
pass
|
|
|
|
# Drop table
|
|
op.drop_table('gdpr_requests')
|