#!/usr/bin/env python2.6

import threading
import logging
import Queue

class _Worker(threading.Thread):
    def __init__(self, uid, pool):
        threading.Thread.__init__(self, name = "Worker%d" % uid)
        self.pool = pool
    def run(self):
        logging.info("%s started", self.getName())
        while True:
            job = self.pool.jobs.get()
            if job:
                logging.debug("%s is running %r", self.getName(), job)
                try:
                    job()
                except:
                    logging.exception("Job %r return an exception", job)
                logging.debug("Job %r completed.", job)
            else:
                self.pool.jobs.put(job)
                break
        logging.info("%s stopped", self.getName())


##
# A basic worker pool... should be replaced with a resizable one.
class ThreadPool(object):
    Worker = _Worker
    
    def __init__(self, size):
        super(ThreadPool, self).__init__()
        self.jobs = Queue.Queue()
        self._workers = [self.Worker(x+1, self) for x in xrange(size)]

    def Start(self):
        logging.info("Starting Workers pool")
        for worker in self._workers:
            worker.start()
        logging.info("Workers started")

    def Stop(self, now = False):
        logging.info("Stopping Workers")
        if now:
            self.jobs = Queue.Queue()
        self.jobs.put(None)
        for worker in self._workers:
            worker.join()
        logging.info("Workers stopped")

    def AddJob(self, job):
        self.jobs.put(job)