Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Rename PostgresBackend to AsyncpgBackend
- Loading branch information
ansipunk
committed
Mar 2, 2024
1 parent
37c450c
commit 8cbcccb
Showing
5 changed files
with
154 additions
and
36 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,118 @@ | ||
| import typing | ||
| from collections.abc import Sequence | ||
|
|
||
| import psycopg_pool | ||
| from sqlalchemy.sql import ClauseElement | ||
|
|
||
| from databases.backends.dialects.psycopg import get_dialect | ||
| from databases.core import DatabaseURL | ||
| from databases.interfaces import ( | ||
| ConnectionBackend, | ||
| DatabaseBackend, | ||
| TransactionBackend, | ||
| ) | ||
|
|
||
|
|
||
| class PsycopgBackend(DatabaseBackend): | ||
| def __init__( | ||
| self, database_url: typing.Union[DatabaseURL, str], **options: typing.Any | ||
| ) -> None: | ||
| self._database_url = DatabaseURL(database_url) | ||
| self._options = options | ||
| self._dialect = get_dialect() | ||
| self._pool: typing.Optional[psycopg_pool.AsyncConnectionPool] = None | ||
|
|
||
| async def connect(self) -> None: | ||
| if self._pool is not None: | ||
| return | ||
|
|
||
| self._pool = psycopg_pool.AsyncConnectionPool( | ||
| self._database_url.url, open=False, **self._options) | ||
| await self._pool.open() | ||
|
|
||
| async def disconnect(self) -> None: | ||
| if self._pool is None: | ||
| return | ||
|
|
||
| await self._pool.close() | ||
| self._pool = None | ||
|
|
||
| def connection(self) -> "PsycopgConnection": | ||
| return PsycopgConnection(self) | ||
|
|
||
|
|
||
| class PsycopgConnection(ConnectionBackend): | ||
| def __init__(self, database: PsycopgBackend) -> None: | ||
| self._database = database | ||
|
|
||
| async def acquire(self) -> None: | ||
| if self._connection is not None: | ||
| return | ||
|
|
||
| if self._database._pool is None: | ||
| raise RuntimeError("PsycopgBackend is not running") | ||
|
|
||
| # TODO: Add configurable timeouts | ||
| self._connection = await self._database._pool.getconn() | ||
|
|
||
| async def release(self) -> None: | ||
| if self._connection is None: | ||
| return | ||
|
|
||
| await self._database._pool.putconn(self._connection) | ||
| self._connection = None | ||
|
|
||
| async def fetch_all(self, query: ClauseElement) -> typing.List["Record"]: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
| async def fetch_one(self, query: ClauseElement) -> typing.Optional["Record"]: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
| async def fetch_val( | ||
| self, query: ClauseElement, column: typing.Any = 0 | ||
| ) -> typing.Any: | ||
| row = await self.fetch_one(query) | ||
| return None if row is None else row[column] | ||
|
|
||
| async def execute(self, query: ClauseElement) -> typing.Any: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
| async def execute_many(self, queries: typing.List[ClauseElement]) -> None: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
| async def iterate( | ||
| self, query: ClauseElement | ||
| ) -> typing.AsyncGenerator[typing.Mapping, None]: | ||
| raise NotImplementedError() # pragma: no cover | ||
| # mypy needs async iterators to contain a `yield` | ||
| # https://github.com/python/mypy/issues/5385#issuecomment-407281656 | ||
| yield True # pragma: no cover | ||
|
|
||
| def transaction(self) -> "TransactionBackend": | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
| @property | ||
| def raw_connection(self) -> typing.Any: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
|
|
||
| class PsycopgTransaction(TransactionBackend): | ||
| async def start( | ||
| self, is_root: bool, extra_options: typing.Dict[typing.Any, typing.Any] | ||
| ) -> None: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
| async def commit(self) -> None: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
| async def rollback(self) -> None: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
|
|
||
| class Record(Sequence): | ||
| @property | ||
| def _mapping(self) -> typing.Mapping: | ||
| raise NotImplementedError() # pragma: no cover | ||
|
|
||
| def __getitem__(self, key: typing.Any) -> typing.Any: | ||
| raise NotImplementedError() # pragma: no cover |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters