forked from angerstoner/schneiderbot-tg
Telegram version of schneiderbot.
I'll totally do something w/ this, this is never going to be effectively a mirror.
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
229 lines
6.1 KiB
229 lines
6.1 KiB
#!/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
|
|
|
|
MENSA_URL = {
|
|
"zentral": "zentralmensa",
|
|
"nord": "nordmensa",
|
|
"turm": "turmmensa"
|
|
}
|
|
|
|
MENSA_NAME = {
|
|
"zentral": "Zentralmensa",
|
|
"nord": "Nordmensa",
|
|
"turm": "Turmmensa"
|
|
}
|
|
|
|
## TODO: rather use filter list, this is bloated; mensa italia missing
|
|
CATEGORIES_LIGHT = {
|
|
"Eintopf",
|
|
"Menü I",
|
|
"Menü II",
|
|
"Vegan",
|
|
"Vegetarisch",
|
|
"Grill III",
|
|
"Al dente",
|
|
"Pasta und Gratins",
|
|
"Kartoffeln und Co.",
|
|
## Nordmensa:
|
|
"Stamm 1 vegetarisch",
|
|
"Nordmensa-Fit",
|
|
"Stamm 2",
|
|
"Stamm 3",
|
|
## Turmmensa:
|
|
"Turm vegetarisch Kombi",
|
|
"Turm 2b",
|
|
"Turm 3a",
|
|
"Turm Vegan",
|
|
"Regionale Linie"
|
|
}
|
|
|
|
CATEGORIES_FULL = {
|
|
"Eintopf",
|
|
"Menü I",
|
|
"Menü II",
|
|
"Vegan",
|
|
"Vegetarisch",
|
|
"Grill III",
|
|
"CampusCurry",
|
|
"natürlich fit",
|
|
"Al dente",
|
|
"Pasta und Gratins",
|
|
"Fitnesscenter",
|
|
"Salatbuffet",
|
|
"Studentenfutter",
|
|
"Süße Versuchung",
|
|
"Süße Spezial Tagesangebot",
|
|
"Kartoffeln und Co.",
|
|
"Vollwert & Co. Stärke",
|
|
"Vollwert & Co. Gemüse",
|
|
"Bio-Beilagen",
|
|
"Dessertbuffet",
|
|
"Last Minute ab 14:30 Uhr",
|
|
## Nordmensa:
|
|
"Stamm 1 vegetarisch",
|
|
"Nordmensa-Fit",
|
|
"Stamm 2",
|
|
"Stamm 3",
|
|
"Salatbuffet/Pastapoint",
|
|
"Last Minute ab 13:30 Uhr",
|
|
## Turmmensa:
|
|
"Turm vegetarisch Kombi",
|
|
"Turm 2b",
|
|
"Turm 3a",
|
|
"Turm Vegan",
|
|
"Regionale Linie",
|
|
"Beilagen",
|
|
"Last Minute ab 14:00Uhr"
|
|
}
|
|
|
|
MODES = {
|
|
"light" : CATEGORIES_LIGHT,
|
|
"full" : 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()
|
|
|
|
# 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*:
|
|
/magie - random Axel Stoll qoute
|
|
/manta - random Opel Manta joke
|
|
|
|
/help - this help text""", parse_mode=ParseMode.MARKDOWN)
|
|
|
|
def ccat(bot, update):
|
|
"""CUTE"""
|
|
catfile = cat.getCat();
|
|
bot.send_photo(chat_id=update.message.chat_id, photo=open(catfile, 'rb'))
|
|
os.remove(catfile)
|
|
|
|
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 mensa(bot, update, args):
|
|
which = "zentral"
|
|
categories = CATEGORIES_LIGHT
|
|
for arg in args:
|
|
if arg in MENSA_NAME:
|
|
which = args[0]
|
|
elif arg in MODES:
|
|
categories = MODES[arg]
|
|
else:
|
|
update.message.reply_text("Falscher Aufruf! RTFM und versuchs nochmal.")
|
|
return
|
|
|
|
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
|
|
|
|
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" % (data["date"], MENSA_NAME[which])
|
|
|
|
if len(data["meals"]) > 1:
|
|
for meal in data["meals"]:
|
|
if meal["category"] in categories:
|
|
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("mensa", mensa, pass_args=True))
|
|
dp.add_handler(CommandHandler("cat", ccat))
|
|
|
|
|
|
# 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()
|
|
|