Reversing First k Elements in Queue

reverse_first_n_queue.py

from Queue import MyQueue
from Stack import MyStack


def reverseK(queue: MyQueue, k: int):
    rem_size = queue.size() - k
    if rem_size < 0 or k < 1:
        return None
    stack = MyStack()
    for _ in range(k):
        stack.push(queue.dequeue())
    for _ in range(k):
        queue.enqueue(stack.pop())
    for _ in range(rem_size):
        queue.enqueue(queue.dequeue())
    
    return queue