Ticket #3530: permalink_with_additional_function_arg.diff

django/db/models/__init__.py

@@ -18 +18 @@
 
 # Decorator. Takes a function that returns a tuple in this format:
 #     (viewname, viewargs, viewkwargs)
+#   Optionally takes a function which should either return an object with
+#     an attribute 'urlconf' or directly a python list which is used instead of
+#     settings.ROOT_URLCONF
 # Returns a function that calls urlresolvers.reverse() on that data, to return
 # the URL for those parameters.
-def permalink(func):
+def permalink(func, get_urlconf_func = None):
     from django.core.urlresolvers import reverse
     def inner(*args, **kwargs):
+        # Find urlconf ...
+        urlconf = None
+        if get_urlconf_func != None:
+            urlconf = get_urlconf_func()
+            if type(urlconf) != list:
+                # If type is no list, we assume it is a request object and
+                # look for a 'urlconf' attribute
+                urlconf = getattr(urlconf, 'urlconf')
+        
         bits = func(*args, **kwargs)
         viewname = bits[0]
-        return reverse(bits[0], None, *bits[1:3])
+        return reverse(bits[0], urlconf, *bits[1:3])
     return inner
 
 class LazyDate(object):