from django.core.management.base import BaseCommand
from service.utils import run_sql_command, log
from service.models import TaskStatus
from postamates.settings import STATUS_TASK_NAME, STATUS_TASK_NAME_IMPORT


class Command(BaseCommand):
    help = 'Kill update processes'

    def handle(self, *args, **kwargs):
        TaskStatus.objects.filter(task_name=STATUS_TASK_NAME).update(status='Перерасчет ML завершен')
        TaskStatus.objects.filter(task_name=STATUS_TASK_NAME_IMPORT).update(status='Перерасчет ML завершен')