首页 > 代码库 > Python 使用list实现队列 (基于class, 包含迭代器)

Python 使用list实现队列 (基于class, 包含迭代器)

#!/usr/bin/python 
# -*- coding: utf-8 -*-

'''
Created on 2015-1-27
@author: beyondzhou
@name: test_listqueue.py
'''

def test_listqueue():
    
    # import pyListQueue
    from myqueue import pyListQueue
    
    print '#Init a queue named smith using enqueue'
    smith = pyListQueue()
    smith.enqueue('CSCI-112')
    smith.enqueue('MATH-121')
    smith.enqueue('HIST-340')
    smith.enqueue('ECON-101')
    
    print '\n#output smith queue'
    for element in smith:
        print element
           
    print '\n#dequeue one item'
    smith.dequeue()
    
    print '\n#output smith after dequeue'
    for element in smith:
        print element 
        
    print '\n#get the length of queue'
    print 'the lenght of queue is ', len(smith)
    
    print '\n#check wheter the queue is empty'
    if smith.isEmpty():
        print 'queue is empty!'
    else:
        print 'queue is not empty!'
        
    print '\n#dequeue all items'
    while not smith.isEmpty():
        smith.dequeue()
    
    print '\n#check wheter the queue is empty after dequeue all items'
    if smith.isEmpty():
        print 'queue is empty!'
    else:
        print 'queue is not empty!'
    
if __name__ == "__main__":
    test_listqueue()

# Implementation of iter
class _pyListQueueIterator:
    def __init__(self, theList):
        self._setItems = theList
        self._curItem = 0
    def __iter__(self):
        return self
    def next(self):
        if self._curItem < len(self._setItems):
            item = self._setItems[self._curItem]
            self._curItem += 1
            return item
        else:
            raise StopIteration
        
# Implementation of the Queue ADT using a Python list
class pyListQueue:
    # Creates an empty queue
    def __init__(self):
        self._qList = list()

    # Returns True if the queue is empty
    def isEmpty(self):
        return len(self) == 0

    # Returns the number of items in the queue
    def __len__(self):
        return len(self._qList)

    # Adds the given item to the queue
    def enqueue(self, item):
        self._qList.append(item)

    # Removes and returns the first item in the queue
    def dequeue(self):
        assert not self.isEmpty(), "Cannot dequeue from an empty queue."
        return self._qList.pop(0)

    # Returns an iterator for traversing the list of items
    def __iter__(self):
        return _pyListQueueIterator(self._qList)

#Init a queue named smith using enqueue

#output smith queue
CSCI-112
MATH-121
HIST-340
ECON-101

#dequeue one item

#output smith after dequeue
MATH-121
HIST-340
ECON-101

#get the length of queue
the lenght of queue is  3

#check wheter the queue is empty
queue is not empty!

#dequeue all items

#check wheter the queue is empty after dequeue all items
queue is empty!

Python 使用list实现队列 (基于class, 包含迭代器)