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.

237 lines
7.2 KiB

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