diff options
Diffstat (limited to 'rest_framework/tests/models.py')
| -rw-r--r-- | rest_framework/tests/models.py | 28 | 
1 files changed, 28 insertions, 0 deletions
| diff --git a/rest_framework/tests/models.py b/rest_framework/tests/models.py new file mode 100644 index 00000000..4cae68b6 --- /dev/null +++ b/rest_framework/tests/models.py @@ -0,0 +1,28 @@ +from django.db import models +from django.contrib.auth.models import Group + +class CustomUser(models.Model): +    """ +    A custom user model, which uses a 'through' table for the foreign key +    """ +    username = models.CharField(max_length=255, unique=True) +    groups = models.ManyToManyField( +        to=Group, blank=True, null=True, through='UserGroupMap' +    ) + +    @models.permalink +    def get_absolute_url(self): +        return ('custom_user', (), { +            'pk': self.id +    }) + + +class UserGroupMap(models.Model): +    user = models.ForeignKey(to=CustomUser) +    group = models.ForeignKey(to=Group)       +     +    @models.permalink +    def get_absolute_url(self): +        return ('user_group_map', (), { +            'pk': self.id +        }) | 
