#!/usr/bin/env python3

from telegram.ext import Updater, CommandHandler
import config
import requests
import json
import logging
from PIL import Image
from io import BytesIO

logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - \
                    %(message)s', level=logging.INFO)


def cursed(bot, update, args, subreddit):
    try:
        lim = args[0]
    except(IndexError):
        lim = 1
    url = f"https://www.reddit.com/r/{subreddit}/top.json?limit={lim}"
    headers = {
            'user-agent': ('windows / chrome 67 [desktop]: mozilla/5.0 '
                           '(windows nt 10.0; win64; x64)')}
    res = json.loads(requests.get(url, headers=headers).text)
    image_url = res["data"]["children"][-1]["data"]["url"]
    image = requests.get(image_url, headers=headers)
    Image.open(BytesIO(image.content)).save('test.webp')
    chat_id = update.message.chat_id
    bot.send_sticker(chat_id, open('test.webp', 'rb'))


updater = Updater(token=config.api_key)
dispatcher = updater.dispatcher

dispatcher.add_handler(CommandHandler('hmmm', lambda x, y, args:
                                      cursed(x, y, args, "hmmm"),
                                      pass_args=True))
dispatcher.add_handler(CommandHandler('cursed', lambda x, y, args:
                                      cursed(x, y, args, "CursedImages"),
                                      pass_args=True))
updater.start_polling()