handle None parameters in query, returning NULL

MagicStack · Sep 8, 2024 · 3eb681d · 3eb681d
1 parent 0a322a2
commit 3eb681d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/asyncpg/utils.py b/asyncpg/utils.py
@@ -42,4 +42,11 @@ async def _mogrify(conn, query, args):
 
     # Finally, replace $n references with text values.
     return re.sub(
-        r'\$(\d+)\b', lambda m: textified[int(m.group(1)) - 1], query)
+        r"\$(\d+)\b",
+        lambda m: (
+            textified[int(m.group(1)) - 1]
+            if textified[int(m.group(1)) - 1] is not None
+            else "NULL"
+        ),
+        query,
+    )