from django.db.models.signals import pre_save
from django.dispatch import receiver

from core.utils import naturalize
from .models import Album, Artist, Game, Song


@receiver(pre_save, sender=Album)
@receiver(pre_save, sender=Artist)
@receiver(pre_save, sender=Game)
@receiver(pre_save, sender=Song)
def update_sorted_fields(sender, instance, **kwargs):
    if sender == Artist:
        instance.sorted_full_name = naturalize(getattr(instance, 'full_name'))
    else:
        instance.sorted_title = naturalize(getattr(instance, 'title'))