#! /usr/bin/env python3 """ Code for schneiderbot """ import logging import datetime import random # import urllib.request # import json import requests from discord.ext import commands logging.basicConfig(level=logging.INFO) DESCRIPTION = '''Schneiderbot, bald mit tollen Funktionen''' BOT = commands.Bot(command_prefix='!', description=DESCRIPTION) MENSA_URL = { "zentral": "zentralmensa", "nord": "nordmensa", "turm": "turmmensa" } MENSA_NAME = { "zentral": "Zentralmensa", "nord": "Nordmensa", "turm": "Turmmensa" } with open("stoll.txt", "r") as tmp_file: STOLL = tmp_file.readlines() with open("manta.txt", "r") as tmp_file: MANTA = tmp_file.readlines() @BOT.event async def on_ready(): """ On ready """ print('Logged in as') print(BOT.user.name) print(BOT.user.id) print('------') @BOT.command() async def hello(): """Was ist das hier für eins Bot?""" await BOT.say("""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!""") @BOT.command() async def mensa(which="zentral"): """Gibt leckere, äh, nützliche Infos über die Mensa aus. Standardmäßig für die Zentralmensa, aber probier auch mal nord oder turm aus.""" if which not in MENSA_NAME: await BOT.say("Diese Mensa gibt es nicht! 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() await BOT.say("Das Essen für %s in der %s" % (data["date"], MENSA_NAME[which])) message = "" for meal in data["meals"]: 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: await BOT.say(message) message = meal_line else: message += meal_line await BOT.say(message) @BOT.command() async def magie(): """MAGIEEEEEE""" rand = random.SystemRandom() await BOT.say(STOLL[rand.randrange(0, len(STOLL), 1)]) @BOT.command() async def manta(): """Boah ey, geile Karre!""" rand = random.SystemRandom() await BOT.say(MANTA[rand.randrange(0, len(MANTA), 1)]) def main(): """ entry point """ token = open("token").read() BOT.run(token.strip()) if __name__ == '__main__': main()