Add bot.py
This commit is contained in:
parent
7e51aa02d0
commit
5873ffabe6
|
@ -0,0 +1,36 @@
|
||||||
|
#!/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):
|
||||||
|
try:
|
||||||
|
lim = args[0]
|
||||||
|
except(IndexError):
|
||||||
|
lim = 1
|
||||||
|
url = f"https://www.reddit.com/r/cursedimages/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('cursed', cursed, pass_args=True))
|
||||||
|
updater.start_polling()
|
Loading…
Reference in New Issue