Add __bool__ and __next__

sebastinas · sebastinas · commit 4e477f8a429d · 2015-02-04T18:13:39.000+01:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -199,6 +199,9 @@ def __nonzero__(self):
         """MatchesIterator is False when word hasn't been replaced yet"""
         return self.index != -1
 
+    def __bool__(self):
+        return self.__nonzero__()
+
     @property
     def candidate_selected(self):
         """True when word selected/replaced, False when word hasn't been replaced yet"""
@@ -212,10 +215,13 @@ def current(self):
             raise ValueError('No current match.')
         return self.matches[self.index]
 
-    def next(self):
+    def __next__(self):
         self.index = (self.index + 1) % len(self.matches)
         return self.matches[self.index]
 
+    def next(self):
+        return self.__next__()
+
     def previous(self):
         if self.index <= 0:
             self.index = len(self.matches)
