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

Add @database_access to /cast

Browse source
This commit is contained in:
Steffo 2019-02-10 02:15:10 +01:00
parent 33fa9825dd
commit 51b8deaa9c
1 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
telegrambot.py
View file

@ -173,19 +173,16 @@ def cmd_cv(bot: telegram.Bot, update: telegram.Update):
@command @command
def cmd_cast(bot: telegram.Bot, update: telegram.Update): @database_access
def cmd_cast(bot: telegram.Bot, update: telegram.Update, session: db.Session):
try: try:
spell: str = update.message.text.split(" ", 1)[1] spell: str = update.message.text.split(" ", 1)[1]
except IndexError: except IndexError:
bot.send_message(update.message.chat.id, "⚠️ Non hai specificato nessun incantesimo!\n" bot.send_message(update.message.chat.id, "⚠️ Non hai specificato nessun incantesimo!\n"
"Sintassi corretta: `/cast <nome_incantesimo>`", parse_mode="Markdown") "Sintassi corretta: `/cast <nome_incantesimo>`", parse_mode="Markdown")
return return
# Open a new db session
session = db.Session()
# Find a target for the spell # Find a target for the spell
target = random.sample(session.query(db.Telegram).all(), 1)[0] target = random.sample(session.query(db.Telegram).all(), 1)[0]
# Close the session
session.close()
# END # END
bot.send_message(update.message.chat.id, cast.cast(spell_name=spell, bot.send_message(update.message.chat.id, cast.cast(spell_name=spell,
target_name=target.username if target.username is not None target_name=target.username if target.username is not None