quick commit

migrations/versions/20260117_add_database_indexes.py

@@ -0,0 +1,125 @@
+"""Add database indexes for performance and security.
+
+Revision ID: 20260117_add_database_indexes
+Revises: 20260117_add_automod_thresholds
+Create Date: 2026-01-17 12:00:00.000000
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+# revision identifiers, used by Alembic.
+revision = "20260117_add_database_indexes"
+down_revision = "20260117_add_automod_thresholds"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Add indexes for common query patterns and performance optimization."""
+    
+    # Indexes for moderation_logs table
+    # Primary lookup patterns: by guild, by target user, by moderator, by timestamp
+    op.create_index("idx_moderation_logs_guild_id", "moderation_logs", ["guild_id"])
+    op.create_index("idx_moderation_logs_target_id", "moderation_logs", ["target_id"])
+    op.create_index("idx_moderation_logs_moderator_id", "moderation_logs", ["moderator_id"])
+    op.create_index("idx_moderation_logs_created_at", "moderation_logs", ["created_at"])
+    op.create_index("idx_moderation_logs_action", "moderation_logs", ["action"])
+    op.create_index("idx_moderation_logs_is_automatic", "moderation_logs", ["is_automatic"])
+    
+    # Compound indexes for common filtering patterns
+    op.create_index("idx_moderation_logs_guild_target", "moderation_logs", ["guild_id", "target_id"])
+    op.create_index("idx_moderation_logs_guild_created", "moderation_logs", ["guild_id", "created_at"])
+    op.create_index("idx_moderation_logs_target_created", "moderation_logs", ["target_id", "created_at"])
+    
+    # Indexes for strikes table
+    # Primary lookup patterns: by guild, by user, active strikes, expiration
+    op.create_index("idx_strikes_guild_id", "strikes", ["guild_id"])
+    op.create_index("idx_strikes_user_id", "strikes", ["user_id"])
+    op.create_index("idx_strikes_moderator_id", "strikes", ["moderator_id"])
+    op.create_index("idx_strikes_is_active", "strikes", ["is_active"])
+    op.create_index("idx_strikes_expires_at", "strikes", ["expires_at"])
+    op.create_index("idx_strikes_created_at", "strikes", ["created_at"])
+    
+    # Compound indexes for active strike counting and user history
+    op.create_index("idx_strikes_guild_user_active", "strikes", ["guild_id", "user_id", "is_active"])
+    op.create_index("idx_strikes_user_active", "strikes", ["user_id", "is_active"])
+    op.create_index("idx_strikes_guild_active", "strikes", ["guild_id", "is_active"])
+    
+    # Indexes for banned_words table
+    # Primary lookup patterns: by guild, by pattern (for admin management)
+    op.create_index("idx_banned_words_guild_id", "banned_words", ["guild_id"])
+    op.create_index("idx_banned_words_is_regex", "banned_words", ["is_regex"])
+    op.create_index("idx_banned_words_action", "banned_words", ["action"])
+    op.create_index("idx_banned_words_added_by", "banned_words", ["added_by"])
+    
+    # Compound index for guild-specific lookups
+    op.create_index("idx_banned_words_guild_regex", "banned_words", ["guild_id", "is_regex"])
+    
+    # Indexes for user_notes table (if it exists)
+    # Primary lookup patterns: by guild, by user, by moderator
+    op.create_index("idx_user_notes_guild_id", "user_notes", ["guild_id"])
+    op.create_index("idx_user_notes_user_id", "user_notes", ["user_id"])
+    op.create_index("idx_user_notes_moderator_id", "user_notes", ["moderator_id"])
+    op.create_index("idx_user_notes_created_at", "user_notes", ["created_at"])
+    
+    # Compound indexes for user note history
+    op.create_index("idx_user_notes_guild_user", "user_notes", ["guild_id", "user_id"])
+    op.create_index("idx_user_notes_user_created", "user_notes", ["user_id", "created_at"])
+    
+    # Indexes for guild_settings table
+    # These are mostly for admin dashboard filtering
+    op.create_index("idx_guild_settings_automod_enabled", "guild_settings", ["automod_enabled"])
+    op.create_index("idx_guild_settings_ai_enabled", "guild_settings", ["ai_moderation_enabled"])
+    op.create_index("idx_guild_settings_verification_enabled", "guild_settings", ["verification_enabled"])
+    
+    # Indexes for guilds table
+    op.create_index("idx_guilds_owner_id", "guilds", ["owner_id"])
+    op.create_index("idx_guilds_premium", "guilds", ["premium"])
+    op.create_index("idx_guilds_created_at", "guilds", ["created_at"])
+
+
+def downgrade() -> None:
+    """Remove the indexes."""
+    
+    # Remove all indexes in reverse order
+    op.drop_index("idx_guilds_created_at")
+    op.drop_index("idx_guilds_premium")
+    op.drop_index("idx_guilds_owner_id")
+    
+    op.drop_index("idx_guild_settings_verification_enabled")
+    op.drop_index("idx_guild_settings_ai_enabled")
+    op.drop_index("idx_guild_settings_automod_enabled")
+    
+    op.drop_index("idx_user_notes_user_created")
+    op.drop_index("idx_user_notes_guild_user")
+    op.drop_index("idx_user_notes_created_at")
+    op.drop_index("idx_user_notes_moderator_id")
+    op.drop_index("idx_user_notes_user_id")
+    op.drop_index("idx_user_notes_guild_id")
+    
+    op.drop_index("idx_banned_words_guild_regex")
+    op.drop_index("idx_banned_words_added_by")
+    op.drop_index("idx_banned_words_action")
+    op.drop_index("idx_banned_words_is_regex")
+    op.drop_index("idx_banned_words_guild_id")
+    
+    op.drop_index("idx_strikes_guild_active")
+    op.drop_index("idx_strikes_user_active")
+    op.drop_index("idx_strikes_guild_user_active")
+    op.drop_index("idx_strikes_created_at")
+    op.drop_index("idx_strikes_expires_at")
+    op.drop_index("idx_strikes_is_active")
+    op.drop_index("idx_strikes_moderator_id")
+    op.drop_index("idx_strikes_user_id")
+    op.drop_index("idx_strikes_guild_id")
+    
+    op.drop_index("idx_moderation_logs_target_created")
+    op.drop_index("idx_moderation_logs_guild_created")
+    op.drop_index("idx_moderation_logs_guild_target")
+    op.drop_index("idx_moderation_logs_is_automatic")
+    op.drop_index("idx_moderation_logs_action")
+    op.drop_index("idx_moderation_logs_created_at")
+    op.drop_index("idx_moderation_logs_moderator_id")
+    op.drop_index("idx_moderation_logs_target_id")
+    op.drop_index("idx_moderation_logs_guild_id")