From 700593f889109e668fad151813ef41679be8db55 Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Sat, 11 May 2019 19:52:21 +0200
Subject: [PATCH] Rewrite in Telethon

---
 bot.py | 80 +++++++++++++++-------------------------------------------
 1 file changed, 21 insertions(+), 59 deletions(-)

diff --git a/bot.py b/bot.py
index 440775f..be17666 100644
--- a/bot.py
+++ b/bot.py
@@ -1,21 +1,13 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-
-from uuid import uuid4
 from os import environ
 from collections import namedtuple
 from json import loads
 
 import logging
 
-from telegram import InlineQueryResultArticle
-from telegram import InputTextMessageContent
-from telegram import ParseMode
-from telegram.ext import CommandHandler
-from telegram.ext import InlineQueryHandler
-from telegram.ext import Updater
-
 import requests
+from telethon import TelegramClient, events
 from bs4 import BeautifulSoup
 
 # Enable logging
@@ -32,6 +24,8 @@ MSG_FMT_STR = "[{number}]({link}): *{title}*\n\n_{alt}_"
 XKCD_JSON_FMT_STR = "https://xkcd.com/{}/info.0.json"
 MAX_SEARCH_RESULTS = 10
 
+bot = TelegramClient('xkcd', 6, 'eb06d4abfb49dc3eeb1aeb98ae0f581e')
+
 # blockquote element -> Xkcd
 def parse_blockquote(elem):
     children = list(elem.children)
@@ -66,68 +60,36 @@ def get_xkcds(text):
 
 # 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):
+@bot.on(events.NewMessage(pattern='/start$'))
+async def start(event):
     """Send a message when the command /start is issued."""
     # TODO
-    update.message.reply_text('Hi!')
+    await event.respond('Hi!')
 
 
-def help(bot, update):
+@bot.on(events.NewMessage(pattern='/help$'))
+async def help(event):
     """Send a message when the command /help is issued."""
     # TODO
-    update.message.reply_text('Help!')
+    await event.respond('Help!')
 
-def inlinequery(bot, update):
+
+@bot.on(events.InlineQuery)
+async def inlinequery(event):
     """Handle the inline query."""
     # TODO show transcript in result but not message?
-    query = update.inline_query.query
-    results = []
-    for xkcd in get_xkcds(query):
-        number = xkcd.number
-        link = xkcd.link
-        title = xkcd.title
-        alt = xkcd.alt
-
-        results.append(InlineQueryResultArticle(
-            id=uuid4(),
-            title=xkcd.title,
-            url=xkcd.link,
-            input_message_content=InputTextMessageContent(
-                MSG_FMT_STR.format(number=number, link=link, title=title, alt=alt),
-                parse_mode=ParseMode.MARKDOWN)))
-
-    update.inline_query.answer(results)
-
-
-def error(bot, update, error):
-    """Log Errors caused by Updates."""
-    logger.warning('Update "%s" caused error "%s"', update, error)
+    builder = event.builder
+    await event.answer([builder.article(
+        title=xkcd.title,
+        url=xkcd.link,
+        text=MSG_FMT_STR.format(number=xkcd.number, link=xkcd.link, title=xkcd.title, alt=xkcd.alt)
+    ) for xkcd in get_xkcds(event.text)])
 
 
 def main():
-    # Create the Updater and pass it your bot's token.
-    updater = Updater(environ["TOKEN"])
-
-    # 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))
-
-    # on noncommand i.e message - echo the message on Telegram
-    dp.add_handler(InlineQueryHandler(inlinequery))
-
-    # log all errors
-    dp.add_error_handler(error)
-
-    # Start the Bot
-    updater.start_polling()
-
-    # Block until the user 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()
+    bot.start(bot_token=environ['TOKEN'])
+    with bot:
+        bot.run_until_disconnected()
 
 
 if __name__ == '__main__':