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

class User(AbstractUser):
    ROLE_CHOICES = [('USER','User'), ('ADMIN','Admin')]
    role = models.CharField(max_length=10, choices=ROLE_CHOICES, default='USER')
    dive_certificate = models.CharField(max_length=255, blank=True)
    dive_base = models.ForeignKey('divebases.DiveBase', null=True, blank=True, on_delete=models.SET_NULL)