AssertImportHook.enabled predicate property

DasIch · Feb 12, 2011 · 699dea7 · 699dea7
1 parent ca4b369
commit 699dea7
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/attest/hook.py b/attest/hook.py
@@ -189,6 +189,12 @@ def visit_Assert(self, node):
                    args=args, keywords=[], starargs=None, kwargs=None)), node)
 
 
+class AssertImportHookEnabledDescriptor(object):
+
+    def __get__(self, instance, owner):
+        return any(isinstance(ih, owner) for ih in sys.meta_path)
+
+
 class AssertImportHook(object):
     """A :term:`finder` and :term:`loader` that transforms imported modules
     with :class:`AssertTransformer`.
@@ -197,6 +203,9 @@ class AssertImportHook(object):
 
     """
 
+    #: Class property, :const:`True` if the hook is enabled.
+    enabled = AssertImportHookEnabledDescriptor()
+
     @classmethod
     def enable(cls):
         """Enable the import hook."""