diff options
Diffstat (limited to 'mediagoblin/routing.py')
| -rw-r--r-- | mediagoblin/routing.py | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/mediagoblin/routing.py b/mediagoblin/routing.py index defbc4ba..dc22ba09 100644 --- a/mediagoblin/routing.py +++ b/mediagoblin/routing.py @@ -14,12 +14,33 @@ # 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 +from mediagoblin.tools.pluginapi import PluginManager + + +_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 @@ -33,6 +54,7 @@ def add_route(endpoint, url, controller): url_map.add(Rule(url, endpoint=endpoint)) + def mount(mountpoint, routes): """ Mount a bunch of routes to this mountpoint @@ -41,6 +63,14 @@ def mount(mountpoint, routes): url = "%s/%s" % (mountpoint.rstrip('/'), url.lstrip('/')) add_route(endpoint, url, controller) + +def get_url_map(): + for route in PluginManager().get_routes(): + add_route(*route) + + return url_map + + add_route('index', '/', 'mediagoblin.views:root_view') from mediagoblin.admin.routing import admin_routes |