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

from sorl.thumbnail import ImageField


class User(AbstractUser):
    USERNAME_FIELD = "email"
    REQUIRED_FIELDS = []

    username = None
    email = models.EmailField(unique=True)


class FallbackUrl(models.Model):
    gras_id = models.CharField(blank=False, unique=True, max_length=50)
    url = models.URLField(blank=False)

    def __str__(self):
        return f"{self.gras_id}: {self.url}"


class LabelImage(models.Model):
    label_id = models.CharField(blank=False, unique=True, max_length=50)
    image = ImageField(upload_to="labels", blank=False)

    def __str__(self):
        return self.label_id