RPGnotes/loot/models.py

from django.db import models
from django.urls import reverse
from django.utils.translation import gettext_lazy as _

from characters.models import Character
from common.models import DescriptionModel, HistoryModel
from locations.models import Location


class Loot(DescriptionModel, HistoryModel):
    name = models.CharField(_("Name"), max_length=1000)
    quantity = models.PositiveSmallIntegerField(_("Quantity"))
    value_gold = models.DecimalField(_("Value (Gold)"), max_digits=7, decimal_places=2)
    weight = models.FloatField(_("Weight (lb)"), null=True, blank=True)
    owner = models.ForeignKey(
        Character, on_delete=models.PROTECT,
        blank=True, null=True,
        verbose_name=_("Claimant"),
        related_name="loot"
    )
    location = models.ForeignKey(
        Location, on_delete=models.PROTECT,
        blank=True, null=True,
        verbose_name=_("Location"),
        related_name="loot"
    )
    magic_item = models.BooleanField(_("Magic Item"), default=False)

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

    @property
    def value_per_unit(self):
        return self.value_gold / self.quantity

    def __str__(self):
        return self.name

    def get_absolute_url(self):
        return reverse('lootedit', args=[self.id])