charasheet/src/character/admin.py

130 lines
3.9 KiB
Python

from django.contrib import admin
from character import models
@admin.register(models.Capability)
class CapabilityAdmin(admin.ModelAdmin):
list_display = ["name", "path", "rank", "limited", "spell"]
list_filter = ["path", "path__profile", "path__race", "rank", "limited", "spell"]
search_fields = ["name", "description"]
@admin.register(models.Path)
class PathAdmin(admin.ModelAdmin):
list_display = ["name", "category", "related_to"]
list_filter = ["category"]
search_fields = ["name"]
fieldsets = [
(None, {"fields": ["name"]}),
("Related to", {"fields": ["category", ("profile", "race")]}),
("Notes", {"fields": ["notes"]}),
]
def related_to(self, instance: models.Path) -> str:
category = models.Path.Category(instance.category)
if category == models.Path.Category.PROFILE:
return str(instance.profile)
elif category == models.Path.Category.RACE:
return str(instance.race)
else:
return ""
@admin.register(models.RacialCapability)
class RacialCapabilityAdmin(admin.ModelAdmin):
list_display = ["name", "race"]
list_filter = ["race"]
search_fields = ["name", "description"]
class PathInline(admin.TabularInline):
model = models.Path
fields = ["name"]
extra = 0
@admin.register(models.Profile)
class ProfileAdmin(admin.ModelAdmin):
list_display = ["name", "life_dice", "magical_strength"]
list_filter = ["life_dice", "magical_strength"]
search_fields = ["name"]
inlines = [PathInline]
class RacialCapabilityInline(admin.TabularInline):
model = models.RacialCapability
extra = 0
@admin.register(models.Race)
class RaceAdmin(admin.ModelAdmin):
list_display = ["name"]
search_fields = ["name"]
inlines = [RacialCapabilityInline, PathInline]
@admin.register(models.Character)
class CharacterAdmin(admin.ModelAdmin):
list_display = ["name", "player", "race", "profile", "level"]
list_filter = ["race", "profile"]
search_fields = ["name", "notes"]
fieldsets = [
(
"Identity",
{"fields": ["name", "player", "profile", "level", "race"]},
),
("Appearance", {"fields": ["gender", "age", "height", "weight"]}),
(
"Abilities",
{
"fields": [
("value_strength", "modifier_strength"),
("value_dexterity", "modifier_dexterity"),
("value_constitution", "modifier_constitution"),
("value_intelligence", "modifier_intelligence"),
("value_wisdom", "modifier_wisdom"),
("value_charisma", "modifier_charisma"),
]
},
),
(
"Fight",
{"fields": ["initiative", "attack_melee", "attack_range", "attack_magic"]},
),
("Health", {"fields": ["health_max", "health_remaining"]}),
("Defense", {"fields": ["armor", "shield", "defense_misc", "defense"]}),
("Weapons & equipment", {"fields": ["weapons", "equipment"]}),
("Racial", {"fields": ["racial_capability"]}),
("Capabilities", {"fields": ["capabilities"]}),
("Luck", {"fields": ["luck_points_max", "luck_points_remaining"]}),
("Mana", {"fields": ["mana_max", "mana_consumed", "mana_remaining"]}),
("Notes", {"fields": ["notes"]}),
]
readonly_fields = [
"modifier_strength",
"modifier_dexterity",
"modifier_constitution",
"modifier_intelligence",
"modifier_wisdom",
"modifier_charisma",
"initiative",
"attack_melee",
"attack_range",
"attack_magic",
"defense",
"mana_max",
"mana_remaining",
]
filter_horizontal = [
"capabilities",
"weapons",
]
@admin.register(models.Weapon)
class WeaponAdmin(admin.ModelAdmin):
list_display = ["name", "damage"]
search_fields = ["name", "special", "damage"]