Guest User

Untitled

a guest
Jul 21st, 2018
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.68 KB | None | 0 0
  1. accounts/models.py
  2.  
  3.  
  4. class Role(models.Model):
  5.  
  6. ROLE_CHOICES = (
  7. ('agent', 'Agent'),
  8. ('agency', 'Agency'),
  9. ('manufacturer', 'Manufacturer'),
  10. )
  11. role = models.CharField(max_length=15, choices=ROLE_CHOICES)
  12.  
  13. def __str__(self):
  14. return self.role
  15.  
  16.  
  17. class User(AbstractUser):
  18.  
  19. role = models.ForeignKey(
  20. Role,
  21. on_delete=models.CASCADE,
  22. blank=True,
  23. null=True,
  24. )
  25.  
  26. def __str__(self):
  27. return self.username
  28.  
  29. def save(self, *args, **kwargs):
  30. if not self.pk: # the instance is created
  31. self.role, created = Role.objects.get_or_create(role="agent")
  32. return super().save(*args, **kwargs)
Add Comment
Please, Sign In to add comment