mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-14 13:46:59 +03:00
28 lines
829 B
Python
28 lines
829 B
Python
|
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
|
||
|
|
})
|