Improve code climate.

c0fec0de · Nov 15, 2017 · 733bbda · 733bbda
1 parent 911a675
commit 733bbda
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/anytree/iterators.py b/anytree/iterators.py
@@ -32,15 +32,17 @@ def __init__(self, node, filter_=None, stop=None, maxlevel=None):
 
     def __init(self):
         node = self.node
-        filter_ = self.filter_
-        stop = self.stop
         maxlevel = self.maxlevel
-        if filter_ is None:
+        if self.filter_ is None:
             def filter_(node):
                 return True
-        if stop is None:
+        else:
+            filter_ = self.filter_
+        if self.stop is None:
             def stop(node):
                 return False
+        else:
+            stop = self.stop
         children = [] if AbstractIter._abort_at_level(1, maxlevel) else AbstractIter._get_children([node], stop)
         return self._iter(children, filter_, stop, maxlevel)