5

我正在寻找一种方法来告诉 sqlalchemy 将一些标签上的复杂查询映射到自定义类MyResult而不是默认RowProxy类。这是一个简单的工作示例

'''
create table foo(id integer, title text);
create table bar(id integer, foo_id integer, name text);
insert into foo values(0, 'null');
insert into foo values(1, 'eins');
insert into bar values(0,0, 'nullnull');
insert into bar values(1,0, 'einsnull');
insert into bar values(2,1, 'zweieins');
'''

和以下代码：

from sqlalchemy import *
from itertools import imap

db = create_engine('sqlite:///test.db')
metadata = MetaData(db)

class MyResult(object):
    def __init__(self, id, title, name):
        self.id = id
        self.title = title
        self.name = name

foo = Table('foo', metadata, autoload=True)
bar = Table('bar', metadata, autoload=True)

result = select([foo.c.id, foo.c.title, bar.c.name], foo.c.id == bar.c.foo_id).execute().fetchall()

现在我正在寻找一种方法来告诉 sqlalchemy 执行从结果行到 MyResult 的映射。

row = result[0]
print type(row)
#<class 'sqlalchemy.engine.base.RowProxy'>
print row.items()
#[(u'id', 0), (u'title', u'null'), (u'name', u'einsnull')]

我知道我可以用类似的东西手动进行映射

my_result = imap(lambda x: MyResult(**x), result)

但我觉得这不是在 sqlalchemy 中处理它的方法。

4

2 回答 2

9

从您的示例中可以看出，将返回超过 1 个 Foo Foo.id = 0，这将导致主键的重复值，而这反过来只会导致返回结果集的子集。在这种情况下，您可能primary_key还应该将其扩展到其他Bar列（如果它是唯一的，则 包含Bar.id或使用）。Bar.name

然后您可以使用from_statement（如Using Literal SQL中所述）来实现此目的：

sql_qry = select([foo.c.id.label("id"), 
                  foo.c.title.label("title"), 
                  bar.c.name.label("name")], 
                 foo.c.id == bar.c.foo_id)
my_result_qry = session.query(MyResult).from_statement(sql_qry)
for x in my_result_qry.all():
    print x

但是，MyResult必须映射模型。您可以将其映射到一些虚拟（不存在的）表或视图。列的labels 也很重要，因为它们必须与类的列定义完全匹配（无论如何都不会使用构造函数）。

于 2012-04-17T15:55:25.787 回答
5

通过直接调用select，您将忽略 ORM 功能。您需要在课堂上mapper使用MyResult。正如你所拥有的，MyResult只是一个普通的类。

像这样的东西：

Foo = mapper(MyResult, foo)
Bar = mapper(MyResult, bar)  # NOTE: MyResult itself is unchanged by this

session = Session()
# query against the mapper class
result = session.query(Foo).filter(Foo.title == 'xyz').one()
print result.name
于 2012-04-16T21:52:54.537 回答