GermanFilipp · June 12, 2020 11:30 · Jun 12, 2020 · Jun 10, 2020 · Jun 10, 2020
diff --git a/print_parametrized_query.py b/print_parametrized_query.py
@@ -8,9 +8,9 @@ def literal_processor(self, dialect):
 
         def process(value):
             if isinstance(value, int):
-                return text(value)
+                return str(value)
             if not isinstance(value, str):
-                value = text(value)
+                value = str(value)
             result = super_processor(value)
             if isinstance(result, bytes):
                 result = result.decode(dialect.encoding)

diff --git a/print_parametrized_query.py b/print_parametrized_query.py
@@ -1,13 +1,6 @@
 from sqlalchemy.engine.default import DefaultDialect
 from sqlalchemy.sql.sqltypes import String, DateTime, NullType
 
-# python2/3 compatible.
-PY3 = str is not bytes
-text = str if PY3 else unicode
-int_type = int if PY3 else (int, long)
-str_type = str if PY3 else (str, unicode)
-
-
 class StringLiteral(String):
     """Teach SA how to literalize various things."""
     def literal_processor(self, dialect):

diff --git a/print_parametrized_query.py b/print_parametrized_query.py
@@ -0,0 +1,47 @@
+from sqlalchemy.engine.default import DefaultDialect
+from sqlalchemy.sql.sqltypes import String, DateTime, NullType
+
+# python2/3 compatible.
+PY3 = str is not bytes
+text = str if PY3 else unicode
+int_type = int if PY3 else (int, long)
+str_type = str if PY3 else (str, unicode)
+
+
+class StringLiteral(String):
+    """Teach SA how to literalize various things."""
+    def literal_processor(self, dialect):
+        super_processor = super(StringLiteral, self).literal_processor(dialect)
+
+        def process(value):
+            if isinstance(value, int):
+                return text(value)
+            if not isinstance(value, str):
+                value = text(value)
+            result = super_processor(value)
+            if isinstance(result, bytes):
+                result = result.decode(dialect.encoding)
+            return result
+        return process
+
+
+class LiteralDialect(DefaultDialect):
+    colspecs = {
+        # prevent various encoding explosions
+        String: StringLiteral,
+        # teach SA about how to literalize a datetime
+        DateTime: StringLiteral,
+        # don't format py2 long integers to NULL
+        NullType: StringLiteral,
+    }
+
+
+def literalquery(statement):
+    """NOTE: This is entirely insecure. DO NOT execute the resulting strings."""
+    import sqlalchemy.orm
+    if isinstance(statement, sqlalchemy.orm.Query):
+        statement = statement.statement
+    return statement.compile(
+        dialect=LiteralDialect(),
+        compile_kwargs={'literal_binds': True},
+    ).string
No results found