Dovodi_Backend/apps/hadis/models/hadis.py

from django.db import models
from django.db.models import F
from django.utils.translation import gettext_lazy as _
from django.conf import settings


class HadisTag(models.Model):
    title = models.CharField(max_length=355, verbose_name=_('title'))
    status = models.BooleanField(default=True, verbose_name=_('status'))
    created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))
    updated_at = models.DateTimeField(auto_now=True, verbose_name=_('updated at'))
    
    def __str__(self):
        return f"{self.title}"


class HadisStatus(models.Model):
    class ColorChoices(models.TextChoices):
        RED = 'red', _('Red')
        GREEN = 'green', _('Green')
        BLUE = 'blue', _('Blue')
        YELLOW = 'yellow', _('Yellow')
        ORANGE = 'orange', _('Orange')
        PURPLE = 'purple', _('Purple')
        GRAY = 'gray', _('Gray')

    title = models.CharField(max_length=119, verbose_name=_('title'))
    
    color = models.CharField(max_length=20, choices=ColorChoices.choices, verbose_name=_('color'))
    order = models.IntegerField(default=0, verbose_name=_('order'))

    def __str__(self):
        return self.title

    class Meta:
        verbose_name = _('hadis status')
        verbose_name_plural = _('hadis statuses')
        ordering = ('order',)




class Hadis(models.Model):
    category = models.ForeignKey("hadis.HadisCategory", on_delete=models.SET_NULL, null=True, blank=True, verbose_name=_('category'))
    number = models.PositiveIntegerField(verbose_name=_('number'), default=1)
    title = models.CharField(max_length=255, verbose_name=_('title'), null=True, blank=True)
    text = models.TextField(verbose_name=_('text'))
    translation = models.JSONField(verbose_name=_('translation'), default=list)
    status = models.BooleanField(default=True, verbose_name=_('visibility'))

    hadis_status = models.ForeignKey(HadisStatus, on_delete=models.SET_NULL, null=True, blank=True, verbose_name=_('hadis status'))
    hadis_status_text = models.CharField(max_length=255, verbose_name=_('hadis status text'), null=True, blank=True)
    address = models.TextField(verbose_name=_('address'), null=True, blank=True)
    links = models.JSONField(verbose_name=_('links'), null=True, blank=True, default=dict)
    tags = models.ManyToManyField("HadisTag", related_name="hadis_overview", verbose_name=_('tags'), blank=True)

    share_link = models.CharField(max_length=255, verbose_name=_('share link'), null=True, blank=True)
    explanation = models.TextField(verbose_name=_('explanation'), null=True, blank=True)

    created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))
    updated_at = models.DateTimeField(auto_now=True, verbose_name=_('updated at'))

    def __str__(self):
        return f"{self.number} - {self.title}" if self.title else f"Hadis {self.number}"

    def get_translation(self, lang):
        """
        Get translation for a specific language
        """
        if not self.translation or not isinstance(self.translation, list):
            return None

        for tr in self.translation:
            if isinstance(tr, dict) and tr.get('language_code') == lang:
                return tr.get('title', '')

        for tr in self.translation:
            if isinstance(tr, dict) and tr.get('language_code') == 'en':
                return tr.get('title', '')
        return None

    def save(self, *args, **kwargs):
        # ساخت share_link قبل از ذخیره
        if not self.share_link:
            self.share_link = f"{settings.SITE_DOMAIN}/hadis/{self.id}"
        super().save(*args, **kwargs)

    class Meta:
        verbose_name = _('hadis')
        verbose_name_plural = _('hadises')
        ordering = ('category', 'number')



class HadisReference(models.Model):
    hadis = models.ForeignKey(
        Hadis,
        on_delete=models.CASCADE,
        verbose_name=_('hadis'),
        related_name='references'
    )
    book = models.ForeignKey("library.Book", on_delete=models.SET_NULL, null=True, blank=True, verbose_name=_('book'), related_name='hadis_references')
    description = models.TextField(verbose_name=_('description'), blank=True, null=True)
    created_at = models.DateTimeField(auto_now_add=True, verbose_name=_('created at'))

    class Meta:
        verbose_name = _('Hadis Reference')
        verbose_name_plural = _('Hadis References')
        unique_together = ('hadis', 'book')

    def __str__(self):
        return f'{self.hadis.number}-{self.book.title if self.book else "No Book"}'

class ReferenceImage(models.Model):
    reference = models.ForeignKey(HadisReference, verbose_name="Hadis Reference", on_delete=models.CASCADE)
    thumbnail = models.ImageField(upload_to='hadis/reference_images/', null=True, blank=True, verbose_name=_('thumbnail'))
    priority = models.IntegerField(
        default=0, 
        verbose_name=_("Priority"),
        help_text=_("Priority of the image, lower values mean higher priority.")
    )


    class Meta:
        verbose_name = _('Reference Image')
        verbose_name_plural = _('Reference Images')

    def __str__(self):
        return f'{self.reference.title}-{self.id}'
    
    def save(self, *args, **kwargs):
        if ReferenceImage.objects.filter(reference=self.reference, priority=self.priority).exists():
            ReferenceImage.objects.filter(
                reference=self.reference,
                priority__gte=self.priority
            ).update(priority=F('priority') + 1)

        super().save(*args, **kwargs)