Ejemplo n.º 1
0
class Queue_Animal_array(object):
    def __init__(self):
        self.cat = Queue()
        self.dog = Queue()
        self.count = 0

    def animal_enqueue(self, item):
        animal = Animal(item, self.count)
        if animal.cat_or_dog() == "cat":
            self.cat.enqueue(animal)
            self.count += 1
        elif animal.cat_or_dog() == "dog":
            self.dog.enqueue(animal)
            self.count += 1
        else:
            return None
    
    def animal_dequeueAny(self):
        if self.cat.isEmpty() and self.dog.isEmpty():
            return None
        elif self.cat.isEmpty() and not self.dog.isEmpty():
            return self.dog.dequeue()
        elif not self.cat.isEmpty() and self.dog.isEmpty():
            return self.cat.dequeue()
        else:
            cat = self.cat.peak().id
            dog = self.dog.peak().id
            if cat<dog:
                return self.cat.dequeue()
            else:
                return self.dog.dequeue()
    
    def animal_dequeueCat(self):
        if self.cat.isEmpty():
            return None
        else:
            return self.cat.dequeue()
        
    def animal_dequeueDog(self):
        if self.dog.isEmpty():
            return None
        else:
            return self.dog.dequeue()