get SQL refs from session (bp sql-query-get)

Change-Id: I2200e33868d50bb69089f3108a5a4c061afccd6e

1 files changed, 3 insertions, 3 deletions

diff --git a/keystone/policy/backends/sql.py b/keystone/policy/backends/sql.py
index 2472c1ed..c1eff268 100644
--- a/keystone/policy/backends/sql.py
+++ b/keystone/policy/backends/sql.py
@@ -53,10 +53,10 @@ class Policy(sql.Base, rules.Policy):
 
     def _get_policy(self, session, policy_id):
         """Private method to get a policy model object (NOT a dictionary)."""
-        try:
-            return session.query(PolicyModel).filter_by(id=policy_id).one()
-        except sql.NotFound:
+        ref = session.query(PolicyModel).get(policy_id)
+        if not ref:
             raise exception.PolicyNotFound(policy_id=policy_id)
+        return ref
 
     def get_policy(self, policy_id):
         session = self.get_session()