You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.5 KiB
105 lines
3.5 KiB
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.core.exceptions import ValidationError
|
|
from dj_category.models import BaseCategoryAbstract
|
|
|
|
|
|
class HadisCategory(BaseCategoryAbstract):
|
|
class SourceType(models.TextChoices):
|
|
SHIA = 'shia', _('Shia')
|
|
SUNNI = 'sunni', _('Sunni')
|
|
|
|
class ContentType(models.TextChoices):
|
|
QURAN = 'quran', _('Quran')
|
|
HADITH = 'hadith', _('Hadith')
|
|
|
|
class LevelChoices(models.IntegerChoices):
|
|
LEVEL_1 = 1, _('Level 1 (Root)')
|
|
LEVEL_2 = 2, _('Level 2 (Child)')
|
|
LEVEL_3 = 3, _('Level 3 (Grandchild)')
|
|
|
|
source_type = models.CharField(max_length=10, choices=SourceType.choices, default=SourceType.SHIA, verbose_name=_('Source Type'), blank=True)
|
|
category_type = models.CharField(max_length=10, choices=ContentType.choices, verbose_name=_('Category Content Type'), blank=True, null=True)
|
|
name = models.CharField(max_length=355, verbose_name=_('name'))
|
|
order = models.IntegerField(default=0, verbose_name=_('order'))
|
|
slug = None
|
|
content_type = None
|
|
language = None
|
|
language_id = None
|
|
|
|
# This field is not stored in the database, it's only used for the form
|
|
level_choice = None
|
|
|
|
class Meta:
|
|
verbose_name = _('Hadis Category')
|
|
verbose_name_plural = _('Hadis Categories')
|
|
ordering = ('order',)
|
|
|
|
def __str__(self):
|
|
return f'<{str(self.level_p)}>{self.name}'
|
|
|
|
def __repr__(self):
|
|
return f'<{str(self.level_p)}>{self.name}'
|
|
|
|
def clean(self):
|
|
super().clean()
|
|
|
|
# Skip validation for new objects that haven't been saved yet
|
|
# This allows the admin form to set these values properly
|
|
if self.pk is None:
|
|
return
|
|
|
|
# For existing objects, apply the validation rules
|
|
if self.level_p == 1 and self.category_type:
|
|
raise ValidationError(_("Level 1 cannot have content type"))
|
|
|
|
if self.level_p == 2 and not self.category_type:
|
|
raise ValidationError(_("Level 2 must have content type"))
|
|
|
|
if self.level_p == 3 and (self.source_type or self.category_type):
|
|
raise ValidationError(_("Level 3 cannot have source/content type"))
|
|
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.clean()
|
|
|
|
# Get the level from the parent structure
|
|
level = self.level_p
|
|
|
|
# Apply level-specific logic
|
|
# if level == 2 and self.parent:
|
|
# For level 2, inherit source_type from parent
|
|
# self.source_type = self.parent.source_type
|
|
# elif level == 3:
|
|
# For level 3, inherit both from parent
|
|
# if self.parent and self.parent.parent:
|
|
# self.source_type = self.parent.source_type
|
|
# self.category_type = self.parent.category_type
|
|
|
|
# Call the parent class's save method
|
|
super().save(*args, **kwargs)
|
|
|
|
@property
|
|
def level_p(self):
|
|
if not self.parent:
|
|
return 1
|
|
elif not self.parent.parent:
|
|
return 2
|
|
else:
|
|
return 3
|
|
def get_level_info(self):
|
|
info = {
|
|
'level': self.level_p,
|
|
'source_type': None,
|
|
'category_type': None,
|
|
}
|
|
if self.level_p == 1:
|
|
info['source_type'] = self.source_type
|
|
elif self.level_p == 2:
|
|
info['source_type'] = self.parent.source_type
|
|
info['category_type'] = self.category_type
|
|
return info
|
|
|
|
|
|
|
|
|