threadedqueue.py

from threading import Lock, Condition
import _thread
import Queue
import time


class ThreadedQueue:
    def __init__(self):
        self.items = []
        self.globallock = Lock()
        self.globalcond = Condition(self.globallock)

    def get(self, wait=-1):
        starttime = time.time()
        id = _thread.get_ident()
        v = self.items
        with self.globallock:
            for i in range(0, len(v)):
                u = v[i]
                if id == u[0]:
                    del v[i]
                    return u[1]
            while True:
                v = time.time() - starttime
                if (v >= wait) & (wait >= 0):
                    raise Queue.Empty
                if wait >= 0:
                    self.globalcond.wait(wait - v)
                else:
                    self.globalcond.wait()
                if len(self.items) == 0:
                    continue
                w = self.items[-1]
                if w[0] == id:
                    return w[1]

    def get_nowait(self):
        return self.get(0)

    def put(self, item, threadid):
        with self.globallock:
            self.items.append((threadid, item))
            self.globalcond.notifyAll()