최대 용량이 정해진 FIFO 큐 클래스

오호 스택 두개로 FIFO 큐를 구현할 수 있구만 몰랐던 사실!

class MyStack(object):
    def __init__(self):
        self.lst = list()

    def push(self, x):
        self.lst.append(x)

    def pop(self):
        return self.lst.pop()

    def size(self):
        return len(self.lst)


class MyQueue(object):
    def __init__(self, max_size):
        self.stack1 = MyStack()
        self.stack2 = MyStack()
        self.max_size = max_size

    def qsize(self):
        # 큐가 가진 원소 수를 리턴합니다.
        return self.stack1.size()

    def push(self, item):
        # 입력받은 인자 x를 큐에 넣습니다.
        # 단, 현재 큐가 꽉찬 경우 인자를 넣지말고 False를 리턴하세요.
        if self.stack1.size() == self.max_size:
            return False
        else:
            self.stack1.push(item)
            return True

    def pop(self):
        # 큐가 가진 원소 중, 가장 처음에 들어온 원소를 큐에서 제거하고 리턴합니다.
        # 큐에 원소가 없다면 Empty Exception을 raise 하세요.
        # Empty Exception은 본인이 직접 만드셔야 합니다.
        try:
            if self.stack1.size() == 0:
                raise emptyException

            while self.stack1.size() > 0:
                self.stack2.push(self.stack1.pop())

            item = self.stack2.pop()

            while self.stack2.size() > 0:
                self.stack1.push(self.stack2.pop())

            return item

        except emptyException:
            return False


class emptyException(Exception):
    pass


n, max_size = map(int, input().strip().split(' '))
q = MyQueue(max_size)
for i in range(n):
    order = input()

    if order == "POP":
        print(q.pop())
    elif order == "SIZE":
        print(q.qsize())
    else:
        order, a = order.strip().split(' ')
        print(q.push(int(a)))