Ticket #11891: get_absolute_url_attributes.diff

django/db/models/base.py

@@ -18 +18 @@
 from django.db import connection, transaction, DatabaseError
 from django.db.models import signals
 from django.db.models.loading import register_models, get_model
-from django.utils.functional import curry
+from django.utils.functional import curry, update_wrapper
 from django.utils.encoding import smart_str, force_unicode, smart_unicode
 from django.conf import settings
 
@@ -232 +232 @@
             cls.__doc__ = "%s(%s)" % (cls.__name__, ", ".join([f.attname for f in opts.fields]))
 
         if hasattr(cls, 'get_absolute_url'):
-            cls.get_absolute_url = curry(get_absolute_url, opts, cls.get_absolute_url)
+            cls.get_absolute_url = update_wrapper(curry(get_absolute_url, opts, cls.get_absolute_url),
+                                                  cls.get_absolute_url)
 
         signals.class_prepared.send(sender=cls)
 

tests/regressiontests/views/models.py

@@ -39 +39 @@
 
     def get_absolute_url(self):
         return '/urlarticles/%s/' % self.slug
+    get_absolute_url.purge = True
 
 class DateArticle(BaseArticle):
     """

tests/regressiontests/views/tests/defaults.py

@@ -4 +4 @@
 from django.test import TestCase
 from django.contrib.contenttypes.models import ContentType
 
-from regressiontests.views.models import Author, Article
+from regressiontests.views.models import Author, Article, UrlArticle
 
 class DefaultsTests(TestCase):
     """Test django views in django/views/defaults.py"""
@@ -37 +37 @@
         "The server_error view raises a 500 status"
         response = self.client.get('/views/server_error/')
         self.assertEquals(response.status_code, 500)
+
+    def test_get_absolute_url_attributes(self):
+        "A model can set attributes on the get_absolute_url method"
+        self.assertTrue(getattr(UrlArticle.get_absolute_url, 'purge', False), 
+                        'The attributes of the original get_absolute_url must be added.')
+        article = UrlArticle.objects.get(pk=1)
+        self.assertTrue(getattr(article.get_absolute_url, 'purge', False),
+                        'The attributes of the original get_absolute_url must be added.')