Remove generator deprecated warning

falhenaki · May 24, 2018 · e3cfa00 · e3cfa00
1 parent e706239
commit e3cfa00
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/wfuzz/core.py b/src/wfuzz/core.py
@@ -69,7 +69,10 @@ def __iter__(self):
 
         def _gen(self):
             while 1:
-                payload_list = next(self.__payload)
+                try:
+                    payload_list = next(self.__payload)
+                except StopIteration:
+                    return
 
                 for name in self.__encoders:
                     if name.find('@') > 0: