Chore: Remove deprecated pyscopg v2 (pyscopg2)

airbyte/_processors/sql/postgres.py

@@ -6,6 +6,7 @@
 import functools
 
 from overrides import overrides
+from sqlalchemy import URL
 
 from airbyte._util.name_normalizers import LowerCaseNormalizer
 from airbyte._writers.jsonl import JsonlWriter
@@ -28,9 +29,15 @@ class PostgresConfig(SqlConfig):
     @overrides
     def get_sql_alchemy_url(self) -> SecretString:
         """Return the SQLAlchemy URL to use."""
-        return SecretString(
-            f"postgresql://{self.username}:{self.password}@{self.host}:{self.port}/{self.database}"
+        url = URL.create(
+            drivername="postgresql+psycopg",
+            username=self.username,
+            password=self.password,
+            host=self.host,
+            port=self.port,  # defaults to 5432
+            database=self.database,
         )
+        return SecretString(str(url))
 
     @overrides
     def get_database_name(self) -> str:

pyproject.toml

@@ -30,7 +30,6 @@ orjson = "^3.10"
 overrides = "^7.4.0"
 pandas = { version = ">=1.5.3,<3.0" }
 psycopg = {extras = ["binary", "pool"], version = "^3.1.19"}
-psycopg2-binary =  "^2.9.9"
 pyarrow = ">=16.1,<18.0"
 pydantic = ">=2.0,<=3.0"
 pydantic-core = "*"

tests/conftest.py

@@ -13,6 +13,8 @@
 import warnings
 from pathlib import Path
 
+from sqlalchemy import URL
+
 import airbyte
 import docker
 import psycopg
@@ -125,7 +127,16 @@ def remove_postgres_container():
 
 
 def test_pg_connection(host) -> bool:
-    pg_url = f"postgresql://postgres:postgres@{host}:{PYTEST_POSTGRES_PORT}/postgres"
+    pg_url = str(
+        URL.create(
+            drivername="postgresql+psycopg",
+            username="postgres",
+            password="postgres",
+            host=host,
+            port=PYTEST_POSTGRES_PORT,  # defaults to 5432
+            database="postgres",
+        )
+    )
 
     max_attempts = 120
     for attempt in range(max_attempts):