Implement more efficient __contains__

sebastinas · sebastinas · commit da150e4284d4 · 2021-05-12T23:30:30.000+02:00
diff --git a/bpython/curtsiesfrontend/manual_readline.py b/bpython/curtsiesfrontend/manual_readline.py
@@ -22,13 +22,6 @@ class AbstractEdits:
         "cut_buffer": "there",
     }
 
-    def __contains__(self, key):
-        try:
-            self[key]
-        except KeyError:
-            return False
-        else:
-            return True
 
     def add(self, key, func, overwrite=False):
         if key in self:
@@ -75,6 +68,9 @@ def call_without_cut(self, key, **kwargs):
         r = self.call_for_two(key, **kwargs)
         return r[:2]
 
+    def __contains__(self, key):
+        return key in self.simple_edits or key in self.cut_buffer_edits
+
     def __getitem__(self, key):
         if key in self.simple_edits:
             return self.simple_edits[key]
