#!/usr/bin/env python # -*- coding: utf-8 -*- import logging import random import pydog import requests from telegram import ParseMode from telegram.ext import Updater, CommandHandler from coding_love import send_coding_love_gif from reddit import wrong_dog, hamster, reddit, reddit_img from db import Db from mensa import mensa # Enable logging logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO) logger = logging.getLogger(__name__) with open("res/stoll.txt", "r") as tmp_file: STOLL = tmp_file.readlines() with open("res/manta.txt", "r") as tmp_file: MANTA = tmp_file.readlines() with open("res/goodlife.txt", "r") as tmp_file: GOOD_LIFE = tmp_file.readlines() with open("res/simon.txt", "r") as tmp_file: SIMON = tmp_file.readlines() # Define a few command handlers. These usually take the two arguments bot and # update. Error handlers also receive the raised TelegramError object in error. def start(bot, update): """Was ist das hier für eins Bot?""" update.message.reply_text("""Hallo, ich bin ein Bot. Ich liefere tolle Informationen über die Mensa. Die Infos kommen von https://mensa.schneider-hosting.de Mhmm, lecker. Guten Appetit!""") def help(bot, update): """Send a message when the command /help is issued.""" update.message.reply_text("""Commands: *Mensa*: /mensa _mensa_ - prints filtered list of meals for _mensa_ (if no _mensa_ given, zentralmensa is used) /mensa _mensa_ full - prints full list of meals for _mensa_ (if no _mensa_ given, zentralmensa is used) *Fun*: /cat - random cat (using thecatapi.com) /dog - random dog (using dog.ceo/dog-api) /codinglove - random gif from the coding love (https://thecodinglove.com) /magie - random Axel Stoll quote /manta - random Opel Manta joke /goodlife - random good life quote /shrug - prints shrug /kill - kills you /revive - revives you /sudo kill/revive - force kill/revive /graveyard - shows the dead people /help - this help text""", parse_mode=ParseMode.MARKDOWN) def send_cat(bot, update): """CUTE""" cat_file = requests.get('http://aws.random.cat/meow').json()['file'] bot.send_photo(chat_id=update.message.chat_id, photo=cat_file) def send_dog(bot, update): """CUTE 2.0""" dog = pydog.PyDog() bot.send_photo(chat_id=update.message.chat_id, photo=dog.get_random_image()) def kill(bot, update, sudocall=False): """kill me pls""" db = Db() user = update.message.from_user.first_name if db.is_dead(user, update.message.chat_id): if sudocall: update.message.reply_text("%s killed again" % user) return update.message.reply_text("%s is already dead" % user) return message = update.message.from_user.first_name + " died" if sudocall: message += " for real" db.kill(user, update.message.chat_id) update.message.reply_text(message) def revive(bot, update, sudocall=False): """unkill me pls""" db = Db() user = update.message.from_user.first_name if not db.is_dead(user, update.message.chat_id): update.message.reply_text("Maybe %s should have a litte 'accident' before" % user) return if sudocall: update.message.reply_text("%s is living again, for real" % user) db.revive(user, update.message.chat_id) return update.message.reply_text("You fool! You cannot revive a dead person!") SUDO_CMDS = { "kill": kill, "revive": revive } def graveyard(bot, update): """List the dead""" db = Db() update.message.reply_text("Here are the dead people:\n%s" % db.get_dead_bodies(update.message.chat_id)) def sudo(bot, update, args): """for real""" if not args: update.message.reply_text("Unknown command") return for arg in args: if arg in SUDO_CMDS: SUDO_CMDS[arg](bot, update, True) else: update.message.reply_text("Unknown command") def send_cat_dog(bot, update): """Best of both worlds!""" catdog = "https://upload.wikimedia.org/wikipedia/en/6/64/CatDog.jpeg" bot.send_photo(chat_id=update.message.chat_id, photo=catdog) def magie(bot, update): """MAGIEEEEEE""" rand = random.SystemRandom() update.message.reply_text(STOLL[rand.randrange(0, len(STOLL), 1)]) def manta(bot, update): """Boah ey, geile Karre!""" rand = random.SystemRandom() update.message.reply_text(MANTA[rand.randrange(0, len(MANTA), 1)]) def goodlife(bot, update): """GOOD LIFE MY A...""" rand = random.SystemRandom() update.message.reply_text(GOOD_LIFE[rand.randrange(0, len(GOOD_LIFE), 1)]) def shrug(bot, update): """SHRUG""" update.message.reply_text("¯\_(ツ)_/¯") def simon(bot, update, args): "KAUF DIR N HUND" name = "@thedailysimon" if len(args) > 0: name = args[0] rand = random.SystemRandom() animal = SIMON[rand.randrange(0, len(SIMON), 1)].replace('\n', '') update.message.reply_text( "%s, bitte kauf dir einen Hund, eine Katze oder von mir aus auch %s, du hast zu viel Zeit" % (name, animal)) def echo(bot, update): """Echo the user message.""" update.message.reply_text(update.message.text) def error(bot, update, error): """Log Errors caused by Updates.""" logger.warning('Update "%s" caused error "%s"', update, error) def init_commands(dp): dp.add_handler(CommandHandler("start", start)) dp.add_handler(CommandHandler("help", help)) dp.add_handler(CommandHandler("magie", magie)) dp.add_handler(CommandHandler("manta", manta)) dp.add_handler(CommandHandler("goodlife", goodlife)) dp.add_handler(CommandHandler("mensa", mensa, pass_args=True)) dp.add_handler(CommandHandler("sudo", sudo, pass_args=True)) dp.add_handler(CommandHandler("cat", send_cat)) dp.add_handler(CommandHandler("dog", send_dog)) dp.add_handler(CommandHandler("catdog", send_cat_dog)) dp.add_handler(CommandHandler("shrug", shrug)) dp.add_handler(CommandHandler("kill", kill)) dp.add_handler(CommandHandler("revive", revive)) dp.add_handler(CommandHandler("simon", simon, pass_args=True)) dp.add_handler(CommandHandler("graveyard", graveyard)) dp.add_handler(CommandHandler("codinglove", send_coding_love_gif)) dp.add_handler(CommandHandler("wrongdog", wrong_dog)) dp.add_handler(CommandHandler("hamster", hamster)) dp.add_handler(CommandHandler("reddit", reddit, pass_args=True)) dp.add_handler(CommandHandler("reddit_img", reddit_img, pass_args=True)) def main(): """Start the bot.""" # Create the EventHandler and pass it your bot's token. token = open("token").read() # token = open("test_token").read() updater = Updater(token.strip()) # Get the dispatcher to register handlers dp = updater.dispatcher init_commands(dp) # on different commands - answer in Telegram # log all errors dp.add_error_handler(error) # Start the Bot updater.start_polling() # Run the bot until you press Ctrl-C or the process receives SIGINT, # SIGTERM or SIGABRT. This should be used most of the time, since # start_polling() is non-blocking and will stop the bot gracefully. updater.idle() if __name__ == '__main__': main()