mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
92 lines
2.5 KiB
Python
92 lines
2.5 KiB
Python
from typing import *
|
|
from sqlalchemy import *
|
|
from sqlalchemy.orm import relationship, backref
|
|
from sqlalchemy.ext.declarative import declared_attr
|
|
from ..types import DotaMedal, DotaStars, DotaRank
|
|
import steam.steamid
|
|
|
|
|
|
class Dota:
|
|
__tablename__ = "dota"
|
|
|
|
@declared_attr
|
|
def _steamid(self):
|
|
return Column(BigInteger, ForeignKey("steam._steamid"), primary_key=True)
|
|
|
|
@declared_attr
|
|
def steam(self):
|
|
return relationship("Steam", backref=backref("dota", uselist=False))
|
|
|
|
@property
|
|
def steamid(self):
|
|
return steam.steamid.SteamID(self._steamid)
|
|
|
|
@declared_attr
|
|
def _rank_tier(self):
|
|
return Column(Integer)
|
|
|
|
@property
|
|
def medal(self) -> Optional[DotaMedal]:
|
|
if self._rank_tier is None:
|
|
return None
|
|
return DotaMedal(self._rank_tier // 10)
|
|
|
|
@medal.setter
|
|
def medal(self, value: DotaMedal):
|
|
if not isinstance(value, DotaMedal):
|
|
raise AttributeError("medal can only be set to DotaMedal objects.")
|
|
self._rank_tier = value.value * 10 + self.stars.value
|
|
|
|
@property
|
|
def stars(self) -> Optional[DotaStars]:
|
|
if self._rank_tier is None:
|
|
return None
|
|
return DotaStars(self._rank_tier % 10)
|
|
|
|
@stars.setter
|
|
def stars(self, value: DotaStars):
|
|
if not isinstance(value, DotaStars):
|
|
raise AttributeError("stars can only be set to DotaStars objects.")
|
|
self._rank_tier = self.medal.value * 10 + value.value
|
|
|
|
@property
|
|
def rank(self) -> Optional[DotaRank]:
|
|
if self._rank_tier is None:
|
|
return None
|
|
return DotaRank(self.medal, self.stars)
|
|
|
|
@rank.setter
|
|
def rank(self, value: Optional[DotaRank]):
|
|
if value is None:
|
|
self._rank_tier = None
|
|
return
|
|
if not isinstance(value, DotaRank):
|
|
raise AttributeError("rank can only be set to DotaRank objects (or None).")
|
|
self._rank_tier = value.rank_tier
|
|
|
|
@declared_attr
|
|
def wins(self):
|
|
return Column(Integer)
|
|
|
|
@declared_attr
|
|
def losses(self):
|
|
return Column(Integer)
|
|
|
|
def json(self):
|
|
rank = self.rank
|
|
|
|
return {
|
|
"rank": {
|
|
"raw": self._rank_tier,
|
|
"medal": rank.medal.name,
|
|
"rank": rank.stars.name
|
|
} if self._rank_tier is not None else None,
|
|
"wins": self.wins,
|
|
"losses": self.losses
|
|
}
|
|
|
|
def __repr__(self):
|
|
return f"<Dota account {self._steamid}>"
|
|
|
|
def __str__(self):
|
|
return f"[c]dota:{self._steamid}[/c]"
|