diff options
Diffstat (limited to 'lib/sqlalchemy/orm/shard.py')
| -rw-r--r-- | lib/sqlalchemy/orm/shard.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lib/sqlalchemy/orm/shard.py b/lib/sqlalchemy/orm/shard.py index c38bcdd96..5036bc115 100644 --- a/lib/sqlalchemy/orm/shard.py +++ b/lib/sqlalchemy/orm/shard.py @@ -1,6 +1,6 @@ from sqlalchemy.orm.session import Session from sqlalchemy.orm.query import Query -from sqlalchemy import exceptions +from sqlalchemy import exceptions, util __all__ = ['ShardedSession', 'ShardedQuery'] @@ -100,6 +100,7 @@ class ShardedQuery(Query): if self._shard_id is not None: return super(ShardedQuery, self).get(ident) else: + ident = util.to_list(ident) for shard_id in self.id_chooser(self, ident): o = self.set_shard(shard_id).get(ident, **kwargs) if o is not None: |