from django.core.management.base import BaseCommand
from service.utils import run_sql_command, log

CMD_PIVOT_DIST = """DROP MATERIALIZED VIEW IF EXISTS public.points_with_dist;"""

CMD_PIVOT_DIST_PRE = """DROP MATERIALIZED VIEW IF EXISTS public.prepoints_with_dist;"""
class Command(BaseCommand):
    help = 'delete views'

    def handle(self, *args, **kwargs):
        try:
            log('Deleting views')
            run_sql_command(CMD_PIVOT_DIST)
            log('pivot_dist deleted')
            run_sql_command(CMD_PIVOT_DIST_PRE)
            log('prepivot_dist deleted')
        except Exception as e:
            log('Error deleting views: ' + str(e))