apache-airflow-providers-postgres 5.10.1rc1__py3-none-any.whl → 5.10.2__py3-none-any.whl

airflow/providers/postgres/__init__.py

@@ -27,7 +27,7 @@ import packaging.version
 
 __all__ = ["__version__"]
 
-__version__ = "5.10.1"
+__version__ = "5.10.2"
 
 try:
     from airflow import __version__ as airflow_version

airflow/providers/postgres/datasets/__init__.py

@@ -0,0 +1,16 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.

airflow/providers/postgres/datasets/postgres.py

@@ -0,0 +1,37 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from urllib.parse import SplitResult
+
+
+def sanitize_uri(uri: SplitResult) -> SplitResult:
+    if not uri.netloc:
+        raise ValueError("URI format postgres:// must contain a host")
+    if uri.port is None:
+        host = uri.netloc.rstrip(":")
+        uri = uri._replace(netloc=f"{host}:5432")
+    path_parts = uri.path.split("/")
+    if len(path_parts) != 4:  # Leading slash, database, schema, and table names.
+        raise ValueError("URI format postgres:// must contain database, schema, and table names")
+    if not path_parts[2]:
+        path_parts[2] = "default"
+    return uri._replace(scheme="postgres", path="/".join(path_parts))

airflow/providers/postgres/get_provider_info.py

@@ -28,8 +28,9 @@ def get_provider_info():
         "name": "PostgreSQL",
         "description": "`PostgreSQL <https://www.postgresql.org/>`__\n",
         "state": "ready",
-        "source-date-epoch": 1707636481,
+        "source-date-epoch": 1709556176,
         "versions": [
+            "5.10.2",
             "5.10.1",
             "5.10.0",
             "5.9.0",
@@ -97,4 +98,10 @@ def get_provider_info():
                 "connection-type": "postgres",
             }
         ],
+        "dataset-uris": [
+            {
+                "schemes": ["postgres", "postgresql"],
+                "handler": "airflow.providers.postgres.datasets.postgres.sanitize_uri",
+            }
+        ],
     }

airflow/providers/postgres/hooks/postgres.py

@@ -126,7 +126,7 @@ class PostgresHook(DbApiHook):
             raise ValueError(f"Invalid cursor passed {_cursor}. Valid options are: {valid_cursors}")
 
     def get_conn(self) -> connection:
-        """Establishes a connection to a postgres database."""
+        """Establish a connection to a postgres database."""
         conn_id = getattr(self, self.conn_name_attr)
         conn = deepcopy(self.connection or self.get_connection(conn_id))
 
@@ -162,7 +162,7 @@ class PostgresHook(DbApiHook):
         return self.conn
 
     def copy_expert(self, sql: str, filename: str) -> None:
-        """Executes SQL using psycopg2's ``copy_expert`` method.
+        """Execute SQL using psycopg2's ``copy_expert`` method.
 
         Necessary to execute COPY command without access to a superuser.
 
@@ -193,11 +193,11 @@ class PostgresHook(DbApiHook):
         return uri
 
     def bulk_load(self, table: str, tmp_file: str) -> None:
-        """Loads a tab-delimited file into a database table."""
+        """Load a tab-delimited file into a database table."""
         self.copy_expert(f"COPY {table} FROM STDIN", tmp_file)
 
     def bulk_dump(self, table: str, tmp_file: str) -> None:
-        """Dumps a database table into a tab-delimited file."""
+        """Dump a database table into a tab-delimited file."""
         self.copy_expert(f"COPY {table} TO STDOUT", tmp_file)
 
     @staticmethod
@@ -326,7 +326,7 @@ class PostgresHook(DbApiHook):
         return sql
 
     def get_openlineage_database_info(self, connection) -> DatabaseInfo:
-        """Returns Postgres/Redshift specific information for OpenLineage."""
+        """Return Postgres/Redshift specific information for OpenLineage."""
         from airflow.providers.openlineage.sqlparser import DatabaseInfo
 
         is_redshift = connection.extra_dejson.get("redshift", False)
@@ -363,11 +363,11 @@ class PostgresHook(DbApiHook):
         return f"{cluster_identifier}.{region_name}:{port}"
 
     def get_openlineage_database_dialect(self, connection) -> str:
-        """Returns postgres/redshift dialect."""
+        """Return 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 current schema. This is usually changed with ``SEARCH_PATH`` parameter."""
         return self.get_first("SELECT CURRENT_SCHEMA;")[0]
 
     @classmethod

{apache_airflow_providers_postgres-5.10.1rc1.dist-info → apache_airflow_providers_postgres-5.10.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-postgres
-Version: 5.10.1rc1
+Version: 5.10.2
 Summary: Provider package apache-airflow-providers-postgres for Apache Airflow
 Keywords: airflow-provider,postgres,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -20,15 +20,15 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: System :: Monitoring
-Requires-Dist: apache-airflow-providers-common-sql>=1.3.1.dev0
-Requires-Dist: apache-airflow>=2.6.0.dev0
+Requires-Dist: apache-airflow-providers-common-sql>=1.3.1
+Requires-Dist: apache-airflow>=2.6.0
 Requires-Dist: psycopg2-binary>=2.8.0
 Requires-Dist: apache-airflow-providers-amazon>=2.6.0 ; extra == "amazon"
 Requires-Dist: apache-airflow-providers-common-sql ; extra == "common.sql"
 Requires-Dist: apache-airflow-providers-openlineage ; extra == "openlineage"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.10.1/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.10.1
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.10.2/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.10.2
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Twitter, https://twitter.com/ApacheAirflow
@@ -81,7 +81,7 @@ Provides-Extra: openlineage
 
 Package ``apache-airflow-providers-postgres``
 
-Release: ``5.10.1.rc1``
+Release: ``5.10.2``
 
 
 `PostgreSQL <https://www.postgresql.org/>`__
@@ -94,7 +94,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.10.1/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.10.2/>`_.
 
 Installation
 ------------
@@ -138,4 +138,4 @@ Dependent package
 ==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.10.1/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/5.10.2/changelog.html>`_.

apache_airflow_providers_postgres-5.10.2.dist-info/RECORD

@@ -0,0 +1,13 @@
+airflow/providers/postgres/LICENSE,sha256=ywUBpKZc7Jb96rVt5I3IDbg7dIJAbUSHkuoDcF3jbH4,13569
+airflow/providers/postgres/__init__.py,sha256=my-vOpHPxX8bNXPYr9p5itjxkamH-zBkEaGeTmoAmbQ,1584
+airflow/providers/postgres/get_provider_info.py,sha256=mXiwHrTxGX5wvFNWeoxKCpY3h8sYMQc4jPcxCsw8cNs,3566
+airflow/providers/postgres/datasets/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
+airflow/providers/postgres/datasets/postgres.py,sha256=XNhOJCbOA_soaaiS73JjULMqAM_7PBryhToe8FJREA0,1522
+airflow/providers/postgres/hooks/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
+airflow/providers/postgres/hooks/postgres.py,sha256=L1cPkPcfYAapAYYXibT2hCnSDX1Ek_S1couTMn_8Hk0,15768
+airflow/providers/postgres/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
+airflow/providers/postgres/operators/postgres.py,sha256=pB2mkEh0vOXAr0ANBKJLGSRLSkel815V4vyfET6xG44,3507
+apache_airflow_providers_postgres-5.10.2.dist-info/entry_points.txt,sha256=dhtJi6PTWHd6BwKhmI4OtSPvQVI_p0yYWI0eba83HqY,104
+apache_airflow_providers_postgres-5.10.2.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+apache_airflow_providers_postgres-5.10.2.dist-info/METADATA,sha256=trSdUvE7y3TpGK7ZxnTIPGSJhB1tI9C4w-EIC0Sl-5c,6555
+apache_airflow_providers_postgres-5.10.2.dist-info/RECORD,,

apache_airflow_providers_postgres-5.10.1rc1.dist-info/RECORD

@@ -1,11 +0,0 @@
-airflow/providers/postgres/LICENSE,sha256=ywUBpKZc7Jb96rVt5I3IDbg7dIJAbUSHkuoDcF3jbH4,13569
-airflow/providers/postgres/__init__.py,sha256=0qo2QFnNnN3BFndU0MWjASqnUNUk_6LFRyZiZUHWKck,1584
-airflow/providers/postgres/get_provider_info.py,sha256=O3-FdXvyZLWKXkniheLReeUQ16L_6eOlXByZaQB8NaM,3336
-airflow/providers/postgres/hooks/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/postgres/hooks/postgres.py,sha256=1Wt6Vy8MaZmQW7QU70OmWn2mYl_SbmeHRgafAZlMMlc,15776
-airflow/providers/postgres/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/postgres/operators/postgres.py,sha256=pB2mkEh0vOXAr0ANBKJLGSRLSkel815V4vyfET6xG44,3507
-apache_airflow_providers_postgres-5.10.1rc1.dist-info/entry_points.txt,sha256=dhtJi6PTWHd6BwKhmI4OtSPvQVI_p0yYWI0eba83HqY,104
-apache_airflow_providers_postgres-5.10.1rc1.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-apache_airflow_providers_postgres-5.10.1rc1.dist-info/METADATA,sha256=uXtwxMuy_Yx2DBWbuzKCg-vqwYgtbwXSMSr0l9J7tFQ,6572
-apache_airflow_providers_postgres-5.10.1rc1.dist-info/RECORD,,