stack.py

from typing import Generic, Iterator, TypeVar

T = TypeVar("T")


class Stack(Generic[T]):
    """Класс стека, реализующий стандартные операции работы со стеком
    Атрибуты:
        stack (list[T]): Список, используемый для хранения элементов стека.
    """

    def __init__(self) -> None:
        """Инициализирует пустой стек на основе списка."""
        self.stack: list[T] = []

    def push(self, element: T) -> None:
        """Добавляет элемент в вершину стека.

        Аргументы:
            element (Any): Элемент для добавления в стек.
        """
        self.stack.append(element)

    def pop(self) -> T:
        """Удаляет и возвращает элемент с вершины стека.

        Возвращает:
            Any: Элемент с вершины стека.

        Ошибка:
            IndexError: Если стек пуст.
        """
        if self.is_empty():
            raise IndexError("stack is empty")
        return self.stack.pop()

    def peek(self) -> T:
        """Возвращает элемент с вершины стека без его удаления.

        Возвращает:
            Any: Элемент с вершины стека.

        Ошибка:
            IndexError: Если стек пуст.
        """
        if self.is_empty():
            raise IndexError("stack is empty")
        return self.stack[-1]

    def is_empty(self) -> bool:
        """Проверяет, пуст ли стек.

        Возвращает:
            bool: True если стек пуст, False в противном случае.
        """
        return len(self.stack) == 0

    def size(self) -> int:
        """Возвращает текущий размер стека (количество элементов).

        Возвращает:
            int: Количество элементов в стеке.
        """
        return len(self.stack)

    def __iter__(self) -> Iterator[T]:
        """Реализует итерацию по элементам стека.

        Возвращает:
            Iterator[Any]: Итератор по элементам стека.
        """
        for element in self.stack:
            yield element