mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
143 lines
4.7 KiB
Python
143 lines
4.7 KiB
Python
import datetime
|
|
import sqlalchemy.exc
|
|
import discord
|
|
from sqlalchemy import create_engine, Column, Integer, String, Boolean, DateTime, ForeignKey
|
|
from sqlalchemy.orm import sessionmaker, relationship
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
import lol
|
|
|
|
|
|
class NoUsersMatchingError(Exception):
|
|
pass
|
|
|
|
|
|
class InvalidPasswordError(Exception):
|
|
pass
|
|
|
|
|
|
# Initialize the database
|
|
engine = create_engine("sqlite:///db.sqlite")
|
|
Base = declarative_base()
|
|
Session = sessionmaker(bind=engine)
|
|
|
|
|
|
class Diario(Base):
|
|
__tablename__ = "diario"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
|
|
text = Column(String, nullable=False)
|
|
date = Column(DateTime, nullable=False)
|
|
|
|
def __repr__(self):
|
|
return f"<Diario {self.date} {self.text}>"
|
|
|
|
|
|
class Account(Base):
|
|
__tablename__ = "account"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
|
|
lol = relationship("LoL")
|
|
|
|
|
|
class LoL(Base):
|
|
__tablename__ = "lol"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
parent_id = Column(Integer, ForeignKey("account.id"))
|
|
|
|
last_updated = Column(DateTime)
|
|
summoner_name = Column(String, nullable=False)
|
|
level = Column(Integer)
|
|
soloq_division = Column(Integer)
|
|
soloq_tier = Column(Integer)
|
|
flexq_division = Column(Integer)
|
|
flexq_tier = Column(Integer)
|
|
ttq_division = Column(Integer)
|
|
ttq_tier = Column(Integer)
|
|
|
|
def __repr__(self):
|
|
return f"<LoL {self.id} {self.summoner_name}>"
|
|
|
|
|
|
async def update_data(self):
|
|
try:
|
|
soloq, flexq, ttq = await lol.get_rank_data("euw", self.id)
|
|
except lol.LoLAPIError as e:
|
|
# LoL returns 404 if the account is unranked
|
|
if e.status_code == 404:
|
|
self.soloq_tier = None
|
|
self.soloq_division = None
|
|
self.flexq_tier = None
|
|
self.flexq_division = None
|
|
self.ttq_tier = None
|
|
self.ttq_division = None
|
|
else:
|
|
# Update the user data
|
|
if soloq is not None:
|
|
self.soloq_tier = lol.tiers.index(soloq["tier"])
|
|
self.soloq_division = lol.divisions.index(soloq["entries"][0]["division"])
|
|
else:
|
|
self.soloq_tier = None
|
|
self.soloq_division = None
|
|
if flexq is not None:
|
|
self.flexq_tier = lol.tiers.index(flexq["tier"])
|
|
self.flexq_division = lol.divisions.index(flexq["entries"][0]["division"])
|
|
else:
|
|
self.flexq_tier = None
|
|
self.flexq_division = None
|
|
if ttq is not None:
|
|
self.ttq_tier = lol.tiers.index(ttq["tier"])
|
|
self.ttq_division = lol.divisions.index(ttq["entries"][0]["division"])
|
|
else:
|
|
self.ttq_tier = None
|
|
self.ttq_division = None
|
|
# Mark the user as updated
|
|
self.last_updated = datetime.datetime.now()
|
|
|
|
|
|
def generate_discord_embed(self):
|
|
embed = discord.Embed(type="rich")
|
|
# TODO: change the icon
|
|
embed.set_author(name="League of Legends", url="http://euw.leagueoflegends.com/", icon_url="https://cdn.discordapp.com/attachments/152150752523976704/307558194824216578/icon.png")
|
|
embed.add_field(name="Summoner", value=str(self.summoner_name))
|
|
embed.add_field(name="Level", value=str(self.level))
|
|
if self.soloq_tier is not None:
|
|
embed.add_field(name="Solo/duo SR", value=f"{lol.tiers[self.soloq_tier].capitalize()} {lol.divisions[self.soloq_division]}", inline=False)
|
|
embed.set_thumbnail(url=f"https://royal.steffo.me/loltiers/{lol.tiers[self.soloq_tier].lower()}_{lol.divisions[self.soloq_division].lower()}.png")
|
|
if self.flexq_tier is not None:
|
|
embed.add_field(name="Flex SR", value=f"{lol.tiers[self.flexq_tier].capitalize()} {lol.divisions[self.flexq_division]}", inline=False)
|
|
if self.ttq_tier is not None:
|
|
embed.add_field(name="Twisted Treeline", value=f"{lol.tiers[self.ttq_tier].capitalize()} {lol.divisions[self.ttq_division]}", inline=False)
|
|
embed.colour = discord.Colour(0x09AEBB)
|
|
return embed
|
|
|
|
|
|
Base.metadata.create_all(engine)
|
|
|
|
|
|
def migrate_diario():
|
|
import datetime
|
|
session = Session()
|
|
file = open("diario.txt", encoding="utf8")
|
|
for row in file:
|
|
entry = row.split("|", 1)
|
|
new = Diario()
|
|
new.date = datetime.datetime.fromtimestamp(int(entry[0]))
|
|
new.text = entry[1]
|
|
session.add(new)
|
|
session.commit()
|
|
|
|
|
|
def new_diario_entry(dt, text):
|
|
# Create a new session
|
|
session = Session()
|
|
# Create a new diario entry
|
|
entry = Diario()
|
|
entry.date = dt
|
|
entry.text = text
|
|
# Add the entry to the database
|
|
session.add(entry)
|
|
# Commit the change
|
|
session.commit()
|