CursedImagesBot/bot.py

42 lines
1.4 KiB
Python

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