mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
267 lines
12 KiB
Python
267 lines
12 KiB
Python
import asyncio
|
|
import discord
|
|
import json
|
|
import overwatch
|
|
import league
|
|
import strings as s
|
|
import telegram
|
|
import bs4
|
|
import brawlhalla
|
|
|
|
loop = asyncio.get_event_loop()
|
|
d_client = discord.Client()
|
|
discord_is_ready = False
|
|
|
|
|
|
# When Discord is ready, set discord_is_ready to True
|
|
@d_client.event
|
|
async def on_ready():
|
|
global discord_is_ready
|
|
discord_is_ready = True
|
|
|
|
# Get player database from the db.json file
|
|
file = open("db.json")
|
|
db = json.load(file)
|
|
file.close()
|
|
|
|
# Get the discord bot token from "discordtoken.txt"
|
|
file = open("discordtoken.txt", "r")
|
|
token = file.read()
|
|
file.close()
|
|
|
|
# Every timeout seconds, update player status and check for levelups
|
|
async def overwatch_status_change(timeout):
|
|
while True:
|
|
if discord_is_ready:
|
|
print("[Overwatch] Starting check...")
|
|
# Update data for every player in list
|
|
for player in db:
|
|
if "overwatch" in db[player]:
|
|
try:
|
|
r = await overwatch.get_player_data(**db[player]["overwatch"])
|
|
except overwatch.NotFoundException:
|
|
print("[Overwatch] Player not found.")
|
|
except Exception:
|
|
# If some other error occours, skip the player
|
|
print("[Overwatch] Request returned an unhandled exception.")
|
|
else:
|
|
# Check for levelups
|
|
level = r["data"]["level"]
|
|
try:
|
|
oldlevel = db[player]["overwatch"]["level"]
|
|
except KeyError:
|
|
oldlevel = 0
|
|
if level > oldlevel:
|
|
# Send the message
|
|
loop.create_task(send_event(eventmsg=s.overwatch_level_up, player=player, level=level))
|
|
# Update database
|
|
db[player]["overwatch"]["level"] = level
|
|
f = open("db.json", "w")
|
|
json.dump(db, f)
|
|
f.close()
|
|
# Check for rank changes
|
|
rank = r["data"]["competitive"]["rank"]
|
|
if rank is not None:
|
|
rank = int(rank)
|
|
try:
|
|
oldrank = int(db[player]["overwatch"]["rank"])
|
|
except KeyError:
|
|
oldrank = 0
|
|
if rank != oldrank:
|
|
# Send the message
|
|
loop.create_task(send_event(eventmsg=s.overwatch_rank_change, player=player, change=oldrank-rank, rank=rank, medal=overwatch.rank_to_medal(rank)))
|
|
# Update database
|
|
db[player]["overwatch"]["rank"] = rank
|
|
f = open("db.json", "w")
|
|
json.dump(db, f)
|
|
f.close()
|
|
finally:
|
|
asyncio.sleep(1)
|
|
print("[Overwatch] Check completed successfully.")
|
|
# Wait for the timeout
|
|
await asyncio.sleep(timeout)
|
|
else:
|
|
await asyncio.sleep(1)
|
|
|
|
# Every timeout seconds, update player league and check for rank changes
|
|
async def league_rank_change(timeout):
|
|
while True:
|
|
if discord_is_ready:
|
|
print("[League] Starting check for rank changes...")
|
|
# Update data for every player in list
|
|
for player in db:
|
|
if "league" in db[player]:
|
|
try:
|
|
r = await league.get_player_rank(**db[player]["league"])
|
|
except league.NoRankedGamesCompletedException:
|
|
# If the player has no ranked games completed, skip him
|
|
pass
|
|
except league.RateLimitException:
|
|
# If you've been ratelimited, skip the player and notify the console.
|
|
print("[League] Request rejected for rate limit.")
|
|
except Exception:
|
|
# If some other error occours, skip the player
|
|
print("[League] Request returned an unhandled exception.")
|
|
else:
|
|
# Convert tier into a number
|
|
tier_number = league.ranklist.index(r["tier"])
|
|
roman_number = league.roman.index(r["entries"][0]["division"])
|
|
# Check for tier changes
|
|
if tier_number != db[player]["league"]["tier"] \
|
|
or roman_number != db[player]["league"]["division"]:
|
|
# Send the message
|
|
loop.create_task(send_event(eventmsg=s.league_rank_up,
|
|
player=player,
|
|
tier=s.league_tier_list[tier_number],
|
|
division=r["entries"][0]["division"],
|
|
oldtier=s.league_tier_list[db[player]["league"]["tier"]],
|
|
olddivision=
|
|
s.league_roman_list[db[player]["league"]["division"]]))
|
|
# Update database
|
|
db[player]["league"]["tier"] = tier_number
|
|
db[player]["league"]["division"] = roman_number
|
|
f = open("db.json", "w")
|
|
json.dump(db, f)
|
|
f.close()
|
|
finally:
|
|
# Prevent getting ratelimited by Riot
|
|
await asyncio.sleep(2)
|
|
print("[League] Rank check completed.")
|
|
# Wait for the timeout
|
|
await asyncio.sleep(timeout)
|
|
else:
|
|
await asyncio.sleep(1)
|
|
|
|
# Every timeout seconds, update player level and check for changes
|
|
async def league_level_up(timeout):
|
|
while True:
|
|
if discord_is_ready:
|
|
print("[League] Starting check for level changes...")
|
|
# Update data for every player in list
|
|
for player in db:
|
|
if "league" in db[player]:
|
|
try:
|
|
r = await league.get_player_info(**db[player]["league"])
|
|
except league.RateLimitException:
|
|
# If you've been ratelimited, skip the player and notify the console.
|
|
print("[League] Request rejected for rate limit.")
|
|
except Exception:
|
|
# If some other error occours, skip the player
|
|
print("[League] Request returned an unhandled exception.")
|
|
else:
|
|
# Check for level changes
|
|
if "level" not in db[player]["league"] or r["summonerLevel"] > db[player]["league"]["level"]:
|
|
# Send the message
|
|
loop.create_task(send_event(eventmsg=s.league_level_up,
|
|
player=player,
|
|
level=r["summonerLevel"]))
|
|
# Update database
|
|
db[player]["league"]["level"] = r["summonerLevel"]
|
|
f = open("db.json", "w")
|
|
json.dump(db, f)
|
|
f.close()
|
|
finally:
|
|
# Prevent getting ratelimited by Riot
|
|
await asyncio.sleep(2)
|
|
print("[League] Level check completed.")
|
|
# Wait for the timeout
|
|
await asyncio.sleep(timeout)
|
|
else:
|
|
await asyncio.sleep(1)
|
|
|
|
# Every timeout seconds, update player mmr and rank
|
|
async def brawlhalla_update_mmr(timeout):
|
|
while True:
|
|
if discord_is_ready:
|
|
print("[Brawlhalla] Starting check for mmr changes...")
|
|
# Update mmr for every player in list
|
|
for player in db:
|
|
if "brawlhalla" in db[player]:
|
|
try:
|
|
r = await brawlhalla.get_leaderboard_for(db[player]["brawlhalla"]["username"])
|
|
except None:
|
|
print("[Brawlhalla] Request returned an unhandled exception.")
|
|
else:
|
|
# Parse the page
|
|
bs = bs4.BeautifulSoup(r.text, "html.parser")
|
|
# Divide the page into rows
|
|
rows = bs.find_all("tr")
|
|
# Find the row containing the rank
|
|
for row in rows:
|
|
# Skip header rows
|
|
if row.has_attr('id') and row['id'] == "rheader":
|
|
continue
|
|
# Check if the row belongs to the correct player
|
|
# (Brawlhalla searches aren't case sensitive)
|
|
columns = list(row.children)
|
|
for column in columns:
|
|
# Find the player name column
|
|
if column.has_attr('class') and column['class'][0] == "pnameleft":
|
|
# Check if the name matches the parameter
|
|
if column.string == db[player]["brawlhalla"]["username"]:
|
|
break
|
|
else:
|
|
continue
|
|
# Get the current mmr
|
|
mmr = int(list(row.children)[7].string)
|
|
# Compare the mmr with the value saved in the database
|
|
if mmr != db[player]["brawlhalla"]["mmr"]:
|
|
# Send a message
|
|
loop.create_task(send_event(s.brawlhalla_new_mmr, player, mmr=mmr,
|
|
oldmmr=db[player]["brawlhalla"]["mmr"]))
|
|
# Update database
|
|
db[player]["brawlhalla"]["mmr"] = mmr
|
|
f = open("db.json", "w")
|
|
json.dump(db, f)
|
|
f.close()
|
|
break
|
|
finally:
|
|
await asyncio.sleep(1)
|
|
print("[Brawlhalla] Request returned an unhandled exception.")
|
|
await asyncio.sleep(timeout)
|
|
else:
|
|
await asyncio.sleep(1)
|
|
|
|
# Send a new event to both Discord and Telegram
|
|
async def send_event(eventmsg: str, player: str, **kwargs):
|
|
# Create arguments dict
|
|
mapping = kwargs.copy()
|
|
mapping["eventmsg"] = None
|
|
# Discord
|
|
# The user id is the player argument; convert that into a mention
|
|
mapping["player"] = "<@" + player + ">"
|
|
# Format the event message
|
|
msg = eventmsg.format(**mapping)
|
|
# Send the message
|
|
loop.create_task(d_client.send_message(d_client.get_channel("213655027842154508"), msg))
|
|
# Telegram
|
|
# Find the matching Telegram username inside the db
|
|
mapping["player"] = "@" + db[player]["telegram"]["username"]
|
|
# Convert the Discord Markdown to Telegram Markdown
|
|
msg = eventmsg.replace("**", "*")
|
|
# Format the event message
|
|
msg = msg.format(**mapping)
|
|
# Send the message
|
|
loop.create_task(telegram.send_message(msg, -2141322))
|
|
|
|
|
|
loop.create_task(overwatch_status_change(600))
|
|
print("[Overwatch] Added level up check to the queue.")
|
|
|
|
loop.create_task(league_rank_change(900))
|
|
print("[League] Added rank change check to the queue.")
|
|
|
|
loop.create_task(league_level_up(900))
|
|
print("[League] Added level change check to the queue.")
|
|
|
|
loop.create_task(brawlhalla_update_mmr(1800))
|
|
print("[Brawlhalla] Added mmr change check to the queue.")
|
|
|
|
|
|
try:
|
|
loop.run_until_complete(d_client.start(token))
|
|
except KeyboardInterrupt:
|
|
loop.run_until_complete(d_client.logout())
|
|
# cancel all tasks lingering
|
|
finally:
|
|
loop.close()
|