diff --git a/scrapy/utils/iterators.py b/scrapy/utils/iterators.py
index ce59c9719..b0688791e 100644
--- a/scrapy/utils/iterators.py
+++ b/scrapy/utils/iterators.py
@@ -17,7 +17,7 @@ logger = logging.getLogger(__name__)
def xmliter(obj, nodename):
"""Return a iterator of Selector's over all nodes of a XML document,
- given tha name of the node to iterate. Useful for parsing XML feeds.
+ given the name of the node to iterate. Useful for parsing XML feeds.
obj can be:
- a Response object
@@ -35,7 +35,7 @@ def xmliter(obj, nodename):
header_end = re_rsearch(HEADER_END_RE, text)
header_end = text[header_end[1]:].strip() if header_end else ''
- r = re.compile(r"<{0}[\s>].*?{0}>".format(nodename_patt), re.DOTALL)
+ r = re.compile(r'<%(np)s[\s>].*?%(np)s>' % {'np': nodename_patt}, re.DOTALL)
for match in r.finditer(text):
nodetext = header_start + match.group() + header_end
yield Selector(text=nodetext, type='xml').xpath('//' + nodename)[0]
@@ -48,7 +48,7 @@ def xmliter_lxml(obj, nodename, namespace=None, prefix='x'):
iterable = etree.iterparse(reader, tag=tag, encoding=reader.encoding)
selxpath = '//' + ('%s:%s' % (prefix, nodename) if namespace else nodename)
for _, node in iterable:
- nodetext = etree.tostring(node)
+ nodetext = etree.tostring(node, encoding='unicode')
node.clear()
xs = Selector(text=nodetext, type='xml')
if namespace:
@@ -128,8 +128,11 @@ def csviter(obj, delimiter=None, headers=None, encoding=None, quotechar=None):
def _body_or_str(obj, unicode=True):
- assert isinstance(obj, (Response, six.string_types, bytes)), \
- "obj must be Response or basestring, not %s" % type(obj).__name__
+ expected_types = (Response, six.text_type, six.binary_type)
+ assert isinstance(obj, expected_types), \
+ "obj must be %s, not %s" % (
+ " or ".join(t.__name__ for t in expected_types),
+ type(obj).__name__)
if isinstance(obj, Response):
if not unicode:
return obj.body
diff --git a/tests/test_utils_iterators.py b/tests/test_utils_iterators.py
index d42ed2c91..b2e3889a4 100644
--- a/tests/test_utils_iterators.py
+++ b/tests/test_utils_iterators.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
import os
import six
from twisted.trial import unittest
@@ -46,6 +47,60 @@ class XmliterTestCase(unittest.TestCase):
for e in self.xmliter(response, 'matchme...')]
self.assertEqual(nodenames, [['matchme...']])
+ def test_xmliter_unicode(self):
+ # example taken from https://github.com/scrapy/scrapy/issues/1665
+ body = u"""
+ <þingflokkar>
+ <þingflokkur id="26">
+
+
+ -
+
+
+
+ 80
+
+ þingflokkur>
+ <þingflokkur id="21">
+ Alþýðubandalag
+
+ Ab
+ Alþb.
+
+
+ 76
+ 123
+
+ þingflokkur>
+ <þingflokkur id="27">
+ Alþýðuflokkur
+
+ A
+ Alþfl.
+
+
+ 27
+ 120
+
+ þingflokkur>
+ þingflokkar>"""
+
+ for r in (
+ # with bytes
+ XmlResponse(url="http://example.com", body=body.encode('utf-8')),
+ # Unicode body needs encoding information
+ XmlResponse(url="http://example.com", body=body, encoding='utf-8')):
+
+ attrs = []
+ for x in self.xmliter(r, u'þingflokkur'):
+ attrs.append((x.xpath('@id').extract(),
+ x.xpath(u'./skammstafanir/stuttskammstöfun/text()').extract(),
+ x.xpath(u'./tímabil/fyrstaþing/text()').extract()))
+
+ self.assertEqual(attrs,
+ [([u'26'], [u'-'], [u'80']),
+ ([u'21'], [u'Ab'], [u'76']),
+ ([u'27'], [u'A'], [u'27'])])
def test_xmliter_text(self):
body = u"""onetwo"""
@@ -96,6 +151,10 @@ class XmliterTestCase(unittest.TestCase):
self.assertRaises(StopIteration, next, iter)
+ def test_xmliter_objtype_exception(self):
+ i = self.xmliter(42, 'product')
+ self.assertRaises(AssertionError, next, i)
+
def test_xmliter_encoding(self):
body = b'\n\n - Some Turkish Characters \xd6\xc7\xde\xdd\xd0\xdc \xfc\xf0\xfd\xfe\xe7\xf6
\n\n\n'
response = XmlResponse('http://www.example.com', body=body)
@@ -169,6 +228,9 @@ class LxmlXmliterTestCase(XmliterTestCase):
node = next(my_iter)
self.assertEqual(node.xpath('f:name/text()').extract(), ['African Coffee Table'])
+ def test_xmliter_objtype_exception(self):
+ i = self.xmliter(42, 'product')
+ self.assertRaises(TypeError, next, i)
class UtilsCsvTestCase(unittest.TestCase):
sample_feeds_dir = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'sample_data', 'feeds')