from functools import partial

from twisted.internet.defer import inlineCallbacks, returnValue 

from ..error import NoRouteForTask, RejectRecipient

from ..model.offer import offers
from ..model.job import exJobs

from .pool  import taskPool

class Task:
    '''one single part of a job.
    a task is one message to one recipient
    '''
    def __init__(self, recipient, job):
        self.recipient = recipient
        self.job = job
        self.status = None
        self.error = False

    def setStatus(self,status):
        self.status = status
        return status

    def setError(self,error):
        self.status = error
        self.error = True
        return error

    def start(self):
        self.d = taskPool.run(self._run)
        self.d.addCallback(self.setStatus)
        self.d.addCallback(partial(self.job.setStatus,self))
        self.d.addErrback(self.setError)
        self.d.addErrback(partial(self.job.setError,self))
        return self.d

    @inlineCallbacks
    def _run(self):
        for offer in self.job.offers:
            try:
                ret = yield offers[offer](self.recipient,self.job.message)
                returnValue(ret)
            except RejectRecipient:
                continue
        else:
            raise NoRouteForTask()


@inlineCallbacks
def createJob(user,recipients, msg, offers, info=None):
    job = yield exJobs.create(user, recipients, msg, offers, info)
    for r in recipients:
        task = Task(r,job)
        job.addTask(task)
        task.start()
    returnValue(job)