apache-airflow-providers-postgres 5.5.2rc2__tar.gz → 5.6.0__tar.gz

{apache-airflow-providers-postgres-5.5.2rc2/apache_airflow_providers_postgres.egg-info → apache-airflow-providers-postgres-5.6.0}/PKG-INFO

@@ -1,14 +1,14 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-postgres
-Version: 5.5.2rc2
+Version: 5.6.0
 Summary: Provider for Apache Airflow. Implements apache-airflow-providers-postgres package
 Home-page: https://airflow.apache.org/
 Download-URL: https://archive.apache.org/dist/airflow/providers
 Author: Apache Software Foundation
 Author-email: dev@airflow.apache.org
 License: Apache License 2.0
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/changelog.html
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
@@ -31,6 +31,7 @@ Requires-Python: ~=3.8
 Description-Content-Type: text/x-rst
 Provides-Extra: amazon
 Provides-Extra: common.sql
+Provides-Extra: openlineage
 License-File: LICENSE
 License-File: NOTICE
 
@@ -72,7 +73,7 @@ License-File: NOTICE
 
 Package ``apache-airflow-providers-postgres``
 
-Release: ``5.5.2rc2``
+Release: ``5.6.0``
 
 
 `PostgreSQL <https://www.postgresql.org/>`__
@@ -85,7 +86,7 @@ This is a provider package for ``postgres`` provider. All classes for this provi
 are in ``airflow.providers.postgres`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/>`_.
 
 
 Installation
@@ -121,12 +122,13 @@ You can install such cross-provider dependencies when installing from PyPI. For
     pip install apache-airflow-providers-postgres[amazon]
 
 
-============================================================================================================  ==============
-Dependent package                                                                                             Extra
-============================================================================================================  ==============
-`apache-airflow-providers-amazon <https://airflow.apache.org/docs/apache-airflow-providers-amazon>`_          ``amazon``
-`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_  ``common.sql``
-============================================================================================================  ==============
+==============================================================================================================  ===============
+Dependent package                                                                                               Extra
+==============================================================================================================  ===============
+`apache-airflow-providers-amazon <https://airflow.apache.org/docs/apache-airflow-providers-amazon>`_            ``amazon``
+`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_    ``common.sql``
+`apache-airflow-providers-openlineage <https://airflow.apache.org/docs/apache-airflow-providers-openlineage>`_  ``openlineage``
+==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/changelog.html>`_.

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/README.rst

@@ -36,7 +36,7 @@
 
 Package ``apache-airflow-providers-postgres``
 
-Release: ``5.5.2rc2``
+Release: ``5.6.0``
 
 
 `PostgreSQL <https://www.postgresql.org/>`__
@@ -49,7 +49,7 @@ This is a provider package for ``postgres`` provider. All classes for this provi
 are in ``airflow.providers.postgres`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/>`_.
 
 
 Installation
@@ -85,12 +85,13 @@ You can install such cross-provider dependencies when installing from PyPI. For
     pip install apache-airflow-providers-postgres[amazon]
 
 
-============================================================================================================  ==============
-Dependent package                                                                                             Extra
-============================================================================================================  ==============
-`apache-airflow-providers-amazon <https://airflow.apache.org/docs/apache-airflow-providers-amazon>`_          ``amazon``
-`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_  ``common.sql``
-============================================================================================================  ==============
+==============================================================================================================  ===============
+Dependent package                                                                                               Extra
+==============================================================================================================  ===============
+`apache-airflow-providers-amazon <https://airflow.apache.org/docs/apache-airflow-providers-amazon>`_            ``amazon``
+`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_    ``common.sql``
+`apache-airflow-providers-openlineage <https://airflow.apache.org/docs/apache-airflow-providers-openlineage>`_  ``openlineage``
+==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/changelog.html>`_.

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/airflow/providers/postgres/__init__.py

@@ -28,14 +28,16 @@ import packaging.version
 
 __all__ = ["__version__"]
 
-__version__ = "5.5.2"
+__version__ = "5.6.0"
 
 try:
     from airflow import __version__ as airflow_version
 except ImportError:
     from airflow.version import version as airflow_version
 
-if packaging.version.parse(airflow_version) < packaging.version.parse("2.4.0"):
+if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
+    "2.4.0"
+):
     raise RuntimeError(
         f"The package `apache-airflow-providers-postgres:{__version__}` requires Apache Airflow 2.4.0+"  # NOQA: E501
     )

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/airflow/providers/postgres/get_provider_info.py

@@ -29,6 +29,7 @@ def get_provider_info():
         "description": "`PostgreSQL <https://www.postgresql.org/>`__\n",
         "suspended": False,
         "versions": [
+            "5.6.0",
             "5.5.2",
             "5.5.1",
             "5.5.0",

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/airflow/providers/postgres/hooks/postgres.py

@@ -21,7 +21,7 @@ import os
 import warnings
 from contextlib import closing
 from copy import deepcopy
-from typing import Any, Iterable, Union
+from typing import TYPE_CHECKING, Any, Iterable, Union
 
 import psycopg2
 import psycopg2.extensions
@@ -33,6 +33,9 @@ from airflow.exceptions import AirflowProviderDeprecationWarning
 from airflow.models.connection import Connection
 from airflow.providers.common.sql.hooks.sql import DbApiHook
 
+if TYPE_CHECKING:
+    from airflow.providers.openlineage.sqlparser import DatabaseInfo
+
 CursorType = Union[DictCursor, RealDictCursor, NamedTupleCursor]
 
 
@@ -317,3 +320,46 @@ class PostgresHook(DbApiHook):
                 sql += f"{on_conflict_str} DO NOTHING"
 
         return sql
+
+    def get_openlineage_database_info(self, connection) -> DatabaseInfo:
+        """Returns Postgres/Redshift specific information for OpenLineage."""
+        from airflow.providers.openlineage.sqlparser import DatabaseInfo
+
+        is_redshift = connection.extra_dejson.get("redshift", False)
+
+        if is_redshift:
+            authority = self._get_openlineage_redshift_authority_part(connection)
+        else:
+            authority = DbApiHook.get_openlineage_authority_part(connection, default_port=5432)
+
+        return DatabaseInfo(
+            scheme="postgres" if not is_redshift else "redshift",
+            authority=authority,
+            database=self.database or connection.schema,
+        )
+
+    def _get_openlineage_redshift_authority_part(self, connection) -> str:
+        try:
+            from airflow.providers.amazon.aws.hooks.base_aws import AwsBaseHook
+        except ImportError:
+            from airflow.exceptions import AirflowException
+
+            raise AirflowException(
+                "apache-airflow-providers-amazon not installed, run: "
+                "pip install 'apache-airflow-providers-postgres[amazon]'."
+            )
+        aws_conn_id = connection.extra_dejson.get("aws_conn_id", "aws_default")
+
+        port = connection.port or 5439
+        cluster_identifier = connection.extra_dejson.get("cluster-identifier", connection.host.split(".")[0])
+        region_name = AwsBaseHook(aws_conn_id=aws_conn_id).region_name
+
+        return f"{cluster_identifier}.{region_name}:{port}"
+
+    def get_openlineage_database_dialect(self, connection) -> str:
+        """Returns postgres/redshift dialect."""
+        return "redshift" if connection.extra_dejson.get("redshift", False) else "postgres"
+
+    def get_openlineage_default_schema(self) -> str | None:
+        """Returns current schema. This is usually changed with ``SEARCH_PATH`` parameter."""
+        return self.get_first("SELECT CURRENT_SCHEMA;")[0]

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/airflow/providers/postgres/operators/postgres.py

@@ -28,6 +28,10 @@ class PostgresOperator(SQLExecuteQueryOperator):
     """
     Executes sql code in a specific Postgres database.
 
+    This class is deprecated.
+
+    Please use :class:`airflow.providers.common.sql.operators.sql.SQLExecuteQueryOperator`.
+
     :param sql: the SQL code to be executed as a single string, or
         a list of str (sql statements), or a reference to a template file.
         Template references are recognized by str ending in '.sql'

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0/apache_airflow_providers_postgres.egg-info}/PKG-INFO

@@ -1,14 +1,14 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-postgres
-Version: 5.5.2rc2
+Version: 5.6.0
 Summary: Provider for Apache Airflow. Implements apache-airflow-providers-postgres package
 Home-page: https://airflow.apache.org/
 Download-URL: https://archive.apache.org/dist/airflow/providers
 Author: Apache Software Foundation
 Author-email: dev@airflow.apache.org
 License: Apache License 2.0
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/changelog.html
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
@@ -31,6 +31,7 @@ Requires-Python: ~=3.8
 Description-Content-Type: text/x-rst
 Provides-Extra: amazon
 Provides-Extra: common.sql
+Provides-Extra: openlineage
 License-File: LICENSE
 License-File: NOTICE
 
@@ -72,7 +73,7 @@ License-File: NOTICE
 
 Package ``apache-airflow-providers-postgres``
 
-Release: ``5.5.2rc2``
+Release: ``5.6.0``
 
 
 `PostgreSQL <https://www.postgresql.org/>`__
@@ -85,7 +86,7 @@ This is a provider package for ``postgres`` provider. All classes for this provi
 are in ``airflow.providers.postgres`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/>`_.
 
 
 Installation
@@ -121,12 +122,13 @@ You can install such cross-provider dependencies when installing from PyPI. For
     pip install apache-airflow-providers-postgres[amazon]
 
 
-============================================================================================================  ==============
-Dependent package                                                                                             Extra
-============================================================================================================  ==============
-`apache-airflow-providers-amazon <https://airflow.apache.org/docs/apache-airflow-providers-amazon>`_          ``amazon``
-`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_  ``common.sql``
-============================================================================================================  ==============
+==============================================================================================================  ===============
+Dependent package                                                                                               Extra
+==============================================================================================================  ===============
+`apache-airflow-providers-amazon <https://airflow.apache.org/docs/apache-airflow-providers-amazon>`_            ``amazon``
+`apache-airflow-providers-common-sql <https://airflow.apache.org/docs/apache-airflow-providers-common-sql>`_    ``common.sql``
+`apache-airflow-providers-openlineage <https://airflow.apache.org/docs/apache-airflow-providers-openlineage>`_  ``openlineage``
+==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/changelog.html>`_.

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/apache_airflow_providers_postgres.egg-info/requires.txt

@@ -1,5 +1,5 @@
-apache-airflow-providers-common-sql>=1.3.1.dev0
-apache-airflow>=2.4.0.dev0
+apache-airflow-providers-common-sql>=1.3.1
+apache-airflow>=2.4.0
 psycopg2-binary>=2.8.0
 
 [amazon]
@@ -7,3 +7,6 @@ apache-airflow-providers-amazon>=2.6.0
 
 [common.sql]
 apache-airflow-providers-common-sql
+
+[openlineage]
+apache-airflow-providers-openlineage

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/pyproject.toml

@@ -41,7 +41,8 @@ extend-exclude = [
     # The files generated by stubgen aren't 100% valid syntax it turns out, and we don't ship them, so we can
     # ignore them in ruff
     "airflow/providers/common/sql/*/*.pyi",
-    "airflow/migrations/versions/*.py"
+    "airflow/migrations/versions/*.py",
+    "tests/dags/test_imports.py",
 ]
 
 extend-select = [
@@ -73,6 +74,8 @@ extend-ignore = [
     "E731",
 ]
 
+namespace-packages = ["airflow/providers"]
+
 [tool.pytest.ini_options]
 # * Disable `flaky` plugin for pytest. This plugin conflicts with `rerunfailures` because provide same marker.
 # * Disable `nose` builtin plugin for pytest. This feature deprecated in 7.2 and will be removed in pytest>=8
@@ -99,46 +102,20 @@ filterwarnings = [
     "ignore::DeprecationWarning:apispec.utils",
 ]
 python_files = [
-    "*.py",
+    "test_*.py",
 ]
 testpaths = [
     "tests",
 ]
 
 [tool.ruff.isort]
-known-first-party = ["airflow", "airflow_breeze", "docker_tests", "docs", "kubernetes_tests", "tests"]
 required-imports = ["from __future__ import annotations"]
 combine-as-imports = true
 
-# TODO: for now, https://github.com/charliermarsh/ruff/issues/1817
-known-third-party = [
-    "asana",
-    "atlassian",
-    "celery",
-    "cloudant",
-    "databricks",
-    "datadog",
-    "docker",
-    "elasticsearch",
-    "github",
-    "google",
-    "grpc",
-    "jenkins",
-    "mysql",
-    "neo4j",
-    "papermill",
-    "redis",
-    "sendgrid",
-    "snowflake",
-    "telegram",
-    "trino",
-]
-
 [tool.ruff.per-file-ignores]
 "airflow/models/__init__.py" = ["F401"]
 "airflow/models/sqla_models.py" = ["F401"]
 
-
 # The test_python.py is needed because adding __future__.annotations breaks runtime checks that are
 # needed for the test to work
 "tests/decorators/test_python.py" = ["I002"]

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/setup.cfg

@@ -27,8 +27,8 @@ classifiers =
 	Programming Language :: Python :: 3.11
 	Topic :: System :: Monitoring
 project_urls = 
-	Documentation=https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/
-	Changelog=https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.5.2/changelog.html
+	Documentation=https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/
+	Changelog=https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.6.0/changelog.html
 	Bug Tracker=https://github.com/apache/airflow/issues
 	Source Code=https://github.com/apache/airflow
 	Slack Chat=https://s.apache.org/airflow-slack
@@ -47,8 +47,8 @@ setup_requires =
 	setuptools
 	wheel
 install_requires = 
-	apache-airflow-providers-common-sql>=1.3.1.dev0
-	apache-airflow>=2.4.0.dev0
+	apache-airflow-providers-common-sql>=1.3.1
+	apache-airflow>=2.4.0
 	psycopg2-binary>=2.8.0
 
 [options.entry_points]
@@ -59,6 +59,6 @@ apache_airflow_provider =
 packages = airflow.providers.postgres
 
 [egg_info]
-tag_build = rc2
+tag_build = 
 tag_date = 0
 

{apache-airflow-providers-postgres-5.5.2rc2 → apache-airflow-providers-postgres-5.6.0}/setup.py

@@ -26,7 +26,7 @@
 
 from setuptools import find_namespace_packages, setup
 
-version = "5.5.2"
+version = "5.6.0"
 
 
 def do_setup():
@@ -36,6 +36,7 @@ def do_setup():
         extras_require={
             "amazon": ["apache-airflow-providers-amazon>=2.6.0"],
             "common.sql": ["apache-airflow-providers-common-sql"],
+            "openlineage": ["apache-airflow-providers-openlineage"],
         },
         packages=find_namespace_packages(
             include=[