from queue import Queue


class WaitingQueue:
    def __init__(self):
        self.queue = Queue(maxsize=500)

    def enqueue(self,uid, video_path,audio_path):
        data = {
            "uid": uid,
            "video_path": video_path,
            "audio_path": audio_path
        }
        self.queue.put(data,timeout=10)

    def dequeue(self):
        data = self.queue.get(timeout=10)
        return data["uid"], data["video_path"], data["audio_path"]

    def get_size(self):
        return self.queue.qsize()