Last active
November 29, 2019 00:12
-
-
Save daninfpj/b3fd1ef0133e1923d443301ddc2ee0ff to your computer and use it in GitHub Desktop.
SQLAlchemy print query
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from sqlalchemy.dialects import postgresql | |
print(statement.compile(dialect=postgresql.dialect())) | |
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_type): | |
return text(value) | |
if not isinstance(value, str_type): | |
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 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment