Merge pull request #7 from apavelchuk/fix-mogrify-psycopg3

Fix mogrify() for psycopg3
paul-wolf · May 3, 2023 · 7a76129 · 7a76129
2 parents edb17bd + 15a1455
commit 7a76129
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/djaq/query/__init__.py b/djaq/query/__init__.py
@@ -459,7 +459,10 @@ def mogrify(self, sql, parameters):
 
  conn = connections[self.using]
  cursor = conn.cursor()
- return cursor.mogrify(sql, parameters).decode()
+ mogrified = cursor.mogrify(sql, parameters)
+ if isinstance(mogrified, bytes):
+ return mogrified.decode()
+ return mogrified
 
  def dump(self):
  for k, v in self.__dict__.items():