From 05501c5742c85d0e2c855a02800d1f50f9df6c3d Mon Sep 17 00:00:00 2001 From: Elrond Date: Sun, 16 Dec 2012 00:45:57 +0100 Subject: Rewrite routing using new MGRoute class MGRoute subclasses Rule(): Rule doesn't have a way to tag extra data, like the controller function, we need. So MGRoute has a new attribute .gmg_controller, which holds this. Rewrite everything to use this new Rule variant and drop all the other stuff that mapped endpoints to controller functions, mostly. --- mediagoblin/tools/routing.py | 27 +++++++++++++++------------ 1 file changed, 15 insertions(+), 12 deletions(-) (limited to 'mediagoblin/tools/routing.py') diff --git a/mediagoblin/tools/routing.py b/mediagoblin/tools/routing.py index 6c5acbec..791cd1e6 100644 --- a/mediagoblin/tools/routing.py +++ b/mediagoblin/tools/routing.py @@ -24,18 +24,28 @@ _log = logging.getLogger(__name__) url_map = Map() -view_functions = {} +class MGRoute(Rule): + def __init__(self, endpoint, url, controller): + Rule.__init__(self, url, endpoint=endpoint) + self.gmg_controller = controller -def endpoint_to_controller(endpoint): - view_func = view_functions[endpoint] + def empty(self): + new_rule = Rule.empty(self) + new_rule.gmg_controller = self.gmg_controller + return new_rule + + +def endpoint_to_controller(rule): + endpoint = rule.endpoint + view_func = rule.gmg_controller _log.debug('endpoint: {0} view_func: {1}'.format(endpoint, view_func)) # import the endpoint, or if it's already a callable, call that if isinstance(view_func, basestring): view_func = import_component(view_func) - view_functions[endpoint] = view_func + rule.gmg_controller = view_func return view_func @@ -44,14 +54,7 @@ def add_route(endpoint, url, controller): """ Add a route to the url mapping """ - # XXX: We cannot use this, since running tests means that the plugin - # routes will be populated over and over over the same session. - # - # assert endpoint not in view_functions.keys(), 'Trying to overwrite a rule' - - view_functions.update({endpoint: controller}) - - url_map.add(Rule(url, endpoint=endpoint)) + url_map.add(MGRoute(endpoint, url, controller)) def mount(mountpoint, routes): -- cgit v1.2.3