26 lines
918 B
Python
26 lines
918 B
Python
from telegram.ext import CommandHandler, run_async
|
|
from bot.helper.mirror_utils.gdriveTools import GoogleDriveHelper
|
|
from bot import LOGGER, dispatcher
|
|
from bot.helper.telegram_helper.message_utils import auto_delete_message, sendMessage
|
|
import threading
|
|
from bot.helper.telegram_helper.bot_commands import BotCommands
|
|
|
|
@run_async
|
|
def list_drive(update, context):
|
|
message = update.message.text
|
|
search = message.replace('/list ', '')
|
|
LOGGER.info("Searching: " + search)
|
|
gdrive = GoogleDriveHelper(None)
|
|
msg = gdrive.drive_list(search)
|
|
if msg:
|
|
reply_message = sendMessage(msg, context, update)
|
|
|
|
else:
|
|
reply_message = sendMessage('No result found', context, update)
|
|
|
|
threading.Thread(target=auto_delete_message, args=(context, update.message, reply_message)).start()
|
|
|
|
|
|
list_handler = CommandHandler(BotCommands.ListCommand, list_drive)
|
|
dispatcher.add_handler(list_handler)
|