greenclock add stop task

scheduler.py

import greenclock


class Scheduler(greenclock.Scheduler):
    pool = None
    greenlets = {}

    def run_tasks(self):
        '''
        Runs all assigned task in separate green threads. If the task should not be run, schedule it
        '''
        self.pool = Pool(len(self.tasks))
        for task in self.tasks:
            # Launch a green thread to schedule the task
            # A task will be managed by 2 green thread: execution thread and scheduling thread
            self.pool.spawn(self.do_run, task)
        return self.pool

    def do_run(self, task):
        greenlet, next_greenlet = self.run(task)
        self.greenlets.update({task.name: [greenlet, next_greenlet]})
        return greenlet, next_greenlet

    def stop_task(self, name):
        for key, task in enumerate(self.tasks):
            print key, task.name
            if task.name == name:
                greens = self.greenlets.pop(task.name, [])
                for g in greens:
                    if g is not None:
                        self.tasks.pop(key)
                        self.pool.killone(g)
                        self.pool.discard(g)
                        g.unlink(task.action)
                        g.kill()
                        print "Stopped task: %s" % task.name