mirror of
https://github.com/Findus23/nonsense.git
synced 2024-09-19 16:03:50 +02:00
175 lines
5.3 KiB
Python
175 lines
5.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Simple Bot to reply to Telegram messages
|
|
# This program is dedicated to the public domain under the CC0 license.
|
|
import datetime
|
|
import logging
|
|
from uuid import uuid4
|
|
import yaml
|
|
from telegram import InlineQueryResultArticle
|
|
from telegram import InputTextMessageContent
|
|
from telegram.ext import InlineQueryHandler
|
|
from telegram.ext import Job
|
|
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
|
|
|
|
import config
|
|
import generate
|
|
|
|
subscriptions = dict()
|
|
|
|
# Enable logging
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.DEBUG)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# 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 help(bot, update):
|
|
update.message.reply_text("Hier steht der Hilfetext!")
|
|
|
|
|
|
def single(bot, update):
|
|
update.message.reply_text(generate.get_description())
|
|
|
|
|
|
def subscribe_notification(bot, job):
|
|
bot.sendMessage(job.context, text=generate.get_description())
|
|
|
|
|
|
def time_till_first_run():
|
|
now = datetime.datetime.now()
|
|
nextsend = datetime.datetime(now.year, now.month, now.day, now.hour, now.minute, 0) + datetime.timedelta(minutes=1)
|
|
time_till_next_send = nextsend - now
|
|
return time_till_next_send.total_seconds()
|
|
|
|
|
|
def subscribe(bot, update, job_queue):
|
|
"""Adds a job to the queue"""
|
|
chat_id = update.message.chat_id
|
|
# Add job to queue
|
|
if chat_id in subscriptions:
|
|
update.message.reply_text('You are already subscribed')
|
|
return
|
|
|
|
job = Job(subscribe_notification, 60, repeat=True, context=chat_id)
|
|
subscriptions[chat_id] = job
|
|
|
|
job_queue.put(job, next_t=time_till_first_run())
|
|
update.message.reply_text('Successfully subscribed')
|
|
|
|
|
|
def unsubscribe(bot, update):
|
|
"""Deletes a job"""
|
|
chat_id = update.message.chat_id
|
|
|
|
if chat_id not in subscriptions:
|
|
update.message.reply_text('You have no subscription')
|
|
return
|
|
|
|
# Add job to queue
|
|
job = subscriptions[chat_id]
|
|
job.schedule_removal()
|
|
del subscriptions[chat_id]
|
|
update.message.reply_text('Successfully unsubscribed')
|
|
|
|
|
|
def multiple(bot, update, args):
|
|
try:
|
|
descriptions = []
|
|
print(args)
|
|
count = int(args[0])
|
|
if count > 50:
|
|
update.message.reply_text(str(count) + ' > 50')
|
|
return
|
|
for _ in range(count):
|
|
descriptions.append("+++ " + generate.get_description() + " +++")
|
|
update.message.reply_text("\n".join(descriptions))
|
|
except (IndexError, ValueError):
|
|
update.message.reply_text('Usage: /multiple <count>')
|
|
|
|
|
|
def error(bot, update, error):
|
|
logger.warning('Update "%s" caused error "%s"' % (update, error))
|
|
|
|
|
|
def inlinequery(bot, update):
|
|
if update.inline_query.query == "":
|
|
count = 3
|
|
else:
|
|
count = int(update.inline_query.query)
|
|
results = list()
|
|
if count > 50:
|
|
count = 50
|
|
for i in range(count):
|
|
description = generate.get_description()
|
|
results.append(InlineQueryResultArticle(id=uuid4(),
|
|
title=description,
|
|
input_message_content=InputTextMessageContent(description)))
|
|
update.inline_query.answer(results)
|
|
|
|
|
|
def startup(job_queue):
|
|
with open("save.yaml") as json_file:
|
|
save = yaml.load(json_file)
|
|
for s in save["subscriptions"]:
|
|
job = Job(subscribe_notification, 60, repeat=True, context=s)
|
|
subscriptions[s] = job
|
|
job_queue.put(job, next_t=time_till_first_run())
|
|
|
|
|
|
def shutdown(a=None, b=None):
|
|
print("------------------------------------------------------------------------") #
|
|
save = {
|
|
"subscriptions": []
|
|
}
|
|
for s in subscriptions:
|
|
save["subscriptions"].append(s)
|
|
with open("save.yaml", 'w') as stream:
|
|
yaml.dump(save, stream, default_flow_style=False)
|
|
|
|
|
|
def main():
|
|
# Create the EventHandler and pass it your bot's token.
|
|
updater = Updater(config.telegram_bot_token)
|
|
|
|
# Get the dispatcher to register handlers
|
|
dp = updater.dispatcher
|
|
|
|
# on different commands - answer in Telegram
|
|
dp.add_handler(CommandHandler("start", help))
|
|
dp.add_handler(CommandHandler("help", help))
|
|
dp.add_handler(CommandHandler("single", single))
|
|
dp.add_handler(CommandHandler("multiple", multiple, pass_args=True))
|
|
# dp.add_handler(CommandHandler("single", subscribe_notification))
|
|
dp.add_handler(CommandHandler("subscribe", subscribe, pass_job_queue=True))
|
|
dp.add_handler(CommandHandler("unsubscribe", unsubscribe))
|
|
|
|
dp.add_handler(CommandHandler("save", shutdown))
|
|
|
|
# on noncommand i.e message - echo the message on Telegram
|
|
# dp.add_handler(MessageHandler(Filters.text, single))
|
|
|
|
dp.add_handler(InlineQueryHandler(inlinequery))
|
|
|
|
# log all errors
|
|
dp.add_error_handler(error)
|
|
|
|
# Start the Bot
|
|
updater.start_polling()
|
|
|
|
startup(updater.job_queue)
|
|
|
|
# Run the bot until the you presses 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()
|
|
|
|
shutdown()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|