ryg/royalnet
ryg/royalnet
1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-27 13:34:28 +00:00
Code Activity

Improved timeouts

Browse source
This commit is contained in:
Steffo 2016-08-13 15:36:08 +02:00
parent 3de55d688b
commit 2b2a7051d4
1 changed files with 63 additions and 53 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

116
main.py
View file

@ -8,6 +8,14 @@ import telegram
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
d_client = discord.Client() 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 # Get player database from the db.json file
file = open("db.json") file = open("db.json")
@ -22,70 +30,72 @@ file.close()
# Every timeout seconds, update player status and check for levelups # Every timeout seconds, update player status and check for levelups
async def overwatch_level_up(timeout): async def overwatch_level_up(timeout):
while True: while True:
# Wait for the Discord client to be ready if discord_is_ready:
await d_client.wait_until_ready() print("Checking for Overwatch updates.")
print("Checking for Overwatch updates.") # Update data for every player in list
# Update data for every player in list for player in db:
for player in db: if "overwatch" in db[player]:
if "overwatch" in db[player]: r = await overwatch.get_player_data(**db[player]["overwatch"])
r = await overwatch.get_player_data(**db[player]["overwatch"]) if r["data"]["level"] > db[player]["overwatch"]["level"]:
if r["data"]["level"] > db[player]["overwatch"]["level"]:
# Convert user ID into a mention
user = "<@" + player + ">"
# Prepare the message to send
msg = s.overwatch_level_up.format(player=user, level=r["data"]["level"])
# Send the message to the discord channel
loop.create_task(d_client.send_message(d_client.get_channel("213655027842154508"), msg))
# Send the message to the telegram group chat
loop.create_task(telegram.send_message(msg, -2141322))
# Update database
db[player]["overwatch"]["level"] = r["data"]["level"]
f = open("db.json", "w")
json.dump(db, f)
f.close()
print("Check for Overwatch completed.")
# Wait for the timeout
await asyncio.sleep(timeout)
# Every timeout seconds, update player league and check for rank changes
async def league_rank_change(timeout):
while True:
# Wait for the Discord client to be ready
await d_client.wait_until_ready()
print("Checking for League of Legends updates.")
# 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
continue
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"]:
# Convert user ID into a mention # Convert user ID into a mention
user = "<@" + player + ">" user = "<@" + player + ">"
# Prepare the message to send # Prepare the message to send
msg = s.league_rank_up.format(player=user, msg = s.overwatch_level_up.format(player=user, level=r["data"]["level"])
tier=s.league_tier_list[tier_number],
division=r["entries"][0]["division"])
# Send the message to the discord channel # Send the message to the discord channel
loop.create_task(d_client.send_message(d_client.get_channel("213655027842154508"), msg)) loop.create_task(d_client.send_message(d_client.get_channel("213655027842154508"), msg))
# Send the message to the telegram group chat # Send the message to the telegram group chat
loop.create_task(telegram.send_message(msg, -2141322)) loop.create_task(telegram.send_message(msg, -2141322))
# Update database # Update database
db[player]["league"]["tier"] = tier_number db[player]["overwatch"]["level"] = r["data"]["level"]
db[player]["league"]["division"] = roman_number
f = open("db.json", "w") f = open("db.json", "w")
json.dump(db, f) json.dump(db, f)
f.close() f.close()
print("Check for League of Legends completed.") print("Check for Overwatch completed.")
# Wait for the timeout # Wait for the timeout
await asyncio.sleep(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("Checking for League of Legends updates.")
# 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
continue
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"]:
# Convert user ID into a mention
user = "<@" + player + ">"
# Prepare the message to send
msg = s.league_rank_up.format(player=user,
tier=s.league_tier_list[tier_number],
division=r["entries"][0]["division"])
# Send the message to the discord channel
loop.create_task(d_client.send_message(d_client.get_channel("213655027842154508"), msg))
# Send the message to the telegram group chat
loop.create_task(telegram.send_message(msg, -2141322))
# 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()
print("Check for League of Legends completed.")
# Wait for the timeout
await asyncio.sleep(timeout)
else:
await asyncio.sleep(1)
print("Added Overwatch to the queue.") print("Added Overwatch to the queue.")
loop.create_task(overwatch_level_up(900)) loop.create_task(overwatch_level_up(900))