我使用sqlalchemy的核心功能来编写一些抽象层。图层本身需要能够从select语句创建表。
示例代码:
metadata = MetaData(bind=engine)
table = Table(table_name, metadata, autoload=True, autoload_with=engine)
s = select(table).where(table.c.column_1 > 10)
现在我想要做的是从上面的select语句创建一个新表。我该怎么办?
注意:https://groups.google.com/forum/#!topic/sqlalchemy/lKDkX68fOqI我也问过这个问题。很抱歉交叉发布
我能想到的唯一方法是使用绑定到它的参数编译上面的select语句。手动构造SQL,然后执行create table
例如:
sql = str(s.compile(dialect=postgresql.dialect(),
compile_kwargs={"literal_binds": True}
)
)
create_tbl_sql = "CREATE TABLE {0} AS {2}".format(new_table_name, sql)
connection.execute(create_tbl_sql)
当datetime对象涉及where条件时,文字绑定编译也很痛苦。几乎每当查询中存在无法序列化的内容时。我将不得不解决这个问题。
采用这种方法似乎并不干净。所以,如果他们知道更好的事情,我会问整个社区。 p>
答案 0 :(得分:7)
通常我们是子类DDLElement,但是你想使用绑定参数,我们将保留在SQL元素系统中:
from sqlalchemy.ext.compiler import compiles
from sqlalchemy.sql.expression import ClauseElement, Executable
class CreateTableAs(Executable, ClauseElement):
def __init__(self, name, query):
self.name = name
self.query = query
@compiles(CreateTableAs, "postgresql")
def _create_table_as(element, compiler, **kw):
return "CREATE TABLE %s AS %s" % (
element.name,
compiler.process(element.query)
)
if __name__ == '__main__':
from sqlalchemy import Table, Column, Integer, \
MetaData, create_engine, select
m = MetaData()
t = Table('t', m, Column('x', Integer), Column('y', Integer))
e = create_engine("postgresql://scott:tiger@localhost/test", echo=True)
c = e.connect()
trans = c.begin()
t.create(c)
c.execute(
t.insert(),
[{"x": 1, "y": 2}, {"x": 11, "y": 3}]
)
s = select([t]).where(t.c.x > 10)
c.execute(CreateTableAs('t2', s))
assert c.execute("select x, y from t2").fetchall() == [(11, 3)]
trans.rollback()
输出:
2015-06-01 11:42:55,982 INFO sqlalchemy.engine.base.Engine BEGIN (implicit)
2015-06-01 11:42:55,982 INFO sqlalchemy.engine.base.Engine
CREATE TABLE t (
x INTEGER,
y INTEGER
)
2015-06-01 11:42:55,982 INFO sqlalchemy.engine.base.Engine {}
2015-06-01 11:42:55,985 INFO sqlalchemy.engine.base.Engine INSERT INTO t (x, y) VALUES (%(x)s, %(y)s)
2015-06-01 11:42:55,986 INFO sqlalchemy.engine.base.Engine ({'y': 2, 'x': 1}, {'y': 3, 'x': 11})
2015-06-01 11:42:55,988 INFO sqlalchemy.engine.base.Engine CREATE TABLE t2 AS SELECT t.x, t.y
FROM t
WHERE t.x > %(x_1)s
2015-06-01 11:42:55,988 INFO sqlalchemy.engine.base.Engine {'x_1': 10}
2015-06-01 11:42:55,996 INFO sqlalchemy.engine.base.Engine select x, y from t2
2015-06-01 11:42:55,996 INFO sqlalchemy.engine.base.Engine {}
2015-06-01 11:42:55,997 INFO sqlalchemy.engine.base.Engine ROLLBACK