Implement __contains__() special method for TaskRegistry class

dnaeon · Jan 26, 2015 · 3ca7c39 · 3ca7c39
1 parent 86319b1
commit 3ca7c39
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/vpoller/registry.py b/src/vpoller/registry.py
@@ -38,6 +38,9 @@ class TaskRegistry(object):
     def __init__(self):
         self._registry = {}
 
+    def __contains__(self, item):
+        return item in self._registry
+
     def register(self, name, required, fn):
         """
         Register a new task