from django.core.management.base import BaseCommand
from django.db import connection


def _pg(cursor):
    cursor.execute("""
    DROP TRIGGER IF EXISTS update_alert_after ON api_check;
    """)


def _mysql(cursor):
    cursor.execute("""
    DROP TRIGGER IF EXISTS update_alert_after;
    """)


def _sqlite(cursor):
    cursor.execute("""
    DROP TRIGGER IF EXISTS update_alert_after;
    """)


class Command(BaseCommand):
    help = 'Drops the `update_alert_after` trigger'
    requires_system_checks = False

    def handle(self, *args, **options):
        with connection.cursor() as cursor:
            if connection.vendor == "postgresql":
                _pg(cursor)
                return "Dropped PostgreSQL trigger"
            if connection.vendor == "mysql":
                _mysql(cursor)
                return "Dropped MySQL trigger"
            if connection.vendor == "sqlite":
                _sqlite(cursor)
                return "Dropped SQLite trigger"