RPGnotes/locations/models.py

from django.db import models
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from tree_queries.fields import TreeNodeForeignKey
from tree_queries.models import TreeNode

from common.models import NameSlugModel, DescriptionModel, HistoryModel


class Location(TreeNode, NameSlugModel, DescriptionModel, HistoryModel):
    parent = TreeNodeForeignKey(
        "self",
        blank=True,
        null=True,
        on_delete=models.CASCADE,
        verbose_name=_("Part of"),
        related_name="children",
    )

    class Meta:
        ordering = ["name"]
        verbose_name = _("Location")
        verbose_name_plural = _("Locations")

    def get_absolute_url(self):
        return reverse('locationdetail', args=[self.slug])