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)
|
for p in parts)
|
||||||
return ''.join(p + '/' for p in quoted_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 = {
|
ALL = {
|
||||||
'levels': Levels,
|
'levels': Levels,
|
||||||
|
'id_or_name': IdOrName,
|
||||||
}
|
}
|
||||||
|
@ -50,3 +50,36 @@ class TestLevels(base.TestCase):
|
|||||||
res = self.mapad.build(
|
res = self.mapad.build(
|
||||||
'l', {'levels': [('level1', 'value1'), ('level2', 'value2')]})
|
'l', {'levels': [('level1', 'value1'), ('level2', 'value2')]})
|
||||||
self.assertEqual(res, '/smth/level1/value1/level2/value2/values')
|
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