70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
|
# Copyright (C) 2021 Veez Project
|
||
|
|
|
||
|
|
from pyrogram import Client, errors
|
||
|
|
from pyrogram.types import (
|
||
|
|
InlineQuery,
|
||
|
|
InlineQueryResultArticle,
|
||
|
|
InputTextMessageContent,
|
||
|
|
)
|
||
|
|
from youtubesearchpython import VideosSearch
|
||
|
|
|
||
|
|
|
||
|
|
@Client.on_inline_query()
|
||
|
|
async def inline(client: Client, query: InlineQuery):
|
||
|
|
answers = []
|
||
|
|
search_query = query.query.lower().strip().rstrip()
|
||
|
|
|
||
|
|
if search_query == "menu":
|
||
|
|
await client.answer_inline_query(
|
||
|
|
query.id,
|
||
|
|
results=menus,
|
||
|
|
switch_pm_text="Menu",
|
||
|
|
switch_pm_parameter="help",
|
||
|
|
cache_time=0,
|
||
|
|
)
|
||
|
|
if search_query == "":
|
||
|
|
await client.answer_inline_query(
|
||
|
|
query.id,
|
||
|
|
results=answers,
|
||
|
|
switch_pm_text="search a youtube video",
|
||
|
|
switch_pm_parameter="help",
|
||
|
|
cache_time=0,
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
search = VideosSearch(search_query, limit=50)
|
||
|
|
|
||
|
|
for result in search.result()["result"]:
|
||
|
|
answers.append(
|
||
|
|
InlineQueryResultArticle(
|
||
|
|
title=result["title"],
|
||
|
|
description="{}, {}.".format(
|
||
|
|
result["duration"], result["viewCount"]["short"]
|
||
|
|
),
|
||
|
|
input_message_content=InputTextMessageContent(
|
||
|
|
"/vstream https://www.youtube.com/watch?v={}".format(result["id"])
|
||
|
|
),
|
||
|
|
thumb_url=result["thumbnails"][0]["url"],
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
try:
|
||
|
|
await query.answer(results=answers, cache_time=0)
|
||
|
|
except errors.QueryIdInvalid:
|
||
|
|
await query.answer(
|
||
|
|
results=answers,
|
||
|
|
cache_time=0,
|
||
|
|
switch_pm_text="Error: search timed out",
|
||
|
|
switch_pm_parameter="",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
# ==================
|
||
|
|
# Tested
|
||
|
|
|
||
|
|
menus = [
|
||
|
|
InlineQueryResultArticle(title="Start", description="Start a bot",
|
||
|
|
input_message_content=InputTextMessageContent("/start")),
|
||
|
|
InlineQueryResultArticle(title="Info Bot", description="Info about this bot",
|
||
|
|
input_message_content=InputTextMessageContent("/info")),
|
||
|
|
]
|