mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 21:44:21 +00:00
89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
|
import db
|
||
|
|
import time
|
||
|
|
import logging
|
||
|
|
import raven
|
||
|
|
import configparser
|
||
|
|
import os
|
||
|
|
import typing
|
||
|
|
import telegram
|
||
|
|
import sys
|
||
|
|
|
||
|
|
logging.getLogger().setLevel(level=logging.ERROR)
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
logger.addHandler(logging.StreamHandler())
|
||
|
|
logger.setLevel(level=logging.DEBUG)
|
||
|
|
|
||
|
|
# Init the config reader
|
||
|
|
config = configparser.ConfigParser()
|
||
|
|
config.read("config.ini")
|
||
|
|
|
||
|
|
# Init the Sentry client
|
||
|
|
sentry = raven.Client(config["Sentry"]["token"],
|
||
|
|
release=raven.fetch_git_sha(os.path.dirname(__file__)),
|
||
|
|
install_logging_hook=False,
|
||
|
|
hook_libraries=[])
|
||
|
|
|
||
|
|
telegram_bot = telegram.Bot(config["Telegram"]["bot_token"])
|
||
|
|
|
||
|
|
|
||
|
|
def update_block(block: list, delay: float=0, change_callback: typing.Callable=None):
|
||
|
|
for item in block:
|
||
|
|
logger.debug(f"Updating {repr(item)}.")
|
||
|
|
t = time.clock()
|
||
|
|
try:
|
||
|
|
change = item.update()
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Error {sys.exc_info()} while updating {repr(item)}.")
|
||
|
|
sentry.extra_context({
|
||
|
|
"item": repr(item)
|
||
|
|
})
|
||
|
|
sentry.captureException()
|
||
|
|
continue
|
||
|
|
if change:
|
||
|
|
change_callback(item)
|
||
|
|
sleep_time = delay - time.clock() + t
|
||
|
|
time.sleep(sleep_time if sleep_time > 0 else 0)
|
||
|
|
|
||
|
|
|
||
|
|
def new_dota_rank(item: db.Dota):
|
||
|
|
try:
|
||
|
|
telegram_bot.send_message(config["Telegram"]["main_group"],
|
||
|
|
f"✳️ {item.steam.royal.username} è salito a"
|
||
|
|
f" {item.get_rank_name()} {item.get_rank_number()} su Dota 2!")
|
||
|
|
except Exception:
|
||
|
|
logger.warning(f"Couldn't notify on Telegram: {item}")
|
||
|
|
|
||
|
|
|
||
|
|
def new_lol_rank(item: db.LeagueOfLegends):
|
||
|
|
try:
|
||
|
|
telegram_bot.send_message(config["Telegram"]["main_group"],
|
||
|
|
f"✳️ {item.royal.username} è salito di rank su League of Legends!")
|
||
|
|
except Exception:
|
||
|
|
logger.warning(f"Couldn't notify on Telegram: {item}")
|
||
|
|
|
||
|
|
|
||
|
|
def process():
|
||
|
|
while True:
|
||
|
|
session = db.Session()
|
||
|
|
#logger.info("Now updating Steam data.")
|
||
|
|
#update_block(session.query(db.Steam).all())
|
||
|
|
#session.commit()
|
||
|
|
#logger.info("Now updating Dota data.")
|
||
|
|
#update_block(session.query(db.Dota).all(), delay=1, change_callback=new_dota_rank)
|
||
|
|
#session.commit()
|
||
|
|
#logger.info("Now updating League of Legends data.")
|
||
|
|
#update_block(session.query(db.LeagueOfLegends).all(), delay=0.3, change_callback=new_lol_rank)
|
||
|
|
#session.commit()
|
||
|
|
#logger.info("Now updating osu! data.")
|
||
|
|
#update_block(session.query(db.Osu).all(), delay=0.3)
|
||
|
|
#session.commit()
|
||
|
|
logger.info("Now updating Overwatch data.")
|
||
|
|
update_block(session.query(db.Overwatch).all(), delay=1)
|
||
|
|
session.commit()
|
||
|
|
logger.info("Pausing for 30 minutes.")
|
||
|
|
time.sleep(1800)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
process()
|