diff options
author | Elrond <elrond+mediagoblin.org@samba-tng.org> | 2012-12-16 00:50:20 +0100 |
---|---|---|
committer | Elrond <elrond+mediagoblin.org@samba-tng.org> | 2012-12-23 12:26:34 +0100 |
commit | 3d9143323019e0793451eac60eef8e55c09f6c47 (patch) | |
tree | fad24f3460af09c155d6c61909d54d5864919d51 /mediagoblin/tools/routing.py | |
parent | 48cf435d718b47e33cb98b048e5b58d79a46b486 (diff) | |
download | mediagoblin-3d9143323019e0793451eac60eef8e55c09f6c47.tar.lz mediagoblin-3d9143323019e0793451eac60eef8e55c09f6c47.tar.xz mediagoblin-3d9143323019e0793451eac60eef8e55c09f6c47.zip |
Move things from routing.py to tools/routing.py
This stops a cyclic import.
Move add_route, mount and endpoint_to_controller into
tools/routing.py and change all callers.
Diffstat (limited to 'mediagoblin/tools/routing.py')
-rw-r--r-- | mediagoblin/tools/routing.py | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/mediagoblin/tools/routing.py b/mediagoblin/tools/routing.py new file mode 100644 index 00000000..6c5acbec --- /dev/null +++ b/mediagoblin/tools/routing.py @@ -0,0 +1,63 @@ +# GNU MediaGoblin -- federated, autonomous media hosting +# Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS. +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +import logging + +from werkzeug.routing import Map, Rule +from mediagoblin.tools.common import import_component + + +_log = logging.getLogger(__name__) + +url_map = Map() + +view_functions = {} + + +def endpoint_to_controller(endpoint): + view_func = view_functions[endpoint] + + _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 + + return view_func + + +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)) + + +def mount(mountpoint, routes): + """ + Mount a bunch of routes to this mountpoint + """ + for endpoint, url, controller in routes: + url = "%s/%s" % (mountpoint.rstrip('/'), url.lstrip('/')) + add_route(endpoint, url, controller) |