Method connection

psycopg_pool/psycopg_pool/pool.py:172–192

Context manager to obtain a connection from the pool. Return the connection immediately if available, otherwise wait up to *timeout* or `self.timeout` seconds and throw `PoolTimeout` if a connection is not available in time. Upon context exit, return the connection

(self, timeout: float | None = None)

    @contextmanager
    def connection(self, timeout: float | None = None) -> Iterator[CT]:
        """Context manager to obtain a connection from the pool.

        Return the connection immediately if available, otherwise wait up to
        *timeout* or `self.timeout` seconds and throw `PoolTimeout` if a
        connection is not available in time.

        Upon context exit, return the connection to the pool. Apply the normal
        :ref:`connection context behaviour <with-connection>` (commit/rollback
        the transaction in case of success/error). If the connection is no more
        in working state, replace it with a new one.
        """
        conn = self.getconn(timeout=timeout)
        try:
            t0 = monotonic()
            with conn:
                yield conn
        finally:
            self.putconn(conn)
            t1 = monotonic()
            self._stats[self._USAGE_MS] += int(1000.0 * (t1 - t0))

    def getconn(self, timeout: float | None = None) -> CT:
        """Obtain a connection from the pool.