summaryrefslogtreecommitdiff
path: root/tests/model_permalink/models.py
blob: 9074837a8b56f8053c36fb51ec888f48537ec0a3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import warnings

from django.db import models
from django.utils.deprecation import RemovedInDjango21Warning


def set_attr(name, value):
    def wrapper(function):
        setattr(function, name, value)
        return function
    return wrapper


with warnings.catch_warnings():
    warnings.simplefilter('ignore', category=RemovedInDjango21Warning)

    class Guitarist(models.Model):
        name = models.CharField(max_length=50)
        slug = models.CharField(max_length=50)

        @models.permalink
        def url(self):
            "Returns the URL for this guitarist."
            return ('guitarist_detail', [self.slug])

        @models.permalink
        @set_attr('attribute', 'value')
        def url_with_attribute(self):
            "Returns the URL for this guitarist and holds an attribute"
            return ('guitarist_detail', [self.slug])