mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-27 11:01:44 -04:00
Mike Bayer
93855ed623
Implemented a new feature whereby the :class:`.AliasedClass` construct can now be used as the target of a :func:`.relationship`. This allows the concept of "non primary mappers" to no longer be necessary, as the :class:`.AliasedClass` is much easier to configure and automatically inherits all the relationships of the mapped class, as well as preserves the ability for loader options to work normally. - introduce new name for mapped_table, "persist_selectable". this is the selectable that selects against the local mapper and its superclasses, but does not include columns local only to subclasses. - relationship gains "entity" which is the mapper or aliasedinsp. - clarfiy name "entity" vs. "query_entity" in loader strategies. Fixes: #4423 Fixes: #4422 Fixes: #4421 Fixes: #3348 Change-Id: Ic3609b43dc4ed115006da9ad9189e574dc0c72d9
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
"""Generic mapping to Select statements"""
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import column
|
|
from sqlalchemy import Integer
|
|
from sqlalchemy import select
|
|
from sqlalchemy import String
|
|
from sqlalchemy import testing
|
|
from sqlalchemy.orm import mapper
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy.testing import assert_raises
|
|
from sqlalchemy.testing import assert_raises_message
|
|
from sqlalchemy.testing import AssertsCompiledSQL
|
|
from sqlalchemy.testing import eq_
|
|
from sqlalchemy.testing import fixtures
|
|
from sqlalchemy.testing.schema import Column
|
|
from sqlalchemy.testing.schema import Table
|
|
|
|
|
|
# TODO: more tests mapping to selects
|
|
|
|
|
|
class SelectableNoFromsTest(fixtures.MappedTest, AssertsCompiledSQL):
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
Table(
|
|
"common",
|
|
metadata,
|
|
Column(
|
|
"id", Integer, primary_key=True, test_needs_autoincrement=True
|
|
),
|
|
Column("data", Integer),
|
|
Column("extra", String(45)),
|
|
)
|
|
|
|
@classmethod
|
|
def setup_classes(cls):
|
|
class Subset(cls.Comparable):
|
|
pass
|
|
|
|
def test_no_tables(self):
|
|
Subset = self.classes.Subset
|
|
|
|
selectable = select([column("x"), column("y"), column("z")]).alias()
|
|
mapper(Subset, selectable, primary_key=[selectable.c.x])
|
|
|
|
self.assert_compile(
|
|
Session().query(Subset),
|
|
"SELECT anon_1.x AS anon_1_x, anon_1.y AS anon_1_y, "
|
|
"anon_1.z AS anon_1_z FROM (SELECT x, y, z) AS anon_1",
|
|
use_default_dialect=True,
|
|
)
|
|
|
|
def test_no_table_needs_pl(self):
|
|
Subset = self.classes.Subset
|
|
|
|
selectable = select([column("x"), column("y"), column("z")]).alias()
|
|
assert_raises_message(
|
|
sa.exc.ArgumentError,
|
|
"could not assemble any primary key columns",
|
|
mapper,
|
|
Subset,
|
|
selectable,
|
|
)
|
|
|
|
def test_no_selects(self):
|
|
Subset, common = self.classes.Subset, self.tables.common
|
|
|
|
subset_select = select([common.c.id, common.c.data])
|
|
assert_raises(
|
|
sa.exc.InvalidRequestError, mapper, Subset, subset_select
|
|
)
|
|
|
|
def test_basic(self):
|
|
Subset, common = self.classes.Subset, self.tables.common
|
|
|
|
subset_select = select([common.c.id, common.c.data]).alias()
|
|
mapper(Subset, subset_select)
|
|
sess = Session(bind=testing.db)
|
|
sess.add(Subset(data=1))
|
|
sess.flush()
|
|
sess.expunge_all()
|
|
|
|
eq_(sess.query(Subset).all(), [Subset(data=1)])
|
|
eq_(sess.query(Subset).filter(Subset.data == 1).one(), Subset(data=1))
|
|
eq_(sess.query(Subset).filter(Subset.data != 1).first(), None)
|
|
|
|
subset_select = sa.orm.class_mapper(Subset).persist_selectable
|
|
eq_(
|
|
sess.query(Subset).filter(subset_select.c.data == 1).one(),
|
|
Subset(data=1),
|
|
)
|