forust
20bce5b31c
- Add missing aiohttp, PIL, BytesIO imports to safone.py (F821 runtime bugs) - Apply ruff --unsafe-fixes (27 fixes): ternary operators, .get() patterns, contextlib.suppress, enumerate(), collapsible if/else, remove .keys()
111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
import random
|
|
import re
|
|
from io import BytesIO
|
|
|
|
import requests
|
|
from bs4 import BeautifulSoup as bs
|
|
from pyrogram import Client, filters
|
|
from pyrogram.errors import RPCError
|
|
from pyrogram.types import InputMediaPhoto, Message
|
|
from utils.misc import modules_help, prefix
|
|
|
|
|
|
@Client.on_message(filters.command('icon', prefix) & filters.me)
|
|
async def search_icon(_, message: Message):
|
|
if len(message.command) != 2:
|
|
return await message.edit_text('Please provide some text to search icons from Flaticon.com.')
|
|
query = message.text.split(maxsplit=1)[1]
|
|
|
|
await message.edit_text('Searching for icons...')
|
|
search_query = query.replace(' ', '%20')
|
|
url = f'https://www.flaticon.com/search?word={search_query}'
|
|
|
|
try:
|
|
html_content = requests.get(url).text
|
|
soup = bs(html_content, 'html.parser')
|
|
results = soup.find_all(
|
|
'img',
|
|
src=re.compile(r'https://cdn-icons-png.flaticon.com/128/[0-9]+/[0-9]+.png'),
|
|
)
|
|
|
|
if not results:
|
|
return await message.edit('No results found.')
|
|
|
|
random.shuffle(results)
|
|
icons = []
|
|
for i in range(5):
|
|
icons.append(results[i]['src'].replace('128', '512'))
|
|
|
|
for icon in icons:
|
|
await message.reply_document(icon)
|
|
|
|
return await message.delete()
|
|
except Exception as e:
|
|
await message.edit(f'An error occurred: {e}')
|
|
print(f'Error: {e}')
|
|
|
|
|
|
@Client.on_message(filters.command('freepik', prefix) & filters.me)
|
|
async def freepik_search(client: Client, message: Message):
|
|
parts = message.text.split(' ', 1)
|
|
if len(parts) < 2:
|
|
await message.edit_text('Please provide a search query!')
|
|
return
|
|
|
|
query = parts[1]
|
|
limit = 5
|
|
if ' ; ' in query:
|
|
match, limit_str = query.split(' ; ', 1)
|
|
try:
|
|
limit = int(limit_str)
|
|
except ValueError:
|
|
await message.edit_text('Invalid limit! Using the default value of 5.')
|
|
else:
|
|
match = query
|
|
|
|
match = match.replace(' ', '%20')
|
|
await message.edit_text('Searching Freepik...')
|
|
|
|
try:
|
|
url = f'https://www.freepik.com/api/regular/search?locale=en&term={match}'
|
|
json_content = requests.get(url).json()
|
|
results = []
|
|
for i in json_content['items']:
|
|
results.append(i['preview']['url'])
|
|
|
|
if results is None:
|
|
return await message.edit_text('No results found.')
|
|
|
|
random.shuffle(results)
|
|
img_urls = results[:limit]
|
|
|
|
media_group = []
|
|
for img_url in img_urls:
|
|
icon = requests.get(img_url)
|
|
if icon.status_code == 200:
|
|
media_group.append(InputMediaPhoto(media=BytesIO(icon.content)))
|
|
|
|
if not media_group:
|
|
await message.edit_text('No images could be downloaded.')
|
|
return
|
|
|
|
try:
|
|
await client.send_media_group(chat_id=message.chat.id, media=media_group)
|
|
except RPCError:
|
|
await message.edit_text('Failed to send some images. Retrying individually...')
|
|
for media in media_group:
|
|
try:
|
|
await message.reply_photo(photo=media.media)
|
|
except Exception as e:
|
|
await message.edit_text(f'Error sending image: {e}')
|
|
|
|
except Exception as e:
|
|
await message.edit_text(f'Failed to fetch data: {e}')
|
|
print(f'Error: {e}')
|
|
|
|
|
|
modules_help['icons'] = {
|
|
'icon [query]': 'Search for icons on Flaticon.',
|
|
'freepik [query] [limit]': 'Search for images on Freepik. Limit is optional and defaults to 5.',
|
|
}
|