"""Definition of ConnectionAwareInterface abstract base class."""

from __future__ import annotations

from abc import ABC, abstractmethod


class ContextAwareInterface(ABC):
    """Interface that defined context-related methods."""

    @abstractmethod
    def __enter__(self) -> ContextAwareInterface:
        """Enter the context."""
        raise NotImplementedError()

    @abstractmethod
    def __exit__(
        self, exc_type: type | None, exc_val: object | None, exc_tb: object | None
    ) -> None:
        """Exit the context."""
        raise NotImplementedError()