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.
This commit is contained in:
Elrond 2012-12-16 00:45:57 +01:00
parent 7880f03e2d
commit 05501c5742
2 changed files with 17 additions and 14 deletions

View File

@ -184,7 +184,7 @@ class MediaGoblinApp(object):
mg_request.setup_user_in_request(request) mg_request.setup_user_in_request(request)
try: try:
endpoint, url_values = map_adapter.match() found_rule, url_values = map_adapter.match(return_rule=True)
request.matchdict = url_values request.matchdict = url_values
except HTTPException as exc: except HTTPException as exc:
# Stop and render exception # Stop and render exception
@ -192,7 +192,7 @@ class MediaGoblinApp(object):
request, exc, request, exc,
exc.get_description(environ))(environ, start_response) exc.get_description(environ))(environ, start_response)
controller = endpoint_to_controller(endpoint) controller = endpoint_to_controller(found_rule)
# pass the request through our meddleware classes # pass the request through our meddleware classes
try: try:

View File

@ -24,18 +24,28 @@ _log = logging.getLogger(__name__)
url_map = Map() 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 empty(self):
new_rule = Rule.empty(self)
new_rule.gmg_controller = self.gmg_controller
return new_rule
def endpoint_to_controller(endpoint): def endpoint_to_controller(rule):
view_func = view_functions[endpoint] endpoint = rule.endpoint
view_func = rule.gmg_controller
_log.debug('endpoint: {0} view_func: {1}'.format(endpoint, view_func)) _log.debug('endpoint: {0} view_func: {1}'.format(endpoint, view_func))
# import the endpoint, or if it's already a callable, call that # import the endpoint, or if it's already a callable, call that
if isinstance(view_func, basestring): if isinstance(view_func, basestring):
view_func = import_component(view_func) view_func = import_component(view_func)
view_functions[endpoint] = view_func rule.gmg_controller = view_func
return view_func return view_func
@ -44,14 +54,7 @@ def add_route(endpoint, url, controller):
""" """
Add a route to the url mapping Add a route to the url mapping
""" """
# XXX: We cannot use this, since running tests means that the plugin url_map.add(MGRoute(endpoint, url, controller))
# 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))
def mount(mountpoint, routes): def mount(mountpoint, routes):