Ident refactoring

A. Jesse Jiryu Davis · A. Jesse Jiryu Davis · commit 0339411b01ee · 2013-05-22T20:33:15.000-04:00
diff --git a/pymongo/thread_util.py b/pymongo/thread_util.py
@@ -56,6 +56,10 @@ def watch(self, callback):
         """
         raise NotImplementedError
 
+    def watching(self):
+        """Is the current thread being watched for death?"""
+        return self.get() in self._refs
+
 
 class ThreadIdent(Ident):
     class _DummyLock(object):
@@ -99,15 +103,6 @@ def watch(self, callback):
         vigil = self._make_vigil()
         self._refs[id(vigil)] = weakref.ref(vigil, callback)
 
-    def watching(self):
-        """Is the current thread being watched for death?"""
-        tid = self.get()
-        if tid not in self._refs:
-            return False
-        # Check that the weakref is active, if not the thread has died
-        # This fixes the case where a thread id gets reused
-        return self._refs[tid]()
-
 
 class GreenletIdent(Ident):
     def get(self):
