|
|
#! /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()
|