Path: blob/master/bot/modules/gd_count.py
1624 views
from ..helper.ext_utils.bot_utils import sync_to_async, new_task1from ..helper.ext_utils.links_utils import is_gdrive_link2from ..helper.ext_utils.status_utils import get_readable_file_size3from ..helper.mirror_leech_utils.gdrive_utils.count import GoogleDriveCount4from ..helper.telegram_helper.message_utils import delete_message, send_message567@new_task8async def count_node(_, message):9args = message.text.split()10user = message.from_user or message.sender_chat11if username := user.username:12tag = f"@{username}"13else:14tag = message.from_user.mention1516link = args[1] if len(args) > 1 else ""17if len(link) == 0 and (reply_to := message.reply_to_message):18link = reply_to.text.split(maxsplit=1)[0].strip()1920if is_gdrive_link(link):21msg = await send_message(message, f"Counting: <code>{link}</code>")22name, mime_type, size, files, folders = await sync_to_async(23GoogleDriveCount().count, link, user.id24)25if mime_type is None:26await send_message(message, name)27return28await delete_message(msg)29msg = f"<b>Name: </b><code>{name}</code>"30msg += f"\n\n<b>Size: </b>{get_readable_file_size(size)}"31msg += f"\n\n<b>Type: </b>{mime_type}"32if mime_type == "Folder":33msg += f"\n<b>SubFolders: </b>{folders}"34msg += f"\n<b>Files: </b>{files}"35msg += f"\n\n<b>cc: </b>{tag}"36else:37msg = (38"Send Gdrive link along with command or by replying to the link by command"39)4041await send_message(message, msg)424344454647