Ticket #4566: reverse-speedup.diff

core/urlresolvers.py

@@ -11 +11 @@
-from django.core.exceptions import ImproperlyConfigured, ViewDoesNotExist
-import re
-
+def cacheable(func, cache):
+    def wrapper(arg):
+        if arg in cache:
+            return cache[arg]
+        
+        result = func(arg)
+        cache[arg] = result
+        return result
+    return wrapper
+
+resolver_cache = {}
+
+def get_resolver(urlconf):
+    if urlconf is None:
+        from django.conf import settings
+        urlconf = settings.ROOT_URLCONF
+    return RegexURLResolver(r'^/', urlconf)
+get_resolver = cacheable(get_resolver, resolver_cache)
+
-class Resolver404(Http404):
-    pass
-
@@ -160 +179 @@
-        self.urlconf_name = urlconf_name
-        self.callback = None
-        self.default_kwargs = default_kwargs or {}
+        self.reverse_dict = {}
+        
+        for pattern in reversed(self.urlconf_module.urlpatterns):
+            if isinstance(pattern, RegexURLResolver):
+                for key, value in pattern.reverse_dict.iteritems():
+                    self.reverse_dict[key] = (pattern,) + value
+            else:
+                self.reverse_dict[pattern.callback] = (pattern,)
+                self.reverse_dict[pattern.name] = (pattern,)
-
-    def resolve(self, path):
-        tried = []
@@ -209 +237 @@
-        return self._resolve_special('500')
-
-    def reverse(self, lookup_view, *args, **kwargs):
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+
+
+
+
-        raise NoReverseMatch
-
-    def reverse_helper(self, lookup_view, *args, **kwargs):
@@ -234 +248 @@
-        result = reverse_helper(self.regex, *args, **kwargs)
-        return result + sub_match
-
+callable_cache = {}
+def make_callable(lookup_view):
+    if not callable(lookup_view):
+        mod_name, func_name = get_mod_func(lookup_view)
+        if func_name != '':
+            try:
+                lookup_view = getattr(__import__(mod_name, {}, {}, ['']), func_name)
+            except (ImportError, AttributeError):
+                if func_name != '':
+                    raise NoReverseMatch
+    return lookup_view
+make_callable = cacheable(make_callable, callable_cache)
+
-def resolve(path, urlconf=None):
-
-
-
-
+
-    return resolver.resolve(path)
-
-def reverse(viewname, urlconf=None, args=None, kwargs=None):
-    args = args or []
-    kwargs = kwargs or {}
-
-
-
-
+
-    return '/' + resolver.reverse(viewname, *args, **kwargs)