"""
This model defines the database models to handle Celery AsyncResults.

"""
from __future__ import unicode_literals

from django.db import models
from django.utils.encoding import python_2_unicode_compatible
from django.utils.translation import ugettext as _

from gnuviechadmin.celery import app


class TaskResultManager(models.Manager):
    def create_task_result(self, asyncresult, name):
        taskresult = self.create(task_id=asyncresult.id, task_name=name)
        return taskresult


@python_2_unicode_compatible
class TaskResult(models.Model):
    task_id = models.CharField(_('Task id'), max_length=36)
    task_name = models.CharField(_('Task name'), max_length=64)
    result = models.TextField(_('Task result'))
    finished = models.BooleanField(default=False)
    state = models.CharField(_('Task state'), max_length=16)

    objects = TaskResultManager()

    class Meta:
        verbose_name = _('Task result')
        verbose_name_plural = _('Task results')

    def __str__(self):
        return "{task_name} ({task_id}): {finished}".format(
            task_name=self.task_name,
            task_id=self.task_id,
            finished=_('yes') if self.finished else _('no')
        )

    def fetch_result(self):
        if not self.finished:
            ar = app.AsyncResult(self.task_id)
            res = ar.get(no_ack=True, timeout=1)
            self.result = str(res)
            self.state = ar.state
            self.finished = True