mirror of
https://github.com/Llloooggg/TextSouls.git
synced 2026-03-06 20:46:23 +03:00
232 lines
7.5 KiB
Python
232 lines
7.5 KiB
Python
import datetime
|
|
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from sqlalchemy_serializer import SerializerMixin
|
|
|
|
|
|
db = SQLAlchemy()
|
|
|
|
|
|
class User(db.Model, SerializerMixin):
|
|
|
|
__tablename__ = "users"
|
|
|
|
serialize_rules = ("-character",)
|
|
|
|
id = db.Column(db.BigInteger, primary_key=True, autoincrement=False)
|
|
chat_id = db.Column(db.BigInteger, nullable=False, unique=True)
|
|
first_name = db.Column(db.String(255), nullable=True)
|
|
last_name = db.Column(db.String(255), nullable=True)
|
|
username = db.Column(db.String(255), nullable=False)
|
|
registered_on = db.Column(
|
|
db.DateTime, nullable=False, default=datetime.datetime.now()
|
|
)
|
|
is_admin = db.Column(db.Boolean, nullable=False, default=False)
|
|
characters = db.relationship("Character", backref="user", lazy="dynamic")
|
|
|
|
def __str__(self):
|
|
return f"{self.id}: {self.username}"
|
|
|
|
|
|
class CharacterRace(db.Model, SerializerMixin):
|
|
|
|
__tablename__ = "character_races"
|
|
|
|
serialize_rules = ("-characters",)
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(255), nullable=True, unique=True)
|
|
characters = db.relationship("Character", backref="race", lazy="dynamic")
|
|
endurance_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
strength_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
agility_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
defence_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
wisdom_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class CharacterClass(db.Model, SerializerMixin):
|
|
|
|
__tablename__ = "character_classes"
|
|
|
|
serialize_rules = ("-characters",)
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(255), nullable=True, unique=True)
|
|
characters = db.relationship("Character", backref="class", lazy="dynamic")
|
|
endurance_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
strength_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
agility_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
defence_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
wisdom_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Character(db.Model, SerializerMixin):
|
|
|
|
__tablename__ = "characters"
|
|
|
|
serialize_rules = ("-user", "-race", "-class")
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
owner = db.Column(
|
|
db.BigInteger,
|
|
db.ForeignKey("users.id", ondelete="CASCADE"),
|
|
)
|
|
name = db.Column(db.String(255), nullable=False, unique=True)
|
|
character_race = db.Column(
|
|
db.Integer,
|
|
db.ForeignKey("character_races.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
)
|
|
character_class = db.Column(
|
|
db.Integer,
|
|
db.ForeignKey("character_classes.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
)
|
|
created_on = db.Column(
|
|
db.DateTime, nullable=False, default=datetime.datetime.now()
|
|
)
|
|
endurance_base = db.Column(db.Integer(), nullable=False, unique=False)
|
|
strength_base = db.Column(db.Integer(), nullable=False, unique=False)
|
|
agility_base = db.Column(db.Integer(), nullable=False, unique=False)
|
|
defence_base = db.Column(db.Integer(), nullable=False, unique=False)
|
|
wisdom_base = db.Column(db.Integer(), nullable=False, unique=False)
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
|
|
char_race = CharacterRace.query.get(self.character_race)
|
|
char_class = CharacterClass.query.get(self.character_class)
|
|
|
|
self.endurance_base = (
|
|
100 * char_race.endurance_koef * char_class.endurance_koef
|
|
)
|
|
self.strength_base = (
|
|
100 * char_race.strength_koef * char_class.strength_koef
|
|
)
|
|
self.agility_base = (
|
|
100 * char_race.agility_koef * char_class.agility_koef
|
|
)
|
|
self.defence_base = (
|
|
100 * char_race.defence_koef * char_class.defence_koef
|
|
)
|
|
self.wisdom_base = 100 * char_race.wisdom_koef * char_class.wisdom_koef
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
@property
|
|
def battle_stats(self):
|
|
char_race = CharacterRace.query.get(self.character_race)
|
|
char_class = CharacterClass.query.get(self.character_class)
|
|
|
|
attack_power = (
|
|
self.strength_base
|
|
* char_race.strength_koef
|
|
* char_class.strength_koef
|
|
)
|
|
|
|
defence_chance = (
|
|
self.defence_base
|
|
* char_race.defence_koef
|
|
* char_class.defence_koef
|
|
)
|
|
|
|
dodge_chance = (
|
|
self.agility_base
|
|
* char_race.agility_koef
|
|
* char_class.agility_koef
|
|
)
|
|
|
|
return {
|
|
"attack_power": attack_power,
|
|
"defence_chance": defence_chance,
|
|
"dodge_chance": dodge_chance,
|
|
}
|
|
|
|
|
|
class MobsStat(db.Model, SerializerMixin):
|
|
|
|
__tablename__ = "mobs_stat"
|
|
|
|
serialize_rules = ("-mobs_level", "-mobs")
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(255), nullable=False, unique=True)
|
|
mobs = db.relationship("Mobs", backref="class", lazy="dynamic")
|
|
endurance_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
strength_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
agility_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
defence_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
wisdom_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class MobsLevel(db.Model, SerializerMixin):
|
|
|
|
__tablename__ = "mobs_level"
|
|
|
|
serialize_rules = ("-mobs", "-mobs_stat")
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
level = db.Column(db.Integer(), nullable=False, unique=True)
|
|
mobs = db.relationship("Mobs", backref="class", lazy="dynamic")
|
|
level_koef = db.Column(db.Float(), nullable=False, unique=False)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Mobs(db.Model, SerializerMixin):
|
|
|
|
__tablename__ = "mobsLevel"
|
|
|
|
serialize_rules = ("-mobs_stat", "-mobs_level")
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(255), nullable=False, unique=True)
|
|
|
|
mob_stat = db.Column(
|
|
db.Integer,
|
|
db.ForeignKey("mobs_stat.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
)
|
|
mob_level = db.Column(
|
|
db.Integer,
|
|
db.ForeignKey("mobs_level.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
)
|
|
|
|
endurance = db.Column(db.Float(), nullable=False, unique=False)
|
|
strength = db.Column(db.Float(), nullable=False, unique=False)
|
|
agility = db.Column(db.Float(), nullable=False, unique=False)
|
|
defence = db.Column(db.Float(), nullable=False, unique=False)
|
|
wisdom = db.Column(db.Float(), nullable=False, unique=False)
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
|
|
mobs_level = MobsStat.query.get(self.mob_level)
|
|
mobs_stat = MobsLevel.query.get(self.mob_stat)
|
|
|
|
self.endurance_base = (
|
|
100 * mobs_level.level_koef * mobs_stat.endurance_koef
|
|
)
|
|
self.strength_base = (
|
|
100 * mobs_level.level_koef * mobs_stat.strength_koef
|
|
)
|
|
self.agility_base = (
|
|
100 * mobs_level.level_koef * mobs_stat.agility_koef
|
|
)
|
|
self.defence_base = (
|
|
100 * mobs_level.level_koef * mobs_stat.defence_koef
|
|
)
|
|
self.wisdom_base = 100 * mobs_level.level_koef * mobs_stat.wisdom_koef
|