forust
227e5fda27
- S113: Add timeout=10 to all requests calls (74 fixes) - E722: Replace bare except: with except Exception: - B904: Replace redundant re-raise with bare raise - E402: Add noqa for intentional late imports after import_library() - S102/S307/S310/S311/S603/S605/S606/S607/S108: Add noqa for intentional usage - F601: Fix duplicate dict key in unsplash.py - N802: Rename ReplyCheck -> reply_check with backward compat alias - N813: Rename bs -> BS in icons.py - B007/B020: Rename loop var _j in animations.py - SIM102: Collapse nested if in autofwd.py - SIM113: Use enumerate() in calculator.py - A002: Add noqa for builtin shadowing in admlist.py - F811: Add noqa for cohere redefinition - edu_master: Fix ARG001, S108, S110, SIM117, apply --unsafe-fixes - Add modules_list.txt with full module inventory
111 lines
3.6 KiB
Python
111 lines
3.6 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, timeout=10).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, timeout=10).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, timeout=10)
|
|
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.',
|
|
}
|