heapThreadSafe.py

import heapq
import threading

class HeapThreadSafe(object):
    def __init__(self):
        self.sem = threading.Semaphore(1)
        self.heap = list()

    def push(self, obj):
        self.sem.acquire()
        heapq.heappush(self.heap, obj)
        self.sem.release()

    def pop(self):
        to_ret = None
        self.sem.acquire()
        try:
            to_ret = heapq.heappop(self.heap)
        except:
            pass
        self.sem.release()
        return to_ret