from __future__ import absolute_import

from celery.task import Task

from observation_manager.tasks import execute_plan

from pyrosapp.models import *

import time

DEFAULT_PLAN_NUMBER = 2
DEFAULT_COUNTDOWN = 1

class execute_sequence(Task):

    def run(self, shs_pk):

        print("ex seq : ", self.request.id)

        # ici, je fais les actions de début de séquence (slewing, intruments
        # configuration, etc)
        shs = ScheduleHasSequences.objects.get(pk=shs_pk)
        sequence = shs.sequence

        TaskId.objects.filter(task_id=self.request.id).delete()

        message = 'Start sequence ' + str(sequence.pk) + ' execution'

        Log.objects.create(agent='Majordome', message=message)

        print("execute_sequence : ", sequence.pk)
        countdown = DEFAULT_COUNTDOWN  # déterminer le vrai ...

        # je devrais peut-être faire un groupe à la place de plusieurs taches

        # le .get() du résultat d'un group est bloquant :)

        plan_id = 1  # simple valeur de test, il faudra envoyer le vrai ID

        plans_results = []
        for i in range(DEFAULT_PLAN_NUMBER):

            res = execute_plan.apply_async((str(sequence.pk) + ' - ' + str(plan_id),), countdown=countdown)
            TaskId.objects.create(task_id=res.id, task="execute_plan")
            plans_results.append(res)

            plan_id += 1

        for plan_result in plans_results:

            # attention, un revoke n'annule pas le countdown du get() (à
            # réfléchir)

            try:
                plan_result.get()
            except Exception:
                pass

        # en cas d'alerte reçue, on va aussi supprimer la séquence en cours d'exécution ; ça permettra de virer ce problème du countdown.
        # de toute façon, c'est le execute_plan qui se charge d'envoyer les images à l'analyse, donc osef que ça coupe ici
        # l'unique but de ne pas couper ici dès la création des plans est
        # d'attendre la fin d'une séquence avant d'en lancer une autre.
        message = 'Finished sequence ' + str(sequence.pk) + ' execution'

        Log.objects.create(agent='Majordome', message=message)

class system_pause(Task):

    def run(self):
        time.sleep(5)
        print("system_pause")


class system_restart(Task):

    def run(self):
        time.sleep(5)
        print("system_restart")


class change_obs_conditions(Task):

    def run(self):
        time.sleep(5)
        print("change_obs_conditions")