44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
import asyncio
|
|
|
|
from core import command
|
|
from loguru import logger
|
|
from pyrogram import Client
|
|
from pyrogram.errors import FloodWait
|
|
from pyrogram.types import Message
|
|
from tools.googles import google_search
|
|
from tools.helpers import Parameters
|
|
|
|
|
|
@Client.on_message(command("google"))
|
|
async def google(_: Client, msg: Message):
|
|
"""谷歌搜索并展示第一页结果和链接"""
|
|
replied_msg = msg.reply_to_message
|
|
if not replied_msg:
|
|
_, args = Parameters.get(msg)
|
|
else:
|
|
args = replied_msg.text or replied_msg.caption
|
|
|
|
try:
|
|
res = await google_search(args)
|
|
content = '\n\n'.join(
|
|
f"[{title[0:30]}]({url})" for title, url in res.items()
|
|
)
|
|
text = f"🔎 | **Google** | `{args}`\n{content}"
|
|
await msg.edit_text(
|
|
text=text,
|
|
parse_mode='md',
|
|
disable_web_page_preview=True
|
|
)
|
|
except FloodWait as e:
|
|
await asyncio.sleep(e.x)
|
|
await msg.edit_text(
|
|
text=text,
|
|
parse_mode='md',
|
|
disable_web_page_preview=True
|
|
)
|
|
except Exception as e:
|
|
logger.error(e)
|
|
await msg.edit_text("❗️ Unable to connect to google")
|
|
finally:
|
|
await logger.complete()
|