diff options
author | Eric Siegerman <eric.siegerman@rci.rogers.com> | 2015-09-28 12:08:19 -0400 |
---|---|---|
committer | Eric Siegerman <eric.siegerman@rci.rogers.com> | 2015-09-28 12:08:19 -0400 |
commit | 2e5e6ff96cc92f342bc53242703991f915741966 (patch) | |
tree | 5e830992e1833fe84e086bcf95898dcc6053af28 /lib/sqlalchemy/orm/query.py | |
parent | 91255618ddb47553774c620a23479adf88c27b74 (diff) | |
download | sqlalchemy-pr/203.tar.gz |
Remplement Query.one() in terms of .one_or_none()pr/203
Thanks to Mike Bayer for suggesting a simpler refactoring.
Diffstat (limited to 'lib/sqlalchemy/orm/query.py')
-rw-r--r-- | lib/sqlalchemy/orm/query.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/lib/sqlalchemy/orm/query.py b/lib/sqlalchemy/orm/query.py index f0b8969a2..16ead950e 100644 --- a/lib/sqlalchemy/orm/query.py +++ b/lib/sqlalchemy/orm/query.py @@ -2535,16 +2535,15 @@ class Query(object): :meth:`.Query.one_or_none` """ - ret = list(self) - - l = len(ret) - if l == 1: - return ret[0] - elif l == 0: - raise orm_exc.NoResultFound("No row was found for one()") - else: + try: + ret = self.one_or_none() + except orm_exc.MultipleResultsFound: raise orm_exc.MultipleResultsFound( "Multiple rows were found for one()") + else: + if ret is None: + raise orm_exc.NoResultFound("No row was found for one()") + return ret def scalar(self): """Return the first element of the first result or None |