async def down_load_media_f(client, message):
user_command = message.command[0]
user_id = message.from_user.id
LOGGER.info(user_id)
mess_age = await message.reply_text(
"Is it really a Telegram Media 🤔!\nJust to make sure 🤷♂️ \nIf yes, then wait a few minutes",
quote=True)
if not os.path.isdir(DOWNLOAD_LOCATION):
os.makedirs(DOWNLOAD_LOCATION)
if message.reply_to_message is not None:
start_t = datetime.now()
download_location = str(Path().resolve()) + "/"
c_time = time.time()
prog = Progress(user_id, client, mess_age)
try:
the_real_download_location = await client.download_media(
message=message.reply_to_message,
file_name=download_location,
progress=prog.progress_for_pyrogram,
progress_args=("trying to download", c_time),
)
except Exception as g_e:
await mess_age.edit(str(g_e))
LOGGER.error(g_e)
return
end_t = datetime.now()
ms = (end_t - start_t).seconds
LOGGER.info(the_real_download_location)
await asyncio.sleep(10)
if the_real_download_location:
await mess_age.edit_text(
f"Downloaded to <code>{the_real_download_location}</code> in <u>{ms}</u> seconds"
)
else:
await mess_age.edit_text(
"😔 Download Cancelled or some error happened")
return
the_real_download_location_g = the_real_download_location
if user_command == TELEGRAM_LEECH_UNZIP_COMMAND.lower():
try:
check_ifi_file = get_base_name(the_real_download_location)
file_up = await unzip_me(the_real_download_location)
if os.path.exists(check_ifi_file):
the_real_download_location_g = file_up
except Exception as ge:
LOGGER.info(ge)
LOGGER.info(
f"Unable to extract {os.path.basename(the_real_download_location)}, Uploading the same file"
)
await upload_to_gdrive(the_real_download_location_g, mess_age, message,
user_id)
else:
await mess_age.edit_text(
"Reply to a Telegram Media, to upload to the Cloud Drive.")
async def down_load_media_f(client, message): # to be removed
user_command = message.command[0]
user_id = message.from_user.id
if message.reply_to_message is not None:
the_real_download_location, mess_age = await download_tg(client, message)
the_real_download_location_g = the_real_download_location
if user_command == TELEGRAM_LEECH_UNZIP_COMMAND.lower():
try:
check_ifi_file = get_base_name(the_real_download_location)
file_up = await unzip_me(the_real_download_location)
if os.path.exists(check_ifi_file):
the_real_download_location_g = file_up
except Exception as ge:
LOGGER.info(ge)
LOGGER.info(
f"Can't extract {os.path.basename(the_real_download_location)}, Uploading the same file"
)
await upload_to_gdrive(the_real_download_location_g, mess_age, message, user_id)
else:
await mess_age.edit_text(
"Reply to a Telegram Media, to upload to the Cloud Drive."
)