Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion providers/postgres/README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,8 @@ PIP package Version required
``apache-airflow-providers-common-sql`` ``>=1.32.0``
``psycopg2-binary`` ``>=2.9.9; python_version < "3.13"``
``psycopg2-binary`` ``>=2.9.10; python_version >= "3.13"``
``asyncpg`` ``>=0.30.0``
``psycopg[binary]`` ``>=3.2.9; python_version < "3.14"``
``psycopg[binary]`` ``>=3.3.3; python_version >= "3.14"``
========================================== ======================================

Cross provider package dependencies
Expand Down
9 changes: 9 additions & 0 deletions providers/postgres/docs/changelog.rst
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,15 @@
Changelog
---------

.. warning::
``asyncpg`` is no longer installed by default — it moved to the ``asyncpg`` optional extra, and
``psycopg`` (psycopg3) is now installed by default as the async metadata-database driver. The
default derived async connection URL changed from ``postgresql+asyncpg://`` to
``postgresql+psycopg_async://``, which is safe behind transaction-mode PgBouncer with no extra
configuration. This aligns with the new default async Postgres driver in Airflow 3.4.0. To keep
using asyncpg, install ``apache-airflow-providers-postgres[asyncpg]`` and set
``[database] sql_alchemy_conn_async = postgresql+asyncpg://...`` explicitly.

6.8.0
.....

Expand Down
3 changes: 2 additions & 1 deletion providers/postgres/docs/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,8 @@ PIP package Version required
``apache-airflow-providers-common-sql`` ``>=1.32.0``
``psycopg2-binary`` ``>=2.9.9; python_version < "3.13"``
``psycopg2-binary`` ``>=2.9.10; python_version >= "3.13"``
``asyncpg`` ``>=0.30.0``
``psycopg[binary]`` ``>=3.2.9; python_version < "3.14"``
``psycopg[binary]`` ``>=3.3.3; python_version >= "3.14"``
========================================== ======================================

Cross provider package dependencies
Expand Down
13 changes: 8 additions & 5 deletions providers/postgres/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -62,9 +62,12 @@ dependencies = [
"apache-airflow>=2.11.0",
"apache-airflow-providers-common-compat>=1.12.0",
"apache-airflow-providers-common-sql>=1.32.0",
# psycopg2 remains the sync driver for now; its removal and the sync migration to
# psycopg3 are tracked at https://github.com/apache/airflow/issues/68453
"psycopg2-binary>=2.9.9; python_version < '3.13'",
"psycopg2-binary>=2.9.10; python_version >= '3.13'",
"asyncpg>=0.30.0",
"psycopg[binary]>=3.2.9; python_version < '3.14'",
"psycopg[binary]>=3.3.3; python_version >= '3.14'",
]

# The optional dependencies should be modified in place in the generated file
Expand All @@ -73,6 +76,9 @@ dependencies = [
"amazon" = [
"apache-airflow-providers-amazon>=2.6.0",
]
"asyncpg" = [
"asyncpg>=0.30.0",
]
"microsoft.azure" = [
"apache-airflow-providers-microsoft-azure>=12.8.0"
]
Expand All @@ -87,10 +93,6 @@ dependencies = [
"polars" = [
"polars>=1.26.0"
]
"psycopg" = [
"psycopg[binary]>=3.2.9; python_version < '3.14'",
"psycopg[binary]>=3.3.3; python_version >= '3.14'",
]
"sqlalchemy" = [
"sqlalchemy>=1.4.54"
]
Expand All @@ -108,6 +110,7 @@ dev = [
# Additional devel dependencies (do not remove this line and add extra development dependencies)
"apache-airflow-providers-common-sql[pandas]",
"apache-airflow-providers-common-sql[polars]",
"apache-airflow-providers-postgres[asyncpg]",
"apache-airflow-providers-postgres[sqlalchemy]"
]

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,30 +70,14 @@


class CompatConnection(Protocol):
"""Protocol for type hinting psycopg2 and psycopg3 connection objects."""
"""Protocol for the common interface shared by psycopg2 and psycopg3 connection objects."""

def cursor(self, *args, **kwargs) -> Any: ...
def commit(self) -> None: ...
def close(self) -> None: ...

# Context manager support
def __enter__(self) -> CompatConnection: ...
def __enter__(self) -> Any: ...
def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None: ...

# Common properties
@property
def notices(self) -> list[Any]: ...

# psycopg3 specific (optional)
@property
def adapters(self) -> Any: ...

@property
def row_factory(self) -> Any: ...

# Optional method for psycopg3
def add_notice_handler(self, handler: Any) -> None: ...


class PostgresHook(DbApiHook):
"""
Expand Down
23 changes: 12 additions & 11 deletions uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading