29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
from telegram.ext import CommandHandler
|
|
from bot.helper.mirror_utils.upload_utils.gdriveTools import GoogleDriveHelper
|
|
from bot import LOGGER, dispatcher
|
|
from bot.helper.telegram_helper.message_utils import sendMessage, sendMarkup, editMessage
|
|
from bot.helper.telegram_helper.filters import CustomFilters
|
|
import threading
|
|
from bot.helper.telegram_helper.bot_commands import BotCommands
|
|
|
|
|
|
def list_drive(update, context):
|
|
try:
|
|
search = update.message.text.split(' ',maxsplit=1)[1]
|
|
LOGGER.info(f"Searching: {search}")
|
|
reply = sendMessage('Searching..... Please wait!', context.bot, update)
|
|
gdrive = GoogleDriveHelper(None)
|
|
msg, button = gdrive.drive_list(search)
|
|
|
|
if button:
|
|
editMessage(msg, reply, button)
|
|
else:
|
|
editMessage(f'No result found for <b>{search}</b>', reply, button)
|
|
|
|
except IndexError:
|
|
sendMessage('Send a search key along with command', context.bot, update)
|
|
|
|
|
|
list_handler = CommandHandler(BotCommands.ListCommand, list_drive, filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
|
|
dispatcher.add_handler(list_handler)
|