royalnet/rpgpack/tables/dndcharacters.py

from sqlalchemy import *
from sqlalchemy.orm import *
from sqlalchemy.ext.declarative import *
from ..utils import DndProficiencyType


class DndCharacter:
    __tablename__ = "dndcharacters"

    @declared_attr
    def character_id(self):
        return Column(Integer, primary_key=True)

    @declared_attr
    def creator_id(self):
        return Column(Integer, ForeignKey("users.uid"))

    @declared_attr
    def creator(self):
        return relationship("User", foreign_keys=self.creator_id, backref="dndcharacters_created")

    @declared_attr
    def name(self):
        return Column(String, nullable=False)

    @declared_attr
    def strength_score(self):
        return Column(Integer, nullable=False)

    @property
    def strength(self):
        return (self.strength_score - 10) // 2

    @declared_attr
    def dexterity_score(self):
        return Column(Integer, nullable=False)

    @property
    def dexterity(self):
        return (self.dexterity_score - 10) // 2

    @declared_attr
    def constitution_score(self):
        return Column(Integer, nullable=False)

    @property
    def constitution(self):
        return (self.constitution_score - 10) // 2

    @declared_attr
    def intelligence_score(self):
        return Column(Integer, nullable=False)

    @property
    def intelligence(self):
        return (self.intelligence_score - 10) // 2

    @declared_attr
    def wisdom_score(self):
        return Column(Integer, nullable=False)

    @property
    def wisdom(self):
        return (self.wisdom_score - 10) // 2

    @declared_attr
    def charisma_score(self):
        return Column(Integer, nullable=False)

    @property
    def charisma(self):
        return (self.charisma_score - 10) // 2

    @declared_attr
    def level(self):
        return Column(Integer, nullable=False)

    @property
    def proficiency_bonus(self):
        return ((self.level - 1) // 4) + 2

    @declared_attr
    def current_hp(self):
        return Column(Integer, nullable=False)

    @declared_attr
    def max_hp(self):
        return Column(Integer, nullable=False)

    @declared_attr
    def armor_class(self):
        return Column(Integer, nullable=False)

    @declared_attr
    def strength_save_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def dexterity_save_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def constitution_save_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def intelligence_save_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def wisdom_save_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def charisma_save_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def acrobatics_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def animal_handling_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def arcana_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def athletics_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def deception_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def history_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def insight_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def intimidation_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def investigation_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def medicine_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def nature_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def perception_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def performance_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def persuasion_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def religion_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def sleight_of_hand_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def stealth_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @declared_attr
    def survival_proficiency(self):
        return Column(Enum(DndProficiencyType), nullable=False, default=DndProficiencyType.NONE)

    @property
    def strength_save(self):
        return self.strength + self.proficiency_bonus * self.strength_save_proficiency.value

    @property
    def dexterity_save(self):
        return self.dexterity + self.proficiency_bonus * self.dexterity_save_proficiency.value

    @property
    def constitution_save(self):
        return self.constitution + self.proficiency_bonus * self.constitution_save_proficiency.value

    @property
    def intelligence_save(self):
        return self.intelligence + self.proficiency_bonus * self.intelligence_save_proficiency.value

    @property
    def wisdom_save(self):
        return self.wisdom + self.proficiency_bonus * self.wisdom_save_proficiency.value

    @property
    def charisma_save(self):
        return self.charisma + self.proficiency_bonus * self.charisma_save_proficiency.value

    @property
    def acrobatics(self):
        return self.dexterity + self.proficiency_bonus * self.acrobatics_proficiency.value

    @property
    def animal_handling(self):
        return self.wisdom + self.proficiency_bonus * self.animal_handling_proficiency.value

    @property
    def arcana(self):
        return self.intelligence + self.proficiency_bonus * self.arcana_proficiency.value

    @property
    def athletics(self):
        return self.strength + self.proficiency_bonus * self.athletics_proficiency.value

    @property
    def deception(self):
        return self.charisma + self.proficiency_bonus * self.deception_proficiency.value

    @property
    def history(self):
        return self.intelligence + self.proficiency_bonus * self.history_proficiency.value

    @property
    def insight(self):
        return self.wisdom + self.proficiency_bonus * self.insight_proficiency.value

    @property
    def intimidation(self):
        return self.charisma + self.proficiency_bonus * self.intimidation_proficiency.value

    @property
    def investigation(self):
        return self.intelligence + self.proficiency_bonus * self.investigation_proficiency.value

    @property
    def medicine(self):
        return self.wisdom + self.proficiency_bonus * self.medicine_proficiency.value

    @property
    def nature(self):
        return self.intelligence + self.proficiency_bonus * self.nature_proficiency.value

    @property
    def perception(self):
        return self.wisdom + self.proficiency_bonus * self.perception_proficiency.value

    @property
    def performance(self):
        return self.charisma + self.proficiency_bonus * self.performance_proficiency.value

    @property
    def persuasion(self):
        return self.charisma + self.proficiency_bonus * self.persuasion_proficiency.value

    @property
    def religion(self):
        return self.intelligence + self.proficiency_bonus * self.religion_proficiency.value

    @property
    def sleight_of_hand(self):
        return self.dexterity + self.proficiency_bonus * self.sleight_of_hand_proficiency.value

    @property
    def stealth(self):
        return self.dexterity + self.proficiency_bonus * self.stealth_proficiency.value

    @property
    def survival(self):
        return self.wisdom + self.proficiency_bonus * self.survival_proficiency.value

    def __repr__(self):
        return f"<{self.__class__.__qualname__} {self.name}>"

    def __str__(self):
        return f"{self.name}"

    def character_sheet(self) -> str:
        columns = list(self.__class__.__table__.columns)
        column_names = [column.name for column in columns if (not column.primary_key and
                                                              not column.foreign_keys and
                                                              column.name != "name")]
        message = f"[b]{self.name}[/b]\n"
        for column_name in column_names:
            value = self.__getattribute__(column_name)
            message += f"{column_name} {value}\n"
        return message