Mercurial > p > roundup > code
view roundup/web/router.py @ 4924:6ee1844019d5 routing
routing: Add DEBUG flag for troubleshooting
| author | anatoly techtonik <techtonik@gmail.com> |
|---|---|
| date | Wed, 20 Aug 2014 13:36:08 +0300 |
| parents | 35dc9191394d |
| children | 997fa47c92d5 |
line wrap: on
line source
#!/usr/bin/env python """ The purpose of router is to make Roundup URL scheme configurable and allow extensions add their own handlers and URLs to tracker. Public domain work by: anatoly techtonik <techtonik@gmail.com> """ DEBUG = False import re # --- Example URL mapping class NamedObject(object): """Object that outputs given name when printed""" def __init__(self, name): self.name = name def __repr__(self): return self.name ExampleHandler = NamedObject('ExampleHandler') ExampleFileHandler = NamedObject('ExampleFileHandler') EXAMPLE_URL_MAP = ( 'static/(.*)', ExampleFileHandler, 'index', ExampleHandler ) # --- Regexp based router class Router(object): def __init__(self, urlmap=[]): """ `urlmap` is a list (pattern, handler, pattern, ...) pattern should have no leading slash """ self.urlmap = urlmap def get_handler(self, urlpath): """ `urlpath` is a part of url /that/looks?like=this (leading slash is optional) returns tuple (handler, arguments) or (None, ()) """ # strip leading slashes before matching path = urlpath.lstrip('/') for i in range(0, len(self.urlmap), 2): pattern, handler = self.urlmap[i], self.urlmap[i+1] if DEBUG: print('router: matching %s' % pattern) match = re.match(pattern, path) if match: return handler, match.groups() return (None, ()) # [ ] len(urlmap) should be even to avoid errors # (find a way to explain this to users) if __name__ == '__main__': import unittest class test_Router(unittest.TestCase): def test_example_routes(self): router = Router(EXAMPLE_URL_MAP) self.assertEquals(router.get_handler(''), (None, ())) handler, params = router.get_handler('/index') self.assertEquals(handler, ExampleHandler) self.assertEquals(params, tuple()) unittest.main()
