Add support for environments running psycopg3 to the migration executor

kolonialno · Apr 24, 2024 · db6e68e · db6e68e
1 parent c610453
commit db6e68e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/migration_checker/executor.py b/migration_checker/executor.py
@@ -33,7 +33,12 @@ def __call__(
         context: dict[str, Any],
     ) -> Any:
         cursor = context["cursor"]
-        rendered_sql = cursor.mogrify(sql, params).decode()
+        mogrify_result = cursor.mogrify(sql, params)
+        rendered_sql: str = (
+            mogrify_result
+            if isinstance(mogrify_result, str)
+            else mogrify_result.decode()
+        )
         self.queries.append(rendered_sql)
 
         return execute(sql, params, many, context)