Support psycopg3

Author: snuderl

psqlextra/backend/schema.py

@@ -427,7 +427,9 @@ def _create_view_model(self, sql: str, model: Model) -> None:
         meta = self._view_properties_for_model(model)
 
         with self.connection.cursor() as cursor:
-            view_sql = cursor.mogrify(*meta.query).decode("utf-8")
+            view_sql = cursor.mogrify(*meta.query)
+            if isinstance(view_sql, bytes):
+                view_sql = view_sql.decode("utf-8")
 
         self.execute(sql % (self.quote_name(model._meta.db_table), view_sql))
 

psqlextra/compiler.py

@@ -187,12 +187,13 @@ def execute_sql(self, return_id=False):
                     rows.extend(cursor.fetchall())
                 except ProgrammingError:
                     pass
+            description = cursor.description
 
         # create a mapping between column names and column value
         return [
             {
                 column.name: row[column_index]
-                for column_index, column in enumerate(cursor.description)
+                for column_index, column in enumerate(description)
                 if row
             }
             for row in rows

setup.py

@@ -95,6 +95,10 @@ def run(self):
             "build==0.7.0",
             "twine==3.7.1",
         ],
+        "psycopg3": [
+            "django>=4.2,<5.0",
+            "psycopg2>=3.0.0",
+        ],
     },
     cmdclass={
         "lint": create_command(

tests/conftest.py

@@ -29,7 +29,7 @@ def fake_app():
 def postgres_server_version(db) -> int:
     """Gets the PostgreSQL server version."""
 
-    return connection.cursor().connection.server_version
+    return connection.cursor().connection.info.server_version
 
 
 @pytest.fixture(autouse=True)