Class ObjectPool

patterns/creational/pool.py:35–58

class ObjectPool:
    def __init__(self, queue: Queue, auto_get: bool = False) -> None:
        self._queue = queue
        self.item = self._queue.get() if auto_get else None

    def __enter__(self) -> str:
        if self.item is None:
            self.item = self._queue.get()
        return self.item

    def __exit__(
        self,
        Type: Union[type[BaseException], None],
        value: Union[BaseException, None],
        traceback: Union[TracebackType, None],
    ) -> None:
        if self.item is not None:
            self._queue.put(self.item)
            self.item = None

    def __del__(self) -> None:
        if self.item is not None:
            self._queue.put(self.item)
            self.item = None


def main():