CustomModules/webshot.py at main · FoxUserbot/CustomModules · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from pyrogram import Client, filters
from command import fox_command, fox_sudo, who_message
from requirements_installer import install_library
import os

install_library("requests")
import requests
import io

@Client.on_message(fox_command("webshot", "Webshot", os.path.basename(__file__), "[url]") & fox_sudo())
async def webshot(client, message):
    message = await who_message(client, message)
    try:
        user_link = (message.command[1].replace("https://", "").replace("http://", ""))
        await message.edit("Try create screenshot...")
        full_link = f"https://mini.s-shot.ru/1920x1080/JPEG/1024/Z100/?{user_link}"
        try:
            response = requests.get(full_link)
            if response.status_code == 200:
                response.raise_for_status()
                image_bytes = io.BytesIO(response.content)
                image_bytes.name = "webshot.jpg"
                image_bytes.seek(0)
                await client.send_photo(message.chat.id, image_bytes, caption=f"**Screenshot of the page ⟶** {user_link}")
            else:
                await message.edit(f"**Error:** {response.status_code}")
        except requests.exceptions.RequestException as e:
            await message.edit(f"**Error:** {e}")
            return
        except Exception as error:
            await message.edit(f"**Error:** {error}")
            return
    except IndexError:
        await message.edit("Don't have link!")
    except Exception as error:
        await message.edit(f"**Error:** {error}")