implented suggested code consolidation

zopefoundation · Nov 18, 2016 · 930d15d · 930d15d
1 parent 39d088e
commit 930d15d
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/src/ZODB/Connection.py b/src/ZODB/Connection.py
@@ -1324,13 +1324,8 @@ def _extension(self, v):
 
     def data(self, ob):
         try:
-            data = self._data
-        except AttributeError:
-            raise KeyError(ob)
-
-        try:
-            return data[id(ob)]
-        except KeyError:
+            return self._data[id(ob)]
+        except (AttributeError, KeyError):
             raise KeyError(ob)
 
     def set_data(self, ob, ob_data):