Spaces:
Build error
Build error
File size: 5,883 Bytes
071487b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 |
# ===================================================================== #
# Copyright (c) 2022 Itz-fork #
# #
# This program is distributed in the hope that it will be useful, #
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. #
# See the GNU General Public License for more details. #
# #
# You should have received a copy of the GNU General Public License #
# along with this program. If not, see <http://www.gnu.org/licenses/> #
# ===================================================================== #
from re import match
from time import time
from os import path, remove
from config import Config
from pyrogram import filters
from aiohttp import ClientSession
from pyrogram.types import Message
from unzipper import unzip_client, Buttons
from pyrogram.errors import ReplyMarkupTooLong
from unzipper.lib.extractor import Extractor
from unzipper.lib.downloader import Downloader, dl_regex
from unzipper.helpers_nexa.utils import (TimeFormatter, get_files,
progress_for_pyrogram, humanbytes)
from unzipper.database.split_arc import del_split_arc_user, get_split_arc_user
@unzip_client.on_message(filters.incoming & filters.private & filters.regex(dl_regex) | filters.document)
@unzip_client.handle_erros
async def extract_dis_archive(_, message: Message, texts):
unzip_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
user_id = message.from_user.id
download_path = f"{Config.DOWNLOAD_LOCATION}/{user_id}"
is_url = message.text and (match(dl_regex, message.text))
is_doc = message.document
# Splitted files
is_spl, lfn, ps = await get_split_arc_user(user_id)
if is_spl:
file_name = is_doc.file_name if is_doc else path.basename(message.text)
await unzip_msg.edit(texts["alert_downloading_part"])
# Check file extension
if not path.splitext(file_name)[1].replace(".", "").isnumeric():
return await unzip_msg.edit(texts["no_splitted_arc"])
arc_name = f"{download_path}/archive_from_{user_id}_{file_name}"
if path.isfile(arc_name):
return await unzip_msg.edit(texts["alert_part_exists"])
# Download the file
s_time = time()
if is_url:
async with ClientSession() as ses:
cleng = (await ses.head(message.text)).headers.get("Content-Length")
fsize = humanbytes(cleng) if cleng else "undefined"
# Send logs
await unzip_client.send_message(Config.LOGS_CHANNEL, texts["log"].format(
user_id,
"N/A",
"N/A",
file_name,
fsize)
)
await Downloader().download(message.text, arc_name, unzip_msg)
else:
# Send logs
rchat = message.forward_from_chat
await unzip_client.send_message(Config.LOGS_CHANNEL, texts["log"].format(
user_id,
rchat.title if rchat else "N/A",
rchat.id if rchat else "N/A",
file_name,
humanbytes(is_doc.file_size))
)
await message.download(
file_name=arc_name,
progress=progress_for_pyrogram, progress_args=(
"**Trying to Download!** \n", unzip_msg, s_time)
)
e_time = time()
await unzip_msg.edit(texts["ok_download"].format(file_name, TimeFormatter(round(e_time-s_time) * 1000)))
return
if path.isdir(download_path):
return await unzip_msg.edit(texts["alert_process_running_already"])
if is_url:
await unzip_msg.edit(texts["ask_what_you_want"], reply_markup=Buttons.EXTRACT_URL)
elif is_doc:
await unzip_msg.edit(texts["ask_what_you_want"], reply_markup=Buttons.EXTRACT_FILE)
else:
await unzip_msg.edit(texts["no_archive"])
@unzip_client.on_message(filters.private & filters.command("done"))
@unzip_client.handle_erros
async def extracted_dis_spl_archive(_, message: Message, texts):
spl_umsg = await message.reply(texts["processing"], reply_to_message_id=message.id)
user_id = message.from_user.id
# Retrive data from database
is_spl, lfn, ps = await get_split_arc_user(user_id)
# Path checks
if not is_spl:
return await spl_umsg.edit("`Bruh, why are you sending this command 🤔?`")
ext_path = f"{Config.DOWNLOAD_LOCATION}/{user_id}/extracted"
arc_path = path.dirname(lfn)
if not path.isdir(arc_path):
await spl_umsg.edit(texts["alert_empty_files"])
return await del_split_arc_user(user_id)
# Remove user record from the database
await del_split_arc_user(user_id)
# Extract the archive
ext = Extractor()
s_time = time()
await ext.extract(lfn, ext_path, ps, True)
extdarc = f"{ext_path}/{path.splitext(path.basename(lfn))[0]}"
await ext.extract(extdarc, ext_path, ps)
e_time = time()
await spl_umsg.edit(texts["ok_extract"].format(TimeFormatter(round(e_time-s_time) * 1000)))
# Try to remove merged archive
try:
remove(extdarc)
except:
pass
paths = await get_files(ext_path)
i_e_btns = await Buttons.make_files_keyboard(paths, user_id, message.chat.id)
try:
await spl_umsg.edit(texts["select_files"], reply_markup=i_e_btns)
except ReplyMarkupTooLong:
i_e_btns = await Buttons.make_files_keyboard(paths, user_id, message.chat.id, False)
await spl_umsg.edit(texts["select_files"], reply_markup=i_e_btns)
|