2024-02-18 19:30:41 -08:00
|
|
|
from ttfrog.db.base import Bases, BaseObject, IterableMixin, SavingThrowsMixin, SkillsMixin
|
|
|
|
from ttfrog.db.base import CreatureTypesEnum
|
|
|
|
|
|
|
|
from sqlalchemy import Column
|
|
|
|
from sqlalchemy import Enum
|
|
|
|
from sqlalchemy import Integer
|
|
|
|
from sqlalchemy import String
|
|
|
|
from sqlalchemy import Text
|
|
|
|
from sqlalchemy import ForeignKey
|
2024-02-23 10:45:38 -08:00
|
|
|
from sqlalchemy import UniqueConstraint
|
2024-02-18 19:30:41 -08:00
|
|
|
from sqlalchemy.orm import relationship
|
2024-02-23 10:45:38 -08:00
|
|
|
from sqlalchemy.ext.associationproxy import association_proxy
|
2024-02-18 19:30:41 -08:00
|
|
|
|
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
'Ancestry',
|
|
|
|
'AncestryTrait',
|
|
|
|
'AncestryTraitMap',
|
|
|
|
'CharacterClassMap',
|
|
|
|
'CharacterClassAttributeMap',
|
|
|
|
'Character',
|
|
|
|
]
|
|
|
|
|
2024-02-26 01:12:45 -08:00
|
|
|
|
2024-02-23 10:45:38 -08:00
|
|
|
def class_map_creator(fields):
|
|
|
|
if isinstance(fields, CharacterClassMap):
|
|
|
|
return fields
|
|
|
|
return CharacterClassMap(**fields)
|
|
|
|
|
2024-03-24 16:56:13 -07:00
|
|
|
|
2024-02-26 01:12:45 -08:00
|
|
|
def attr_map_creator(fields):
|
|
|
|
if isinstance(fields, CharacterClassAttributeMap):
|
|
|
|
return fields
|
|
|
|
return CharacterClassAttributeMap(**fields)
|
|
|
|
|
2024-02-18 19:30:41 -08:00
|
|
|
|
|
|
|
class AncestryTraitMap(BaseObject):
|
|
|
|
__tablename__ = "trait_map"
|
2024-03-24 16:56:13 -07:00
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
|
|
ancestry_id = Column(Integer, ForeignKey("ancestry.id"))
|
|
|
|
ancestry_trait_id = Column(Integer, ForeignKey("ancestry_trait.id"))
|
2024-02-18 19:30:41 -08:00
|
|
|
trait = relationship("AncestryTrait", lazy='immediate')
|
|
|
|
level = Column(Integer, nullable=False, info={'min': 1, 'max': 20})
|
|
|
|
|
|
|
|
|
|
|
|
class Ancestry(*Bases):
|
|
|
|
"""
|
|
|
|
A character ancestry ("race"), which has zero or more AncestryTraits.
|
|
|
|
"""
|
|
|
|
__tablename__ = "ancestry"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
|
|
name = Column(String, index=True, unique=True)
|
|
|
|
creature_type = Column(Enum(CreatureTypesEnum))
|
|
|
|
traits = relationship("AncestryTraitMap", lazy='immediate')
|
|
|
|
|
|
|
|
def __repr__(self):
|
|
|
|
return self.name
|
|
|
|
|
|
|
|
|
|
|
|
class AncestryTrait(BaseObject, IterableMixin):
|
|
|
|
"""
|
|
|
|
A trait granted to a character via its Ancestry.
|
|
|
|
"""
|
|
|
|
__tablename__ = "ancestry_trait"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
|
|
name = Column(String, nullable=False)
|
|
|
|
description = Column(Text)
|
|
|
|
|
2024-03-24 16:56:13 -07:00
|
|
|
def __repr__(self):
|
|
|
|
return self.name
|
|
|
|
|
2024-02-18 19:30:41 -08:00
|
|
|
|
2024-02-26 01:12:45 -08:00
|
|
|
class CharacterClassMap(BaseObject, IterableMixin):
|
2024-02-18 19:30:41 -08:00
|
|
|
__tablename__ = "class_map"
|
2024-02-23 10:45:38 -08:00
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
|
|
character_id = Column(Integer, ForeignKey("character.id"))
|
|
|
|
character_class_id = Column(Integer, ForeignKey("character_class.id"))
|
|
|
|
mapping = UniqueConstraint(character_id, character_class_id)
|
2024-03-24 16:56:13 -07:00
|
|
|
level = Column(Integer, nullable=False, info={'min': 1, 'max': 20}, default=1)
|
2024-02-23 10:45:38 -08:00
|
|
|
|
2024-02-18 19:30:41 -08:00
|
|
|
character_class = relationship("CharacterClass", lazy='immediate')
|
2024-03-24 16:56:13 -07:00
|
|
|
character = relationship("Character", uselist=False, viewonly=True)
|
|
|
|
|
|
|
|
def __repr__(self):
|
|
|
|
return f"{self.character.name}, {self.character_class.name}, level {self.level}"
|
2024-02-18 19:30:41 -08:00
|
|
|
|
|
|
|
|
2024-02-26 01:12:45 -08:00
|
|
|
class CharacterClassAttributeMap(BaseObject, IterableMixin):
|
2024-02-18 19:30:41 -08:00
|
|
|
__tablename__ = "character_class_attribute_map"
|
2024-02-26 01:12:45 -08:00
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
|
|
character_id = Column(Integer, ForeignKey("character.id"), nullable=False)
|
|
|
|
class_attribute_id = Column(Integer, ForeignKey("class_attribute.id"), nullable=False)
|
|
|
|
option_id = Column(Integer, ForeignKey("class_attribute_option.id"), nullable=False)
|
|
|
|
mapping = UniqueConstraint(character_id, class_attribute_id)
|
|
|
|
|
|
|
|
class_attribute = relationship("ClassAttribute", lazy='immediate')
|
|
|
|
option = relationship("ClassAttributeOption", lazy='immediate')
|
2024-02-18 19:30:41 -08:00
|
|
|
|
2024-03-24 16:56:13 -07:00
|
|
|
character_class = relationship(
|
|
|
|
"CharacterClass",
|
|
|
|
secondary="class_map",
|
|
|
|
primaryjoin="CharacterClassAttributeMap.character_id == CharacterClassMap.character_id",
|
|
|
|
secondaryjoin="CharacterClass.id == CharacterClassMap.character_class_id",
|
|
|
|
viewonly=True
|
|
|
|
)
|
|
|
|
|
2024-02-18 19:30:41 -08:00
|
|
|
|
|
|
|
class Character(*Bases, SavingThrowsMixin, SkillsMixin):
|
|
|
|
__tablename__ = "character"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
|
|
name = Column(String, default='New Character', nullable=False)
|
|
|
|
armor_class = Column(Integer, default=10, nullable=False, info={'min': 1, 'max': 99})
|
|
|
|
hit_points = Column(Integer, default=1, nullable=False, info={'min': 0, 'max': 999})
|
|
|
|
max_hit_points = Column(Integer, default=1, nullable=False, info={'min': 0, 'max': 999})
|
|
|
|
temp_hit_points = Column(Integer, default=0, nullable=False, info={'min': 0, 'max': 999})
|
|
|
|
speed = Column(Integer, nullable=False, default=30, info={'min': 0, 'max': 99})
|
|
|
|
str = Column(Integer, nullable=False, default=10, info={'min': 0, 'max': 30})
|
|
|
|
dex = Column(Integer, nullable=False, default=10, info={'min': 0, 'max': 30})
|
|
|
|
con = Column(Integer, nullable=False, default=10, info={'min': 0, 'max': 30})
|
|
|
|
int = Column(Integer, nullable=False, default=10, info={'min': 0, 'max': 30})
|
|
|
|
wis = Column(Integer, nullable=False, default=10, info={'min': 0, 'max': 30})
|
|
|
|
cha = Column(Integer, nullable=False, default=10, info={'min': 0, 'max': 30})
|
|
|
|
proficiencies = Column(String)
|
|
|
|
|
2024-02-23 10:45:38 -08:00
|
|
|
class_map = relationship("CharacterClassMap", cascade='all,delete,delete-orphan')
|
|
|
|
classes = association_proxy('class_map', 'id', creator=class_map_creator)
|
|
|
|
|
2024-02-26 01:12:45 -08:00
|
|
|
character_class_attribute_map = relationship("CharacterClassAttributeMap", cascade='all,delete,delete-orphan')
|
|
|
|
class_attributes = association_proxy('character_class_attribute_map', 'id', creator=attr_map_creator)
|
2024-02-18 19:30:41 -08:00
|
|
|
|
|
|
|
ancestry_id = Column(Integer, ForeignKey("ancestry.id"), nullable=False, default='1')
|
|
|
|
ancestry = relationship("Ancestry", uselist=False)
|
2024-03-24 16:56:13 -07:00
|
|
|
|
|
|
|
@property
|
|
|
|
def traits(self):
|
|
|
|
return [mapping.trait for mapping in self.ancestry.traits]
|
|
|
|
|
|
|
|
@property
|
|
|
|
def level(self):
|
|
|
|
return sum(mapping.level for mapping in self.class_map)
|
|
|
|
|
|
|
|
@property
|
|
|
|
def levels(self):
|
|
|
|
return dict([(mapping.character_class.name, mapping.level) for mapping in self.class_map])
|
|
|
|
|
|
|
|
def add_class(self, newclass, level=1):
|
|
|
|
if level == 0:
|
|
|
|
return self.remove_class(newclass)
|
|
|
|
level_in_class = [mapping for mapping in self.class_map if mapping.character_class_id == newclass.id]
|
|
|
|
if level_in_class:
|
|
|
|
level_in_class = level_in_class[0]
|
|
|
|
level_in_class.level = level
|
|
|
|
return
|
|
|
|
self.classes.append(CharacterClassMap(
|
|
|
|
character_id=self.id,
|
|
|
|
character_class_id=newclass.id,
|
|
|
|
level=level
|
|
|
|
))
|
|
|
|
|
|
|
|
def remove_class(self, target):
|
|
|
|
self.class_map = [m for m in self.class_map if m.id != target.id]
|