# Copyright: (c) OpenSpug Organization. https://github.com/openspug/spug
# Copyright: (c) <spug.dev@gmail.com>
# Released under the AGPL-3.0 License.
from django.db import close_old_connections
from apps.account.models import History
from apps.alarm.models import Alarm
from apps.schedule.models import Task
from apps.deploy.models import DeployRequest
from apps.exec.models import ExecHistory
from apps.deploy.utils import dispatch
from libs.utils import parse_time, human_datetime
from datetime import datetime, timedelta
from threading import Thread


def auto_run_by_day():
    close_old_connections()
    date = datetime.now() - timedelta(days=30)
    History.objects.filter(created_at__lt=date.strftime('%Y-%m-%d')).delete()
    Alarm.objects.filter(created_at__lt=date.strftime('%Y-%m-%d')).delete()
    try:
        record = ExecHistory.objects.all()[50]
        ExecHistory.objects.filter(id__lt=record.id).delete()
    except IndexError:
        pass
    for task in Task.objects.all():
        try:
            record = History.objects.filter(task_id=task.id)[50]
            History.objects.filter(task_id=task.id, id__lt=record.id).delete()
        except IndexError:
            pass


def auto_run_by_minute():
    close_old_connections()
    now = datetime.now()
    for req in DeployRequest.objects.filter(status='2'):
        if (now - parse_time(req.do_at)).seconds > 3600:
            req.status = '-3'
            req.save()
    for req in DeployRequest.objects.filter(status='1', plan__lte=now):
        req.status = '2'
        req.do_at = human_datetime()
        req.do_by = req.created_by
        req.save()
        Thread(target=dispatch, args=(req,)).start()