from django.db import models

from ctrack.organisations.models import Organisation

# TODO - thinking about whether TriageAssessment can be converted into an inherited class
# e.g. we inherit from
# class Assessment(models.Model):
#     descriptor = models.CharField(max_length=100)
#     date_entered = models.DateTimeField(auto_now_add=True)
#     modified = models.DateTimeField(auto_now=True)
#
#     class Meta:
#         abstract = True


class Ranking(models.Model):
    RANKING_TYPE = [(1, "Triage"), (2, "First Assessment"), (3, "Validation")]
    descriptor = models.CharField(max_length=100)
    description = models.TextField(max_length=250)
    type = models.IntegerField(choices=RANKING_TYPE, default=1)


    def __str__(self):
        return self.descriptor


class ConfidenceAssessment(models.Model):
    RANKING_TYPE = [("GOOD", "Good"), ("BAD", "Bad"), ("OK", "OK")]
    descriptor = models.CharField(max_length=10, choices=RANKING_TYPE, default=1)
    description = models.TextField(max_length=250)

    def __str__(self):
        return self.descriptor


class CAFFileStore(models.Model):
    descriptor = models.CharField(max_length=100)
    virtual_location = models.CharField(
        max_length=100, help_text="USB, Rosa, email, etc"
    )
    physical_location = models.CharField(
        max_length=100, blank=True, help_text="Cupboard, room, building, etc"
    )  # cupboard, room, building, address
    physical_location_organisation = models.ForeignKey(
        Organisation, on_delete=models.CASCADE
    )

    class Meta:
        verbose_name = "CAF File Store"


class DocumentFile(models.Model):
    FILETYPE_CHOICES = [(1, "Excel"), (2, "Word"), (3, "PDF"), (4, "Hard Copy")]
    name = models.CharField(max_length=255)
    type = models.IntegerField(choices=FILETYPE_CHOICES, default=1)
    file_store_location = models.ForeignKey(CAFFileStore, on_delete=models.CASCADE)


class EssentialService(models.Model):
    def get_sentinel_org():
        """
        We need this so that we can ensure models.SET() is applied with a callable
        to handle when Users are deleted from the system, preventing the Organisation
        objects related to them being deleted also.
        """
        return Organisation.objects.get_or_create(name="DELETED ORGANISATION")[0]

    name = models.CharField(max_length=256)
    description = models.TextField(max_length=1000)
    organisation = models.ForeignKey(
        Organisation, on_delete=models.SET(get_sentinel_org)
    )
    caf = models.ForeignKey("CAF", on_delete=models.CASCADE)

    class Meta:
        verbose_name = "Essential Service"

    def __str__(self):
        return f"{self.organisation.name} | {self.name}"


class CAF(models.Model):
    owner = models.ForeignKey(Organisation, on_delete=models.CASCADE)
    triage_ranking = models.ForeignKey(Ranking, on_delete=models.CASCADE, blank=True, null=True)
    confidence_assessment = models.ForeignKey(ConfidenceAssessment, on_delete=models.CASCADE, blank=True, null=True)
    file = models.ForeignKey(DocumentFile, on_delete=models.CASCADE, blank=True, null=True)

    class Meta:
        verbose_name = "CAF"

    def __str__(self):
        return f"CAF | {self.owner}"