from django.db import models

# Create your models here.

# Create your models here.


class IdolManager(models.Manager):
    def comma_to_qs(self, idols_str):
        final_ids = []
        if idols_str:
            for idol in idols_str.split(','):
                obj, created = self.get_or_create(romanized_name=idol.strip())
                final_ids.append(obj.id)
            qs = self.get_queryset().filter(id__in=final_ids).distinct()
            return qs
        return self.none()


class Idol(models.Model):
    branch = models.ForeignKey("categories.Branch", on_delete=models.PROTECT)
    name = models.CharField(max_length=255, blank=True)
    romanized_name = models.CharField(max_length=255)
    voice_actor_name = models.CharField(max_length=255, blank=True)
    romanized_voice_actor_name = models.CharField(max_length=255, blank=True)

    objects = IdolManager()

    def __str__(self):
        return self.romanized_name+" (CV: "+self.romanized_voice_actor_name+")"