13 lines
435 B
Python
13 lines
435 B
Python
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
|
|
class UserProfile(models.Model):
|
|
USER_TYPE_CHOICES = (
|
|
('guide', 'Guide'),
|
|
('customer', 'Customer'),
|
|
)
|
|
user = models.OneToOneField(User, on_delete=models.CASCADE)
|
|
user_type = models.CharField(max_length=10, choices=USER_TYPE_CHOICES)
|
|
|
|
def __str__(self):
|
|
return f'{self.user.username} - {self.get_user_type_display()}' |