Commit f1d781a6 authored by Stefan Behnel's avatar Stefan Behnel

fix char* string iteration when we actually need to coerce to a Python string for iteration

parent f8e85bee
...@@ -90,15 +90,16 @@ class IterationTransform(Visitor.VisitorTransform): ...@@ -90,15 +90,16 @@ class IterationTransform(Visitor.VisitorTransform):
node, dict_obj=iterator, keys=True, values=False) node, dict_obj=iterator, keys=True, values=False)
# C array (slice) iteration? # C array (slice) iteration?
plain_iterator = iterator
if isinstance(iterator, ExprNodes.CoerceToPyTypeNode): if isinstance(iterator, ExprNodes.CoerceToPyTypeNode):
iterator = iterator.arg plain_iterator = iterator.arg
if isinstance(iterator, ExprNodes.SliceIndexNode) and \ if isinstance(plain_iterator, ExprNodes.SliceIndexNode) and \
(iterator.base.type.is_array or iterator.base.type.is_ptr): (plain_iterator.base.type.is_array or plain_iterator.base.type.is_ptr):
return self._transform_carray_iteration(node, iterator) return self._transform_carray_iteration(node, plain_iterator)
elif isinstance(iterator, ExprNodes.IndexNode) and \ elif isinstance(plain_iterator, ExprNodes.IndexNode) and \
isinstance(iterator.index, (ExprNodes.SliceNode, ExprNodes.CoerceFromPyTypeNode)) and \ isinstance(plain_iterator.index, (ExprNodes.SliceNode, ExprNodes.CoerceFromPyTypeNode)) and \
(iterator.base.type.is_array or iterator.base.type.is_ptr): (plain_iterator.base.type.is_array or plain_iterator.base.type.is_ptr):
return self._transform_carray_iteration(node, iterator) return self._transform_carray_iteration(node, plain_iterator)
elif iterator.type.is_array: elif iterator.type.is_array:
return self._transform_carray_iteration(node, iterator) return self._transform_carray_iteration(node, iterator)
elif iterator.type in (Builtin.bytes_type, Builtin.unicode_type): elif iterator.type in (Builtin.bytes_type, Builtin.unicode_type):
......
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