92 lines
3.7 KiB
Python
92 lines
3.7 KiB
Python
|
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
|
||
|
|
from sqlalchemy.orm import relationship
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
'Ancestry',
|
||
|
|
'AncestryTrait',
|
||
|
|
'AncestryTraitMap',
|
||
|
|
'CharacterClassMap',
|
||
|
|
'CharacterClassAttributeMap',
|
||
|
|
'Character',
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class AncestryTraitMap(BaseObject):
|
||
|
|
__tablename__ = "trait_map"
|
||
|
|
ancestry_id = Column(Integer, ForeignKey("ancestry.id"), primary_key=True)
|
||
|
|
ancestry_trait_id = Column(Integer, ForeignKey("ancestry_trait.id"), primary_key=True)
|
||
|
|
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)
|
||
|
|
|
||
|
|
|
||
|
|
class CharacterClassMap(BaseObject):
|
||
|
|
__tablename__ = "class_map"
|
||
|
|
character_id = Column(Integer, ForeignKey("character.id"), primary_key=True)
|
||
|
|
character_class_id = Column(Integer, ForeignKey("character_class.id"), primary_key=True)
|
||
|
|
character_class = relationship("CharacterClass", lazy='immediate')
|
||
|
|
level = Column(Integer, nullable=False, info={'min': 1, 'max': 20}, default=1)
|
||
|
|
|
||
|
|
|
||
|
|
class CharacterClassAttributeMap(BaseObject):
|
||
|
|
__tablename__ = "character_class_attribute_map"
|
||
|
|
class_attribute_id = Column(Integer, ForeignKey("class_attribute.id"), primary_key=True)
|
||
|
|
character_id = Column(Integer, ForeignKey("character.id"), primary_key=True)
|
||
|
|
attribute = relationship("ClassAttribute", lazy='immediate')
|
||
|
|
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
classes = relationship("CharacterClassMap")
|
||
|
|
attributes = relationship("CharacterClassAttributeMap")
|
||
|
|
|
||
|
|
ancestry_id = Column(Integer, ForeignKey("ancestry.id"), nullable=False, default='1')
|
||
|
|
ancestry = relationship("Ancestry", uselist=False)
|