TelegramBotAdmin/wbb/modules/images.py

import urllib.request
import json
import asyncio
from random import randint
from pyrogram import filters
from wbb import app, arq
from wbb.utils.errors import capture_err

__MODULE__ = "Images"
__HELP__ = '''/cat  - Get Cute Cats Images
/wall - Get Wallpapers'''


async def delete_message_with_delay(delay, message):
    await asyncio.sleep(delay)
    await message.delete()


@app.on_message(filters.command("cat") & ~filters.edited)
@capture_err
async def cat(_, message):
    with urllib.request.urlopen(
            "https://api.thecatapi.com/v1/images/search"
    ) as url:
        data = json.loads(url.read().decode())
    cat_url = (data[0]['url'])
    await message.reply_photo(cat_url)


@app.on_message(filters.command("wall") & ~filters.edited)
@capture_err
async def wall(_, message):
    if len(message.command) < 2:
        await message.reply_text("/wall needs an argument")
        return
    initial_term = message.text.split(None, 1)[1]
    m = await message.reply_text("Searching!")
    term = initial_term.replace(' ', '%20')
    try:
        wallpapers = await arq.wall(term)
    except KeyError:
        await m.edit("Found literally nothing!,"
                     + "You should work on your English.")
        return
    if len(wallpapers) > 10:
        selection = 10
    else:
        selection = len(wallpapers)
    index = randint(0, selection - 1)
    wallpaper = wallpapers[index].url_image
    await message.reply_document(wallpaper)
    await m.delete()