apache-airflow-providers-postgres 5.5.2rc2__py3-none-any.whl → 5.6.0__py3-none-any.whl

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
     )

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",

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]

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.dist-info → apache_airflow_providers_postgres-5.6.0.dist-info}/METADATA

@@ -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,13 +31,15 @@ Requires-Python: ~=3.8
 Description-Content-Type: text/x-rst
 License-File: LICENSE
 License-File: NOTICE
-Requires-Dist: apache-airflow-providers-common-sql (>=1.3.1.dev0)
-Requires-Dist: apache-airflow (>=2.4.0.dev0)
+Requires-Dist: apache-airflow-providers-common-sql (>=1.3.1)
+Requires-Dist: apache-airflow (>=2.4.0)
 Requires-Dist: psycopg2-binary (>=2.8.0)
 Provides-Extra: amazon
 Requires-Dist: apache-airflow-providers-amazon (>=2.6.0) ; extra == 'amazon'
 Provides-Extra: common.sql
 Requires-Dist: apache-airflow-providers-common-sql ; extra == 'common.sql'
+Provides-Extra: openlineage
+Requires-Dist: apache-airflow-providers-openlineage ; extra == 'openlineage'
 
 
 .. Licensed to the Apache Software Foundation (ASF) under one
@@ -77,7 +79,7 @@ Requires-Dist: apache-airflow-providers-common-sql ; extra == 'common.sql'
 
 Package ``apache-airflow-providers-postgres``
 
-Release: ``5.5.2rc2``
+Release: ``5.6.0``
 
 
 `PostgreSQL <https://www.postgresql.org/>`__
@@ -90,7 +92,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
@@ -126,12 +128,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.6.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+airflow/providers/postgres/__init__.py,sha256=8epKsPUPz0y18-ywzTPCsP13mywZ4CpX6ZrQ-EOrX60,1577
+airflow/providers/postgres/get_provider_info.py,sha256=x8rgbDV8-j_KbfnelfcoOInla0qakl0S2iNaRe_Fbxc,3119
+airflow/providers/postgres/hooks/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
+airflow/providers/postgres/hooks/postgres.py,sha256=Yn6NKaDkgPH0QeZs10RNKUsgqok1RKIsZ3mN2i4-puY,15392
+airflow/providers/postgres/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
+airflow/providers/postgres/operators/postgres.py,sha256=jciwF-MeXCuz4sPen7jUWayDXPMPQMOhOI3swCqV4I8,3591
+apache_airflow_providers_postgres-5.6.0.dist-info/LICENSE,sha256=gXPVwptPlW1TJ4HSuG5OMPg-a3h43OGMkZRR1rpwfJA,10850
+apache_airflow_providers_postgres-5.6.0.dist-info/METADATA,sha256=Aa9JdzHP8lVejsZWaq_j90hh7coWUzLCXUJobLDiL4w,6345
+apache_airflow_providers_postgres-5.6.0.dist-info/NOTICE,sha256=m-6s2XynUxVSUIxO4rVablAZCvFq-wmLrqV91DotRBw,240
+apache_airflow_providers_postgres-5.6.0.dist-info/WHEEL,sha256=AtBG6SXL3KF_v0NxLf0ehyVOh0cold-JbJYXNGorC6Q,92
+apache_airflow_providers_postgres-5.6.0.dist-info/entry_points.txt,sha256=boiuBo37TJDstSn7YjGsk80nn8EOyQ_s3sVJmIJ8Sp8,105
+apache_airflow_providers_postgres-5.6.0.dist-info/top_level.txt,sha256=OeMVH5md7fr2QQWpnZoOWWxWO-0WH1IP70lpTVwopPg,8
+apache_airflow_providers_postgres-5.6.0.dist-info/RECORD,,

{apache_airflow_providers_postgres-5.5.2rc2.dist-info → apache_airflow_providers_postgres-5.6.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.40.0)
+Generator: bdist_wheel (0.41.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

apache_airflow_providers_postgres-5.5.2rc2.dist-info/RECORD

@@ -1,13 +0,0 @@
-airflow/providers/postgres/__init__.py,sha256=HC4IJM3f3qy-qI8jkgt1I0LpisWCx0yJJHmaimwOaPo,1533
-airflow/providers/postgres/get_provider_info.py,sha256=J8uMfa6fZk74LkkVQzdNnwHq1_dKmMJtElP4wPCWMWQ,3098
-airflow/providers/postgres/hooks/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/postgres/hooks/postgres.py,sha256=VZMqETR2iWRhXvqBW3z_H4B8mkuYq8ZfA4l5ZUqlieg,13342
-airflow/providers/postgres/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/postgres/operators/postgres.py,sha256=M2_3KAFMx-Cs-hQjiQ--tiM9E1V8xgmX-UroKDArCds,3467
-apache_airflow_providers_postgres-5.5.2rc2.dist-info/LICENSE,sha256=gXPVwptPlW1TJ4HSuG5OMPg-a3h43OGMkZRR1rpwfJA,10850
-apache_airflow_providers_postgres-5.5.2rc2.dist-info/METADATA,sha256=uk4cgyIiyLRRbW97J_oeibzhNvW_MHfVTt5fe-XuhWs,6113
-apache_airflow_providers_postgres-5.5.2rc2.dist-info/NOTICE,sha256=m-6s2XynUxVSUIxO4rVablAZCvFq-wmLrqV91DotRBw,240
-apache_airflow_providers_postgres-5.5.2rc2.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-apache_airflow_providers_postgres-5.5.2rc2.dist-info/entry_points.txt,sha256=boiuBo37TJDstSn7YjGsk80nn8EOyQ_s3sVJmIJ8Sp8,105
-apache_airflow_providers_postgres-5.5.2rc2.dist-info/top_level.txt,sha256=OeMVH5md7fr2QQWpnZoOWWxWO-0WH1IP70lpTVwopPg,8
-apache_airflow_providers_postgres-5.5.2rc2.dist-info/RECORD,,