from django.db import models
from django.conf import settings

# Create your models here.

class EggImage(models.Model):
    image_path = models.FilePathField(path=settings.IMAGE_DIR)
    num_tags =  models.IntegerField()
    def __str__(self):
        return str(self.image_path)+": "+str(self.num_tags)

class EggTagger(models.Model):
    name = models.CharField(max_length=100)
    experience = models.CharField(max_length=1, choices={
        "B":"Beginner",
        "I":"Intermediate",
        "E":"Expert"})
    def __str__(self):
        return self.name
    
class EggTag(models.Model):
    image = models.ForeignKey(EggImage, on_delete=models.CASCADE)
    tagger = models.ForeignKey(EggTagger, on_delete=models.CASCADE)
    stage = models.IntegerField(null=True, blank=True)
    notag_reason = models.CharField(max_length=1, choices={
        "":"",
        "N":"No embryo in image",
        "A":"Abnormal embryo",
        "U":"I'm not sure"})
    confidence = models.IntegerField()
    alt_stage = models.IntegerField(null=True, blank=True)
    def __str__(self):
        return str(self.image)+": "+str(self.stage)+": "+str(self.tagger.name)