如何在SQLAlchemy中执行INSERT INTO t1 (SELECT * FROM t2)？

Question

在SQLAlchemy中，我该如何从一个SELECT语句填充或更新一个表呢？

Answer 1

正如Noslko在评论中提到的，现在你可以不再使用原始的SQL语句了：

http://www.sqlalchemy.org/docs/core/compiler.html#compiling-sub-elements-of-a-custom-expression-construct

from sqlalchemy.ext.compiler import compiles
from sqlalchemy.sql.expression import Executable, ClauseElement

class InsertFromSelect(Executable, ClauseElement):
    def __init__(self, table, select):
        self.table = table
        self.select = select

@compiles(InsertFromSelect)
def visit_insert_from_select(element, compiler, **kw):
    return "INSERT INTO %s (%s)" % (
        compiler.process(element.table, asfrom=True),
        compiler.process(element.select)
    )

insert = InsertFromSelect(t1, select([t1]).where(t1.c.x>5))
print insert

生成的结果是：

INSERT INTO mytable (SELECT mytable.x, mytable.y, mytable.z FROM mytable WHERE mytable.x > :x_1)

Answer 2

从0.8.3版本开始，你可以直接在sqlalchemy中使用这个功能：Insert.from_select：

sel = select([table1.c.a, table1.c.b]).where(table1.c.c > 5)
ins = table2.insert().from_select(['a', 'b'], sel)

Answer 3

SQLalchemy不会自动为你构建这个结构。你可以使用文本查询。

session.execute('INSERT INTO t1 (SELECT * FROM t2)')

---

编辑：

一年多以后，现在在sqlalchemy 0.6+版本中，你可以创建它：

from sqlalchemy.ext import compiler
from sqlalchemy.sql.expression import Executable, ClauseElement

class InsertFromSelect(Executable, ClauseElement):
    def __init__(self, table, select):
        self.table = table
        self.select = select

@compiler.compiles(InsertFromSelect)
def visit_insert_from_select(element, compiler, **kw):
    return "INSERT INTO %s (%s)" % (
        compiler.process(element.table, asfrom=True),
        compiler.process(element.select)
    )

insert = InsertFromSelect(t1, select([t1]).where(t1.c.x>5))
print insert

生成的结果是：

"INSERT INTO mytable (SELECT mytable.x, mytable.y, mytable.z FROM mytable WHERE mytable.x > :x_1)"

---

另一个编辑：

现在，四年后，这个语法已经被整合进SQLAlchemy 0.9，并且也回溯到0.8.3版本；你可以创建任何select()，然后使用Insert对象的新from_select()方法：

>>> from sqlalchemy.sql import table, column
>>> t1 = table('t1', column('a'), column('b'))
>>> t2 = table('t2', column('x'), column('y'))
>>> print(t1.insert().from_select(['a', 'b'], t2.select().where(t2.c.y == 5)))
INSERT INTO t1 (a, b) SELECT t2.x, t2.y
FROM t2
WHERE t2.y = :y_1

更多信息请查看文档。