23 lines
1002 B
Python
23 lines
1002 B
Python
from telegram.ext import CommandHandler, run_async
|
|
from bot.helper.mirror_utils.upload_utils.gdriveTools import GoogleDriveHelper
|
|
from bot.helper.telegram_helper.message_utils import deleteMessage, sendMessage
|
|
from bot.helper.telegram_helper.filters import CustomFilters
|
|
from bot.helper.telegram_helper.bot_commands import BotCommands
|
|
from bot import dispatcher
|
|
|
|
@run_async
|
|
def countNode(update,context):
|
|
args = update.message.text.split(" ",maxsplit=1)
|
|
if len(args) > 1:
|
|
link = args[1]
|
|
msg = sendMessage(f"Counting: <code>{link}</code>",context.bot,update)
|
|
gd = GoogleDriveHelper()
|
|
result = gd.count(link)
|
|
deleteMessage(context.bot,msg)
|
|
sendMessage(result,context.bot,update)
|
|
else:
|
|
sendMessage("Provide G-Drive Shareable Link to Count.",context.bot,update)
|
|
|
|
count_handler = CommandHandler(BotCommands.CountCommand,countNode,filters=CustomFilters.authorized_chat | CustomFilters.authorized_user)
|
|
dispatcher.add_handler(count_handler)
|