Commit 600ee47d authored by Stefan Behnel's avatar Stefan Behnel

small refactoring

parent 03715858
......@@ -45,22 +45,24 @@ class DictIterTransform(Visitor.VisitorTransform):
PyDict_Next_entry = Symtab.Entry(
PyDict_Next_name, PyDict_Next_name, PyDict_Next_func_type)
def visit_Node(self, node):
# descend into statements (loops) and nodes (comprehensions)
self.visitchildren(node)
return node
def visit_ForInStatNode(self, node):
self.visitchildren(node)
iterator = node.iterator.sequence
if iterator.type is Builtin.dict_type:
# like iterating over dict.keys()
dict_obj = iterator
keys = True
values = False
else:
return self._transform_dict_iteration(
node, dict_obj=iterator, keys=True, values=False)
if not isinstance(iterator, ExprNodes.SimpleCallNode):
return node
function = iterator.function
if not isinstance(function, ExprNodes.AttributeNode):
return node
if function.obj.type != Builtin.dict_type:
return node
if isinstance(function, ExprNodes.AttributeNode) and \
function.obj.type == Builtin.dict_type:
dict_obj = function.obj
method = function.attribute
......@@ -73,7 +75,11 @@ class DictIterTransform(Visitor.VisitorTransform):
keys = values = True
else:
return node
return self._transform_dict_iteration(
node, dict_obj, keys, values)
return node
def _transform_dict_iteration(self, node, dict_obj, keys, values):
py_object_ptr = PyrexTypes.c_void_ptr_type
temps = []
......@@ -219,11 +225,6 @@ class DictIterTransform(Visitor.VisitorTransform):
stats = result_code
))
def visit_Node(self, node):
# descend into statements (loops) and nodes (comprehensions)
self.visitchildren(node)
return node
class SwitchTransform(Visitor.VisitorTransform):
"""
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment