Source code for rate_control.queues._lifo

__all__ = [
    'LifoQueue',
]

import sys
from typing import Any, TypeVar

from rate_control._errors import Empty
from rate_control._helpers import mk_repr
from rate_control.queues._abc import Queue

if sys.version_info >= (3, 12):
    from typing import override
else:
    from typing_extensions import override


_T = TypeVar('_T')


class LifoQueue(Queue[_T]):
    """ "Last In, First Out" queue."""

    __slots__ = ('_queue',)

    def __init__(self, *elements: _T, **kwargs: Any) -> None:
        """
        Args:
            elements: The elements to initialize the queue with.
        """
        self._queue = list(elements)
        super().__init__(**kwargs)

    @override
    def __repr__(self) -> str:
        return mk_repr(self, *reversed(self._queue))

    @override
    def __bool__(self) -> bool:
        return bool(self._queue)

    @override
    def head(self) -> _T:
        try:
            return self._queue[-1]
        except IndexError as e:
            raise Empty from e

    @override
    def pop(self) -> _T:
        try:
            return self._queue.pop()
        except IndexError as e:
            raise Empty from e

    @override
    def add(self, element: _T) -> None:
        self._queue.append(element)

    @override
    def remove(self, element: _T) -> None:
        self._queue.remove(element)