[py3] Made 212b9826bd Python 3-friendly

This commit is contained in:
Aymeric Augustin 2012-08-18 17:47:21 +02:00
parent de3ad8bb2d
commit afc1bd7ab8
3 changed files with 11 additions and 11 deletions

View File

@ -126,13 +126,13 @@ strip_tags = allow_lazy(strip_tags)
def remove_tags(html, tags): def remove_tags(html, tags):
"""Returns the given HTML with given tags removed.""" """Returns the given HTML with given tags removed."""
tags = [re.escape(tag) for tag in tags.split()] tags = [re.escape(tag) for tag in tags.split()]
tags_re = u'(%s)' % u'|'.join(tags) tags_re = '(%s)' % '|'.join(tags)
starttag_re = re.compile(ur'<%s(/?>|(\s+[^>]*>))' % tags_re, re.U) starttag_re = re.compile(r'<%s(/?>|(\s+[^>]*>))' % tags_re, re.U)
endtag_re = re.compile(u'</%s>' % tags_re) endtag_re = re.compile('</%s>' % tags_re)
html = starttag_re.sub(u'', html) html = starttag_re.sub('', html)
html = endtag_re.sub(u'', html) html = endtag_re.sub('', html)
return html return html
remove_tags = allow_lazy(remove_tags, unicode) remove_tags = allow_lazy(remove_tags, six.text_type)
def strip_spaces_between_tags(value): def strip_spaces_between_tags(value):
"""Returns the given HTML with spaces between tags removed.""" """Returns the given HTML with spaces between tags removed."""

View File

@ -391,7 +391,7 @@ def slugify(value):
underscores) and converts spaces to hyphens. Also strips leading and underscores) and converts spaces to hyphens. Also strips leading and
trailing whitespace. trailing whitespace.
""" """
value = unicodedata.normalize('NFKD', value).encode('ascii', 'ignore') value = unicodedata.normalize('NFKD', value).encode('ascii', 'ignore').decode('ascii')
value = unicode(re.sub('[^\w\s-]', '', value).strip().lower()) value = re.sub('[^\w\s-]', '', value).strip().lower()
return mark_safe(re.sub('[-\s]+', '-', value)) return mark_safe(re.sub('[-\s]+', '-', value))
slugify = allow_lazy(slugify, unicode) slugify = allow_lazy(slugify, six.text_type)

View File

@ -116,8 +116,8 @@ class TestUtilsText(SimpleTestCase):
def test_slugify(self): def test_slugify(self):
items = ( items = (
(u'Hello, World!', 'hello-world'), ('Hello, World!', 'hello-world'),
(u'spam & eggs', 'spam-eggs'), ('spam & eggs', 'spam-eggs'),
) )
for value, output in items: for value, output in items:
self.assertEqual(text.slugify(value), output) self.assertEqual(text.slugify(value), output)