39 lines
1.8 KiB
Python
39 lines
1.8 KiB
Python
from pyrogram import Client, filters
|
|
from pyrogram.errors.exceptions.forbidden_403 import Forbidden
|
|
from pyrogram.errors.exceptions.bad_request_400 import ChatSendInlineForbidden
|
|
from .. import slave, config, help_dict, log_errors, public_log_errors
|
|
|
|
@Client.on_message(~filters.scheduled & ~filters.forwarded & ~filters.sticker & ~filters.via_bot & ~filters.edited & filters.me & filters.command(['w', 'wiki', 'wikipedia'], prefixes=config['config']['prefixes']))
|
|
@log_errors
|
|
@public_log_errors
|
|
async def wikipedia(client, message):
|
|
bot = await slave.get_me()
|
|
query = message.command
|
|
page = 1
|
|
query.pop(0)
|
|
if len(query) > 1 and query[0].isnumeric():
|
|
page = int(query.pop(0))
|
|
page -= 1
|
|
if page < 0:
|
|
page = 0
|
|
elif page > 9:
|
|
page = 9
|
|
query = ' '.join(query)
|
|
if not query:
|
|
return
|
|
results = await client.get_inline_bot_results(bot.username or bot.id, 'w ' + query)
|
|
if not results.results:
|
|
await message.reply_text('There are no results')
|
|
return
|
|
try:
|
|
await message.reply_inline_bot_result(results.query_id, results.results[page].id)
|
|
except IndexError:
|
|
await message.reply_text(f'There are only {len(results.results)} results')
|
|
except (Forbidden, ChatInlineSendForbidden):
|
|
await message.reply_text({'message': results.results[page].send_message.message, 'entities': results.results[page].send_message.entities}, disable_web_page_preview=True, parse_mode='through')
|
|
|
|
help_dict['wikipedia'] = ('Wikipedia',
|
|
'''{prefix}wikipedia <i>[page]</i> <i><query></i> - Searches for <i><query></i> on Wikipedia
|
|
Aliases: {prefix}w, {prefix}wiki
|
|
Can also be activated inline with: @{bot} wikipedia <i><query></i> or @{bot} wiki <i><query></i> or @{bot} w <i><query></i>''')
|