mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
170 lines
8.2 KiB
Python
170 lines
8.2 KiB
Python
|
from typing import *
|
|||
|
|
|
|||
|
|
import asyncio
|
|||
|
|
import logging
|
|||
|
|
import aiohttp
|
|||
|
|
|
|||
|
|
from royalnet.backpack import tables as rbt
|
|||
|
|
import royalnet.commands as rc
|
|||
|
|
import royalnet.utils as ru
|
|||
|
|
from sqlalchemy import or_, and_
|
|||
|
|
|
|||
|
|
from .abstract.linker import LinkerCommand
|
|||
|
|
from ..tables import Steam, Brawlhalla, BrawlhallaDuo
|
|||
|
|
from ..types import BrawlhallaRank, BrawlhallaMetal, BrawlhallaTier, Updatable
|
|||
|
|
|
|||
|
|
log = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class BrawlhallaCommand(LinkerCommand):
|
|||
|
|
name: str = "brawlhalla"
|
|||
|
|
|
|||
|
|
aliases = ["bh", "bruhalla", "bruhlalla"]
|
|||
|
|
|
|||
|
|
description: str = "Visualizza le tue statistiche di Brawlhalla."
|
|||
|
|
|
|||
|
|
syntax: str = ""
|
|||
|
|
|
|||
|
|
def token(self):
|
|||
|
|
return self.config['brawlhalla']['token']
|
|||
|
|
|
|||
|
|
async def get_updatables_of_user(self, session, user: rbt.User) -> List[Brawlhalla]:
|
|||
|
|
return user.steam
|
|||
|
|
|
|||
|
|
async def get_updatables(self, session) -> List[Brawlhalla]:
|
|||
|
|
return await ru.asyncify(session.query(self.alchemy.get(Steam)).all)
|
|||
|
|
|
|||
|
|
async def create(self,
|
|||
|
|
session,
|
|||
|
|
user: rbt.User,
|
|||
|
|
args: rc.CommandArgs,
|
|||
|
|
data: Optional[rc.CommandData] = None) -> Optional[Brawlhalla]:
|
|||
|
|
raise rc.InvalidInputError("Brawlhalla accounts are automatically linked from Steam.")
|
|||
|
|
|
|||
|
|
async def update(self, session, obj, change: Callable[[str, Any], Awaitable[None]]):
|
|||
|
|
BrawlhallaT = self.alchemy.get(Brawlhalla)
|
|||
|
|
DuoT = self.alchemy.get(BrawlhallaDuo)
|
|||
|
|
log.info(f"Updating: {obj}")
|
|||
|
|
async with aiohttp.ClientSession() as hcs:
|
|||
|
|
bh: Brawlhalla = obj.brawlhalla
|
|||
|
|
if bh is None:
|
|||
|
|
log.debug(f"Checking if player has an account...")
|
|||
|
|
async with hcs.get(f"https://api.brawlhalla.com/search?steamid={obj.steamid.as_64}&api_key={self.token()}") as response:
|
|||
|
|
if response.status != 200:
|
|||
|
|
raise rc.ExternalError(f"Brawlhalla API /search returned {response.status}!")
|
|||
|
|
j = await response.json()
|
|||
|
|
if j == {} or j == []:
|
|||
|
|
log.debug("No account found.")
|
|||
|
|
return
|
|||
|
|
bh = BrawlhallaT(
|
|||
|
|
steam=obj,
|
|||
|
|
brawlhalla_id=j["brawlhalla_id"],
|
|||
|
|
name=j["name"]
|
|||
|
|
)
|
|||
|
|
session.add(bh)
|
|||
|
|
session.flush()
|
|||
|
|
|
|||
|
|
async with hcs.get(f"https://api.brawlhalla.com/player/{bh.brawlhalla_id}/ranked?api_key={self.token()}") as response:
|
|||
|
|
if response.status != 200:
|
|||
|
|
raise rc.ExternalError(f"Brawlhalla API /ranked returned {response.status}!")
|
|||
|
|
j = await response.json()
|
|||
|
|
if j == {} or j == []:
|
|||
|
|
log.debug("No ranked info found.")
|
|||
|
|
else:
|
|||
|
|
await self._change(session=session, obj=bh, attribute="rating_1v1", new=j["rating"])
|
|||
|
|
metal_name, tier_name = j["tier"].split(" ", 1)
|
|||
|
|
metal = BrawlhallaMetal[metal_name.upper()]
|
|||
|
|
tier = BrawlhallaTier(int(tier_name))
|
|||
|
|
rank = BrawlhallaRank(metal=metal, tier=tier)
|
|||
|
|
await self._change(session=session, obj=bh, attribute="rank_1v1", new=rank)
|
|||
|
|
|
|||
|
|
for jduo in j.get("2v2", []):
|
|||
|
|
bhduo: Optional[BrawlhallaDuo] = await ru.asyncify(
|
|||
|
|
session.query(DuoT)
|
|||
|
|
.filter(
|
|||
|
|
or_(
|
|||
|
|
and_(
|
|||
|
|
DuoT.id_one == jduo["brawlhalla_id_one"],
|
|||
|
|
DuoT.id_two == jduo["brawlhalla_id_two"]
|
|||
|
|
),
|
|||
|
|
and_(
|
|||
|
|
DuoT.id_one == jduo["brawlhalla_id_two"],
|
|||
|
|
DuoT.id_two == jduo["brawlhalla_id_one"]
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
.one_or_none
|
|||
|
|
)
|
|||
|
|
if bhduo is None:
|
|||
|
|
if bh.brawlhalla_id == jduo["brawlhalla_id_one"]:
|
|||
|
|
otherbh: Optional[Brawlhalla] = await ru.asyncify(
|
|||
|
|
session.query(BrawlhallaT).get, jduo["brawlhalla_id_two"]
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
otherbh: Optional[Brawlhalla] = await ru.asyncify(
|
|||
|
|
session.query(BrawlhallaT).get, jduo["brawlhalla_id_one"]
|
|||
|
|
)
|
|||
|
|
if otherbh is None:
|
|||
|
|
continue
|
|||
|
|
bhduo = DuoT(
|
|||
|
|
one=bh,
|
|||
|
|
two=otherbh,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
session.add(bhduo)
|
|||
|
|
await self._change(session=session, obj=bhduo, attribute="rating_2v2", new=jduo["rating"])
|
|||
|
|
metal_name, tier_name = jduo["tier"].split(" ", 1)
|
|||
|
|
metal = BrawlhallaMetal[metal_name.upper()]
|
|||
|
|
tier = BrawlhallaTier(int(tier_name))
|
|||
|
|
rank = BrawlhallaRank(metal=metal, tier=tier)
|
|||
|
|
await self._change(session=session, obj=bhduo, attribute="rank_2v2", new=rank)
|
|||
|
|
|
|||
|
|
async def on_increase(self, session, obj: Union[Brawlhalla, BrawlhallaDuo], attribute: str, old: Any, new: Any) -> None:
|
|||
|
|
if attribute == "rank_1v1":
|
|||
|
|
await self.notify(f"📈 [b]{obj.steam.user}[/b] è salito a [b]{new}[/b] ({obj.rating_1v1} MMR) in 1v1 su Brawlhalla! Congratulazioni!")
|
|||
|
|
elif attribute == "rank_2v2":
|
|||
|
|
await self.notify(f"📈 [b]{obj.one.steam.user}[/b] e [b]{obj.two.steam.user}[/b] sono saliti a [b]{new}[/b] ({obj.rating_2v2} MMR) in 2v2 su Brawlhalla! Congratulazioni!")
|
|||
|
|
|
|||
|
|
async def on_unchanged(self, session, obj: Union[Brawlhalla, BrawlhallaDuo], attribute: str, old: Any, new: Any) -> None:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
async def on_decrease(self, session, obj: Union[Brawlhalla, BrawlhallaDuo], attribute: str, old: Any, new: Any) -> None:
|
|||
|
|
if attribute == "rank_1v1":
|
|||
|
|
await self.notify(f"📉 [b]{obj.steam.user}[/b] è sceso a [b]{new}[/b] ({obj.rating_1v1} MMR) in 1v1 su Brawlhalla.")
|
|||
|
|
elif attribute == "rank_2v2":
|
|||
|
|
await self.notify(f"📉 [b]{obj.one.steam.user}[/b] e [b]{obj.two.steam.user}[/b] sono scesi a [b]{new}[/b] ({obj.rating_2v2} MMR) in 2v2 su Brawlhalla.")
|
|||
|
|
|
|||
|
|
async def on_first(self, session, obj: Union[Brawlhalla, BrawlhallaDuo], attribute: str, old: None, new: Any) -> None:
|
|||
|
|
if attribute == "rank_1v1":
|
|||
|
|
await self.notify(f"🌟 [b]{obj.steam.user}[/b] si è classificato a [b]{new}[/b] ({obj.rating_1v1} MMR) in 1v1 su Brawlhalla!")
|
|||
|
|
elif attribute == "rank_2v2":
|
|||
|
|
await self.notify(f"🌟 [b]{obj.one.steam.user}[/b] e [b]{obj.two.steam.user}[/b] si sono classificati a [b]{new}[/b] ({obj.rating_2v2} MMR) in 2v2 su Brawlhalla!")
|
|||
|
|
|
|||
|
|
async def on_reset(self, session, obj: Union[Brawlhalla, BrawlhallaDuo], attribute: str, old: Any, new: None) -> None:
|
|||
|
|
if attribute == "rank_1v1":
|
|||
|
|
await self.notify(f"⬜️ [b]{obj.steam.user}[/b] non ha più un rank su Brawlhalla.")
|
|||
|
|
elif attribute == "rank_2v2":
|
|||
|
|
await self.notify(f"⬜️ [b]{obj.one.steam.user}[/b] e [b]{obj.two.steam.user}[/b] non hanno più un rank su Brawlhalla.")
|
|||
|
|
|
|||
|
|
def describe(self, obj: Steam) -> str:
|
|||
|
|
bh = obj.brawlhalla
|
|||
|
|
|
|||
|
|
string = [f"ℹ️ [b]{bh.name}[/b]", ""]
|
|||
|
|
|
|||
|
|
if bh.rank_1v1:
|
|||
|
|
string.append("👤 [b]1v1[/b]")
|
|||
|
|
string.append(f"[b]{bh.rank_1v1}[/b] ({bh.rating_1v1} MMR)")
|
|||
|
|
string.append("")
|
|||
|
|
|
|||
|
|
if len(bh.duos) != 0:
|
|||
|
|
string.append(f"👥 [b]2v2[/b]")
|
|||
|
|
|
|||
|
|
for duo in sorted(bh.duos, key=lambda d: -d.rating_2v2):
|
|||
|
|
other = duo.other(bh)
|
|||
|
|
string.append(f"Con [b]{other.steam.user}[/b]: [b]{duo.rank_2v2}[/b] ({duo.rating_2v2} MMR)")
|
|||
|
|
|
|||
|
|
if len(bh.duos) != 0:
|
|||
|
|
string.append("")
|
|||
|
|
|
|||
|
|
return "\n".join(string)
|