modeling many-to-many relationships

ttfrog/db/schema/__init__.py

@@ -0,0 +1,4 @@
+from .character import *
+from .classes import *
+from .property import *
+from .transaction import *

ttfrog/db/schema/character.py

@@ -0,0 +1,91 @@
+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)

ttfrog/db/schema/classes.py

@@ -0,0 +1,43 @@
+from ttfrog.db.base import Bases, BaseObject, IterableMixin, SavingThrowsMixin, SkillsMixin
+from ttfrog.db.base import StatsEnum
+
+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__ = [
+    'ClassAttributeMap',
+    'ClassAttribute',
+    'CharacterClass',
+]
+
+
+class ClassAttributeMap(BaseObject):
+    __tablename__ = "class_attribute_map"
+    class_attribute_id = Column(Integer, ForeignKey("class_attribute.id"), primary_key=True)
+    character_class_id = Column(Integer, ForeignKey("character_class.id"), primary_key=True)
+    attribute = relationship("ClassAttribute", lazy='immediate')
+    level = Column(Integer, nullable=False, info={'min': 1, 'max': 20}, default=1)
+
+
+class ClassAttribute(BaseObject, IterableMixin):
+    __tablename__ = "class_attribute"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    name = Column(String, nullable=False)
+    value = Column(String, nullable=False)
+    description = Column(Text)
+
+
+class CharacterClass(*Bases, SavingThrowsMixin, SkillsMixin):
+    __tablename__ = "character_class"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    name = Column(String, index=True, unique=True)
+    hit_dice = Column(String, default='1d6')
+    hit_dice_stat = Column(Enum(StatsEnum))
+    proficiencies = Column(String)
+    attributes = relationship("ClassAttributeMap")

ttfrog/db/schema/property.py

@@ -0,0 +1,46 @@
+from ttfrog.db.base import Bases, BaseObject, IterableMixin
+
+from sqlalchemy import Column
+from sqlalchemy import Integer
+from sqlalchemy import String
+from sqlalchemy import Text
+from sqlalchemy import UniqueConstraint
+
+
+__all__ = [
+    'Skill',
+    'Proficiency',
+    'Modifier',
+]
+
+
+class Skill(*Bases):
+    __tablename__ = "skill"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    name = Column(String, index=True, unique=True)
+    description = Column(Text)
+
+    def __repr__(self):
+        return str(self.name)
+
+
+class Proficiency(*Bases):
+    __tablename__ = "proficiency"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    name = Column(String, index=True, unique=True)
+
+    def __repr__(self):
+        return str(self.name)
+
+
+class Modifier(BaseObject, IterableMixin):
+    __tablename__ = "modifier"
+    __table_args__ = (
+        UniqueConstraint('source_table_name', 'source_table_id', 'value', 'type', 'target'),
+    )
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    source_table_name = Column(String, index=True, nullable=False)
+    source_table_id = Column(Integer, index=True, nullable=False)
+    value = Column(String, nullable=False)
+    type = Column(String, nullable=False)
+    target = Column(String, nullable=False)

ttfrog/db/schema/transaction.py

@@ -0,0 +1,14 @@
+from ttfrog.db.base import BaseObject, IterableMixin
+from sqlalchemy import Column
+from sqlalchemy import Integer
+from sqlalchemy import String
+from sqlalchemy import Text
+
+__all__ = ['TransactionLog']
+
+class TransactionLog(BaseObject, IterableMixin):
+    __tablename__ = "transaction_log"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    source_table_name = Column(String, index=True, nullable=False)
+    primary_key = Column(Integer, index=True)
+    diff = Column(Text)