from django.db import models
from django.contrib.auth.models import User


class UserProfile(models.Model):
    USER_TYPE_CHOICES = [
        ("publico", "Público"),
        ("user", "Usuario"),
        ("administrator", "Administrador"),
    ]
    user = models.OneToOneField(
        User, on_delete=models.CASCADE, related_name="profile"
    )
    user_type = models.CharField(
        max_length=20, choices=USER_TYPE_CHOICES, default="user"
    )

    def __str__(self):
        return f"{self.user.username} - {self.user_type}"