mirror of
https://github.com/Findus23/RPGnotes.git
synced 2024-09-19 15:43:45 +02:00
80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
from django.contrib.postgres.fields import ArrayField
|
|
from django.core.exceptions import ValidationError
|
|
from django.core.validators import MinLengthValidator
|
|
from django.db import models
|
|
from django.urls import reverse
|
|
from django.utils.translation import gettext_lazy as _
|
|
from sorl.thumbnail import ImageField
|
|
|
|
from common.models import NameSlugModel, DescriptionModel, HistoryModel, AliasModel
|
|
from factions.models import Faction
|
|
from locations.models import Location
|
|
from rpg_notes.settings import AUTH_USER_MODEL
|
|
from search.utils import NameSearchIndex
|
|
from utils.colors import get_random_color, is_bright_color
|
|
from utils.random_filename import get_file_path
|
|
|
|
|
|
def validate_color_hex(value: str):
|
|
if not value.startswith("#"):
|
|
raise ValidationError("color hex has to start with a #")
|
|
|
|
|
|
class Character(NameSlugModel, DescriptionModel, AliasModel, HistoryModel):
|
|
aliases = ArrayField(
|
|
models.CharField(_("Nickname"), max_length=100),
|
|
verbose_name=_("Aliases"), blank=True, null=True
|
|
)
|
|
subtitle = models.CharField(_("Subtitle"), max_length=100, blank=True)
|
|
player = models.ForeignKey(
|
|
AUTH_USER_MODEL, on_delete=models.PROTECT, blank=True, null=True,
|
|
related_name="characters", verbose_name=_("Player"),
|
|
help_text=_("If no player is selected, this character is considered an NPC.")
|
|
)
|
|
faction = models.ForeignKey(
|
|
Faction, on_delete=models.PROTECT, blank=True, null=True,
|
|
related_name="characters",
|
|
verbose_name=_("Faction")
|
|
)
|
|
location = models.ForeignKey(
|
|
Location, on_delete=models.PROTECT, blank=True, null=True,
|
|
verbose_name=_("Location")
|
|
)
|
|
archived = models.BooleanField(_("Archived"), default=False)
|
|
color = models.CharField(_("Color"), max_length=7, default=get_random_color, validators=[
|
|
MinLengthValidator(7),
|
|
validate_color_hex
|
|
])
|
|
token_image = ImageField(_("Token Image"), upload_to=get_file_path, blank=True, null=True, help_text=_("round"))
|
|
large_image = ImageField(_("Large Image"), upload_to=get_file_path, blank=True, null=True)
|
|
|
|
class Meta:
|
|
ordering = ["archived", "name"]
|
|
verbose_name = _("Character")
|
|
verbose_name_plural = _("Characters")
|
|
indexes = [
|
|
NameSearchIndex
|
|
]
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('characterdetail', args=[self.slug])
|
|
|
|
@property
|
|
def initials(self):
|
|
return "".join([word[0] for word in self.name.split()][:2]).upper()
|
|
|
|
@property
|
|
def text_color(self):
|
|
return "black" if is_bright_color(self.color) else "white"
|
|
|
|
@property
|
|
def larger_image(self):
|
|
if self.large_image:
|
|
return self.large_image
|
|
return self.token_image
|
|
|
|
@property
|
|
def smaller_image(self):
|
|
if self.token_image:
|
|
return self.token_image
|
|
return self.larger_image
|