ryg/royalnet
ryg/royalnet
1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
Code Activity

Move database connection creation to the @database_access decorator

Browse source
This commit is contained in:
Steffo 2019-02-10 02:13:34 +01:00
parent 223e2f5296
commit 33fa9825dd
1 changed files with 73 additions and 59 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

64
telegrambot.py
View file

@ -81,8 +81,7 @@ def command(func: "function"):
# noinspection PyBroadException # noinspection PyBroadException
try: try:
bot.send_chat_action(update.message.chat.id, telegram.ChatAction.TYPING) bot.send_chat_action(update.message.chat.id, telegram.ChatAction.TYPING)
session = db.Session() return func(bot, update)
return func(bot, update, session)
except TimedOut: except TimedOut:
logger.warning(f"Telegram timed out in {update}") logger.warning(f"Telegram timed out in {update}")
except Exception: except Exception:
@ -107,20 +106,35 @@ def command(func: "function"):
"update": update.to_dict() "update": update.to_dict()
}) })
sentry.captureException() sentry.captureException()
return new_func
def database_access(func: "function"):
def new_func(bot: telegram.Bot, update: telegram.Update):
try:
session = db.Session()
return func(bot, update, session)
except Exception:
if __debug__:
raise
logger.error(f"Database error: {sys.exc_info()}")
sentry.captureException()
finally: finally:
try:
session.close() session.close()
except Exception:
pass
return new_func return new_func
@command @command
def cmd_ping(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_ping(bot: telegram.Bot, update: telegram.Update):
reply(bot, update, strings.PONG) reply(bot, update, strings.PONG)
@command @command
@database_access
def cmd_link(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_link(bot: telegram.Bot, update: telegram.Update, session: db.Session):
session = db.Session()
try:
try: try:
username = update.message.text.split(" ", 1)[1] username = update.message.text.split(" ", 1)[1]
except IndexError: except IndexError:
@ -142,12 +156,10 @@ def cmd_link(bot: telegram.Bot, update: telegram.Update, session: db.Session):
session.add(t) session.add(t)
session.commit() session.commit()
reply(bot, update, strings.LINK.SUCCESS) reply(bot, update, strings.LINK.SUCCESS)
finally:
session.close()
@command @command
def cmd_cv(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_cv(bot: telegram.Bot, update: telegram.Update):
if discord_connection is None: if discord_connection is None:
reply(bot, update, strings.TELEGRAM.ERRORS.INACTIVE_BRIDGE) reply(bot, update, strings.TELEGRAM.ERRORS.INACTIVE_BRIDGE)
return return
@ -161,7 +173,7 @@ def cmd_cv(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
def cmd_cast(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_cast(bot: telegram.Bot, update: telegram.Update):
try: try:
spell: str = update.message.text.split(" ", 1)[1] spell: str = update.message.text.split(" ", 1)[1]
except IndexError: except IndexError:
@ -182,19 +194,19 @@ def cmd_cast(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
def cmd_color(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_color(bot: telegram.Bot, update: telegram.Update):
bot.send_message(update.message.chat.id, "I am sorry, unknown error occured during working with your request," bot.send_message(update.message.chat.id, "I am sorry, unknown error occured during working with your request,"
" Admin were notified") " Admin were notified")
@command @command
def cmd_smecds(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_smecds(bot: telegram.Bot, update: telegram.Update):
ds = random.sample(stagismo.listona, 1)[0] ds = random.sample(stagismo.listona, 1)[0]
bot.send_message(update.message.chat.id, f"Secondo me, è colpa {ds}.") bot.send_message(update.message.chat.id, f"Secondo me, è colpa {ds}.")
@command @command
def cmd_ciaoruozi(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_ciaoruozi(bot: telegram.Bot, update: telegram.Update):
if update.message.from_user.username.lstrip("@") == "MeStakes": if update.message.from_user.username.lstrip("@") == "MeStakes":
bot.send_message(update.message.chat.id, "Ciao me!") bot.send_message(update.message.chat.id, "Ciao me!")
else: else:
@ -202,7 +214,7 @@ def cmd_ciaoruozi(bot: telegram.Bot, update: telegram.Update, session: db.Sessio
@command @command
def cmd_ahnonlosoio(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_ahnonlosoio(bot: telegram.Bot, update: telegram.Update):
if update.message.reply_to_message is not None and update.message.reply_to_message.text in [ if update.message.reply_to_message is not None and update.message.reply_to_message.text in [
"/ahnonlosoio", "/ahnonlosoio@royalgamesbot", "Ah, non lo so io!", "Ah, non lo so neppure io!" "/ahnonlosoio", "/ahnonlosoio@royalgamesbot", "Ah, non lo so io!", "Ah, non lo so neppure io!"
]: ]:
@ -212,9 +224,8 @@ def cmd_ahnonlosoio(bot: telegram.Bot, update: telegram.Update, session: db.Sess
@command @command
@database_access
def cmd_balurage(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_balurage(bot: telegram.Bot, update: telegram.Update, session: db.Session):
session = db.Session()
try:
user = session.query(db.Telegram).filter_by(telegram_id=update.message.from_user.id).one_or_none() user = session.query(db.Telegram).filter_by(telegram_id=update.message.from_user.id).one_or_none()
if user is None: if user is None:
bot.send_message(update.message.chat.id, bot.send_message(update.message.chat.id,
@ -230,13 +241,10 @@ def cmd_balurage(bot: telegram.Bot, update: telegram.Update, session: db.Session
session.add(br) session.add(br)
session.commit() session.commit()
bot.send_message(update.message.chat.id, f"😡 Stai sfogando la tua ira sul bot!") bot.send_message(update.message.chat.id, f"😡 Stai sfogando la tua ira sul bot!")
except Exception:
raise
finally:
session.close()
@command @command
@database_access
def cmd_diario(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_diario(bot: telegram.Bot, update: telegram.Update, session: db.Session):
user = session.query(db.Telegram).filter_by(telegram_id=update.message.from_user.id).one_or_none() user = session.query(db.Telegram).filter_by(telegram_id=update.message.from_user.id).one_or_none()
if user is None: if user is None:
@ -267,6 +275,7 @@ def cmd_diario(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
@database_access
def cmd_vote(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_vote(bot: telegram.Bot, update: telegram.Update, session: db.Session):
user = session.query(db.Telegram).filter_by(telegram_id=update.message.from_user.id).one_or_none() user = session.query(db.Telegram).filter_by(telegram_id=update.message.from_user.id).one_or_none()
if user is None: if user is None:
@ -320,6 +329,7 @@ def generate_search_message(term, entries):
@command @command
@database_access
def cmd_search(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_search(bot: telegram.Bot, update: telegram.Update, session: db.Session):
try: try:
query = update.message.text.split(" ", 1)[1] query = update.message.text.split(" ", 1)[1]
@ -337,6 +347,7 @@ def cmd_search(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
@database_access
def cmd_regex(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_regex(bot: telegram.Bot, update: telegram.Update, session: db.Session):
try: try:
query = update.message.text.split(" ", 1)[1] query = update.message.text.split(" ", 1)[1]
@ -352,6 +363,7 @@ def cmd_regex(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
@database_access
def cmd_mm(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_mm(bot: telegram.Bot, update: telegram.Update, session: db.Session):
user = session.query(db.Telegram).filter_by(telegram_id=update.message.from_user.id).one_or_none() user = session.query(db.Telegram).filter_by(telegram_id=update.message.from_user.id).one_or_none()
if user is None: if user is None:
@ -380,6 +392,7 @@ def cmd_mm(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
@database_access
def on_callback_query(bot: telegram.Bot, update: telegram.Update, session: db.Session): def on_callback_query(bot: telegram.Bot, update: telegram.Update, session: db.Session):
if update.callback_query.data.startswith("vote_"): if update.callback_query.data.startswith("vote_"):
if update.callback_query.data == "vote_yes": if update.callback_query.data == "vote_yes":
@ -488,7 +501,7 @@ def on_callback_query(bot: telegram.Bot, update: telegram.Update, session: db.Se
@command @command
def cmd_eat(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_eat(bot: telegram.Bot, update: telegram.Update):
try: try:
food: str = update.message.text.split(" ", 1)[1].capitalize() food: str = update.message.text.split(" ", 1)[1].capitalize()
except IndexError: except IndexError:
@ -501,7 +514,7 @@ def cmd_eat(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
def cmd_ship(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_ship(bot: telegram.Bot, update: telegram.Update):
try: try:
_, name_one, name_two = update.message.text.split(" ", 2) _, name_one, name_two = update.message.text.split(" ", 2)
except ValueError: except ValueError:
@ -522,7 +535,7 @@ def cmd_ship(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
def cmd_bridge(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_bridge(bot: telegram.Bot, update: telegram.Update):
try: try:
data = update.message.text.split(" ", 1)[1] data = update.message.text.split(" ", 1)[1]
except IndexError: except IndexError:
@ -640,6 +653,7 @@ def cmd_newevent(bot: telegram.Bot, update: telegram.Update, session: db.Session
@command @command
@database_access
def cmd_calendar(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_calendar(bot: telegram.Bot, update: telegram.Update, session: db.Session):
next_events = session.query(db.Event).filter(db.Event.time > datetime.datetime.now()).order_by(db.Event.time).all() next_events = session.query(db.Event).filter(db.Event.time > datetime.datetime.now()).order_by(db.Event.time).all()
msg = "📆 Prossimi eventi\n" msg = "📆 Prossimi eventi\n"
@ -655,7 +669,7 @@ def cmd_calendar(bot: telegram.Bot, update: telegram.Update, session: db.Session
@command @command
def cmd_markov(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_markov(bot: telegram.Bot, update: telegram.Update):
if model is None: if model is None:
bot.send_message(update.message.chat.id, strings.MARKOV.ERRORS.NO_MODEL) bot.send_message(update.message.chat.id, strings.MARKOV.ERRORS.NO_MODEL)
return return
@ -682,7 +696,7 @@ def cmd_markov(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
def cmd_roll(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_roll(bot: telegram.Bot, update: telegram.Update):
dice_string = update.message.text.split(" ", 1)[1] dice_string = update.message.text.split(" ", 1)[1]
try: try:
result = dice.roll(f"{dice_string}t") result = dice.roll(f"{dice_string}t")
@ -693,7 +707,7 @@ def cmd_roll(bot: telegram.Bot, update: telegram.Update, session: db.Session):
@command @command
def cmd_start(bot: telegram.Bot, update: telegram.Update, session: db.Session): def cmd_start(bot: telegram.Bot, update: telegram.Update):
reply(bot, update, strings.TELEGRAM.BOT_STARTED) reply(bot, update, strings.TELEGRAM.BOT_STARTED)