#!/usr/bin/env python # -*- coding: utf-8 -*- from telegram.ext import Updater, CommandHandler, MessageHandler, Filters from telegram import ParseMode import logging import os import random import requests import datetime import cat import pydog DIETS = { "veggy" : "fleischlos", "fleisch" : "mit Fleisch", "fisch" : "mit Fisch/ Meeresfrüchten" } WEEKDAYS = { "montag" : 1, "dienstag" : 2, "mittwoch" : 3, "donnerstag" : 4, "freitag" : 5, "samstag" : 6 } MENSA_URL = { "zentral": "zentralmensa", "nord": "nordmensa", "turm": "turmmensa", "italia": "mensaitalia", "fasthochschule": "bistrohawk" } MENSA_NAME = { "zentral": "Zentralmensa", "nord": "Nordmensa", "turm": "Turmmensa", "italia": "Mensa Italia", "fasthochschule": "Bistro Fasthochschule" } HIDE_CATEGORIES_LIGHT = { "CampusCurry", "natürlich fit", "Fitnesscenter", "Salatbuffet", "Studentenfutter", "Süße Versuchung", "Süße Spezial Tagesangebot", "Vollwert & Co. Stärke", "Vollwert & Co. Gemüse", "Bio-Beilagen", "Dessertbuffet", "Last Minute ab 14:30 Uhr", ## Nordmensa: "Salatbuffet/Pastapoint", "Last Minute ab 13:30 Uhr", ## Turmmensa: "Beilagen", "Last Minute ab 14:00Uhr" } HIDE_CATEGORIES_FULL = { "Last Minute ab 14:30 Uhr", "Last Minute ab 13:30 Uhr", "Last Minute ab 14:00Uhr" } MODES = { "light" : HIDE_CATEGORIES_LIGHT, "full" : HIDE_CATEGORIES_FULL } # Enable logging logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO) logger = logging.getLogger(__name__) with open("stoll.txt", "r") as tmp_file: STOLL = tmp_file.readlines() with open("manta.txt", "r") as tmp_file: MANTA = tmp_file.readlines() with open("goodlife.txt", "r") as tmp_file: GOOD_LIFE = 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 (https://dog.ceo/dog-api) /magie - random Axel Stoll quote /manta - random Opel Manta joke /goodlife - random good life quote /help - this help text""", parse_mode=ParseMode.MARKDOWN) def sendCat(bot, update): """CUTE""" catfile = cat.getCat(); bot.send_photo(chat_id=update.message.chat_id, photo=open(catfile, 'rb')) os.remove(catfile) def sendDog(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""" message = update.message.from_user.first_name + " died" if sudocall: message += " for real" update.message.reply_text(message) def revive(bot, update, sudocall = False): """unkill me pls""" message = update.message.from_user.first_name + " is living again" if sudocall: message += " for real" update.message.reply_text(message) SUDOCMDS = { "kill" : kill, "revive" : revive } def sudo(bot, update, args): """for real""" if not args : update.message.reply_text("Unknown command") return for arg in args: if arg in SUDOCMDS: SUDOCMDS[arg](bot, update, True) else: update.message.reply_text("Unknown command") def sendCatDog(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): "KAUF DIR N HUND" update.message.reply_text("@Justus_vonRamme bitte kauf dir einen Hund oder eine Katze, du hast zu viel Zeit") def mensa(bot, update, args): which = "zentral" filter_categories = MODES["light"] diet = "" today = datetime.datetime.now().date().weekday() + 1 if datetime.datetime.now().time() > datetime.time(hour=16): # Es ist zu spät am Tag, zeig das essen für morgen an today += 1 today = today % 7 for arg in args: if arg in MENSA_NAME: which = arg elif arg in MODES: filter_categories = MODES[arg] elif arg in WEEKDAYS: today = WEEKDAYS[arg] elif arg in DIETS: diet = DIETS[arg] else: update.message.reply_text("Falscher Aufruf! RTFM und versuchs nochmal.") return if today == 0: update.message.reply_text("Sonntags hat die Mensa zu") return url = "https://mensa.schneider-hosting.de/static/%s.%d.json" % (MENSA_URL[which], today) request = requests.get(url) request.encoding = 'utf-8' data = request.json() message = "Das Essen für %s in der %s \n\n" % (data["date"], MENSA_NAME[which]) if len(data["meals"]) > 1: for meal in data["meals"]: if meal["category"] not in filter_categories and diet in meal["diet"]: meal_line = "*%s*\n" % meal["category"] meal_line += meal["title"].strip() + "\n" # Discord only allows up to 2000 chars per message if len(message) + len(meal_line) > 2000: update.message.reply_text(message, parse_mode=ParseMode.MARKDOWN) message = meal_line + "\n" else: message += meal_line + "\n" update.message.reply_text(message, parse_mode=ParseMode.MARKDOWN) else: update.message.reply_text("Nix zu futtern heute :(") 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 main(): """Start the bot.""" # Create the EventHandler and pass it your bot's token. token = open("token").read() updater = Updater(token.strip()) # Get the dispatcher to register handlers dp = updater.dispatcher # on different commands - answer in Telegram 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", sendCat)) dp.add_handler(CommandHandler("dog", sendDog)) dp.add_handler(CommandHandler("catdog", sendCatDog)) dp.add_handler(CommandHandler("shrug", shrug)) dp.add_handler(CommandHandler("kill", kill)) dp.add_handler(CommandHandler("revive", revive)) dp.add_handler(CommandHandler("simon", simon)) # 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()