changes

Author: iNishant

.github/workflows/test.yml

@@ -14,6 +14,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
+        psycopg-package: ["psycopg2", "psycopg"]
         python-version: ["3.9", "3.10", "3.11"]
         django-version: ["4.0", "4.1", "4.2", "5.0", "5.1", "5.2"]
         exclude:
@@ -53,6 +54,7 @@ jobs:
           # Using -e .[dev] installs the current package in editable mode with dev extras
           uv pip install --system -e .[dev]
           uv pip install --system Django==${{ matrix.django-version }}
+          uv pip install --system ${{ matrix.psycopg-package }}
       - name: Run tests
         run: |
           ./check-lint.sh

django_querysets_single_query_fetch/service.py

@@ -24,7 +24,6 @@
     get_related_populators,
 )
 from django.utils.dateparse import parse_datetime
-from psycopg2.extensions import QuotedString
 
 logger = logging.getLogger(__name__)
 
@@ -189,6 +188,19 @@ def _get_compiler_from_queryset(self, queryset: QuerysetWrapperType) -> Any:
 
         return compiler
 
+    def _get_sanitized_sql_param(self, param: str) -> str:
+        try:
+            from psycopg import sql
+
+            return sql.quote(param)
+        except ImportError:
+            try:
+                from psycopg2.extensions import QuotedString
+
+                return QuotedString(param).getquoted().decode("utf-8")
+            except ImportError:
+                raise ImportError("psycopg or psycopg2 not installed")
+
     def _get_django_sql_for_queryset(self, queryset: QuerysetWrapperType) -> str:
         """
         gets the sql that django would normally execute for the queryset, return empty string
@@ -212,7 +224,7 @@ def _get_django_sql_for_queryset(self, queryset: QuerysetWrapperType) -> str:
         for param in params:
             if isinstance(param, str):
                 # this is to handle special char handling
-                param = QuotedString(param).getquoted().decode("utf-8")
+                param = self._get_sanitized_sql_param(param)
             elif isinstance(param, UUID) or isinstance(param, datetime.datetime):
                 param = f"'{param}'"
             elif isinstance(param, int) or isinstance(param, float):
@@ -419,7 +431,7 @@ def execute(self) -> list[list[Any]]:
             raw_sql = f"""
                 SELECT
                     json_build_object(
-                        {', '.join([f"'{i}', {sql}" for i, sql in enumerate(non_empty_django_sqls_for_querysets)])}
+                        {", ".join([f"'{i}', {sql}" for i, sql in enumerate(non_empty_django_sqls_for_querysets)])}
                 )
             """
             with connections["default"].cursor() as cursor:

pyproject.toml

@@ -31,7 +31,6 @@ classifiers = [
 ]
 dependencies = [
     "django>=4.0,<5.3", # Updated to include Django 5.2
-    "psycopg2==2.9.9" # Keep specific version for now
 ]
 
 [project.urls]