trac#687: Add unit tests for `redirect` and `redirect_obj`.

[mediagoblin.git] / mediagoblin / routing.py

diff --git a/mediagoblin/routing.py b/mediagoblin/routing.py
index defbc4ba80cf2cb8438cef8a8c43cdb62668cb67..1fbc6e9b83f60e1a448f1db103b796ada197e2e4 100644 (file)
--- a/mediagoblin/routing.py
+++ b/mediagoblin/routing.py
@@ -14,42 +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/>.
 
-from werkzeug.routing import Map, Rule
+import logging
 
-url_map = Map()
+from mediagoblin.tools.routing import add_route, mount, url_map
+from mediagoblin.tools.pluginapi import PluginManager
+from mediagoblin.moderation.routing import moderation_routes
+from mediagoblin.auth.routing import auth_routes
 
-view_functions = {}
 
-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'
+_log = logging.getLogger(__name__)
 
-    view_functions.update({endpoint: controller})
 
-    url_map.add(Rule(url, endpoint=endpoint))
+def get_url_map():
+    add_route('index', '/', 'mediagoblin.views:root_view')
+    add_route('terms_of_service','/terms_of_service',
+        'mediagoblin.views:terms_of_service'),
+    mount('/auth', auth_routes)
+    mount('/mod', moderation_routes)
 
-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)
+    import mediagoblin.submit.routing
+    import mediagoblin.user_pages.routing
+    import mediagoblin.edit.routing
+    import mediagoblin.listings.routing
+    import mediagoblin.notifications.routing
+    import mediagoblin.oauth.routing
+    import mediagoblin.api.routing
 
-add_route('index', '/', 'mediagoblin.views:root_view')
+    for route in PluginManager().get_routes():
+        add_route(*route)
 
-from mediagoblin.admin.routing import admin_routes
-from mediagoblin.auth.routing import auth_routes
-mount('/auth', auth_routes)
-mount('/a', admin_routes)
-
-import mediagoblin.submit.routing
-import mediagoblin.user_pages.routing
-import mediagoblin.edit.routing
-import mediagoblin.webfinger.routing
-import mediagoblin.listings.routing
+    return url_map