Add id_or_name URL converter
Change-Id: I615d28eb17963466c232562ff1a1791fe0d43879
This commit is contained in:
parent
37b566a65f
commit
b0e7786d9f
@ -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,
|
||||
}
|
||||
|
@ -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')
|
||||
|
Loading…
x
Reference in New Issue
Block a user