Add id_or_name URL converter

Change-Id: I615d28eb17963466c232562ff1a1791fe0d43879
This commit is contained in:
Yuriy Taraday 2016-03-25 15:46:32 +03:00
parent 37b566a65f
commit b0e7786d9f
2 changed files with 49 additions and 0 deletions

View File

@ -41,6 +41,22 @@ class Levels(routing.BaseConverter):
for p in parts)
return ''.join(p + '/' for p in quoted_parts)
class IdOrName(routing.BaseConverter):
"""Converter that matches either int or URL part including "/" as string"""
regex = '[^/].*?'
def to_python(self, value):
try:
return int(value)
except ValueError:
return value
def to_url(self, value):
return super(IdOrName, self).to_url(str(value))
ALL = {
'levels': Levels,
'id_or_name': IdOrName,
}

View File

@ -50,3 +50,36 @@ class TestLevels(base.TestCase):
res = self.mapad.build(
'l', {'levels': [('level1', 'value1'), ('level2', 'value2')]})
self.assertEqual(res, '/smth/level1/value1/level2/value2/values')
class TestIdOrName(base.TestCase):
def setUp(self):
super(TestIdOrName, self).setUp()
self.map = routing.Map([
routing.Rule('/<id_or_name:id_or_name>', endpoint='i'),
], converters={'id_or_name': converters.IdOrName})
self.mapad = self.map.bind('example.org', '/')
def test_int(self):
route, kwargs = self.mapad.match('/1')
self.assertEqual(kwargs['id_or_name'], 1)
def test_name(self):
route, kwargs = self.mapad.match('/name')
self.assertEqual(kwargs['id_or_name'], 'name')
def test_name_with_slashes(self):
route, kwargs = self.mapad.match('/name/with/slashes')
self.assertEqual(kwargs['id_or_name'], 'name/with/slashes')
def test_reverse_int(self):
res = self.mapad.build('i', {'id_or_name': 1})
self.assertEqual(res, '/1')
def test_reverse_name(self):
res = self.mapad.build('i', {'id_or_name': 'name'})
self.assertEqual(res, '/name')
def test_reverse_name_with_slashes(self):
res = self.mapad.build('i', {'id_or_name': 'name/with/slashes'})
self.assertEqual(res, '/name/with/slashes')