' \
+ f'
Drive Link'
if INDEX_URLS[index] is not None:
if isRecur:
url_path = "/".join([rquote(n, safe='') for n in self.__get_recursive_list(file, parent_id)])
else:
url_path = rquote(f'{file.get("name")}', safe='')
url = f'{INDEX_URLS[index]}/{url_path}/'
- msg += f'
| Index Link'
+ msg += '
| ' \
+ f'
Index Link'
elif mime_type == 'application/vnd.google-apps.shortcut':
- msg += f"⁍
{file.get('name')}" \
- f" (shortcut)"
- # Excluded index link as indexes cant download or open these shortcuts
+ furl = f"https://drive.google.com/drive/folders/{file.get('id')}"
+ msg += '
' \
+ f"📁 {file.get('name')} (shortcut)
" \
+ ''
else:
furl = f"https://drive.google.com/uc?id={file.get('id')}&export=download"
- msg += f"📄
{file.get('name')}
({get_readable_file_size(int(file.get('size', 0)))})"
- msg += f"
Drive Link"
+ msg += '
' \
+ f"📄 {file.get('name')} ({get_readable_file_size(int(file.get('size', 0)))})
" \
+ '' \
+ f'
Drive Link'
if INDEX_URLS[index] is not None:
if isRecur:
- url_path = "/".join(
- rquote(n, safe='')
- for n in self.__get_recursive_list(file, parent_id)
- )
-
+ url_path = "/".join(rquote(n, safe='') for n in self.__get_recursive_list(file, parent_id))
else:
url_path = rquote(f'{file.get("name")}')
url = f'{INDEX_URLS[index]}/{url_path}'
- msg += f'
| Index Link'
+ msg += '
| ' \
+ f'
Index Link'
if VIEW_LINK:
urlv = f'{INDEX_URLS[index]}/{url_path}?a=view'
- msg += f'
| View Link'
- msg += '
'
+ msg += '
| ' \
+ f'
View Link'
+ msg += '
'
contents_count += 1
- if len(msg.encode('utf-8')) > 39000:
- telegraph_content.append(msg)
- msg = ""
if noMulti:
break
- if msg != '':
- telegraph_content.append(msg)
+ if contents_count == 0:
+ return "", ""
- if len(telegraph_content) == 0:
- return "", None
+ rmsg = f"
Found {contents_count} result for {fileName}"
- for content in telegraph_content:
- path.append(
- telegraph.create_page(
- title='Mirror-Leech-Bot Drive Search',
- content=content
- )["path"]
- )
- if len(path) > 1:
- telegraph.edit_telegraph(path, telegraph_content)
-
- msg = f"
Found {contents_count} result for {fileName}"
- buttons = ButtonMaker()
- buttons.buildbutton("🔎 VIEW", f"https://telegra.ph/{path[0]}")
-
- return msg, InlineKeyboardMarkup(buttons.build_menu(1))
+ return rmsg, msg
def count(self, link):
try:
@@ -769,17 +747,16 @@ class GoogleDriveHelper:
return "", size, name, files
def download(self, link):
- self.is_downloading = True
+ self.__is_downloading = True
file_id = self.__getIdFromUrl(link)
- self.updater = setInterval(self.update_interval, self._on_download_progress)
+ self.__updater = setInterval(self.__update_interval, self._progress)
try:
meta = self.__getFileMetadata(file_id)
- path = f"{DOWNLOAD_DIR}{self.__listener.uid}/"
if meta.get("mimeType") == self.__G_DRIVE_DIR_MIME_TYPE:
- self.__download_folder(file_id, path, self.name)
+ self.__download_folder(file_id, self.__path, self.name)
else:
makedirs(path)
- self.__download_file(file_id, path, self.name, meta.get('mimeType'))
+ self.__download_file(file_id, self.__path, self.name, meta.get('mimeType'))
except Exception as err:
if isinstance(err, RetryError):
LOGGER.info(f"Total Attempts: {err.last_attempt.attempt_number}")
@@ -791,21 +768,21 @@ class GoogleDriveHelper:
token_service = self.__alt_authorize()
if token_service is not None:
self.__service = token_service
- self.updater.cancel()
+ self.__updater.cancel()
return self.download(link)
self.__listener.onDownloadError(err)
- self.is_cancelled = True
+ self.__is_cancelled = True
finally:
- self.updater.cancel()
- if self.is_cancelled:
+ self.__updater.cancel()
+ if self.__is_cancelled:
return
self.__listener.onDownloadComplete()
def __download_folder(self, folder_id, path, folder_name):
folder_name = folder_name.replace('/', '')
- if not ospath.exists(path + folder_name):
- makedirs(path + folder_name)
- path += folder_name + '/'
+ if not ospath.exists(f"{path}/{folder_name}"):
+ makedirs(f"{path}/{folder_name}")
+ path += f"/{folder_name}"
result = self.__getFilesByFolderId(folder_id)
if len(result) == 0:
return
@@ -821,9 +798,9 @@ class GoogleDriveHelper:
mime_type = item.get('mimeType')
if mime_type == self.__G_DRIVE_DIR_MIME_TYPE:
self.__download_folder(file_id, path, filename)
- elif not ospath.isfile(path + filename) and not filename.lower().endswith(tuple(EXTENSION_FILTER)):
+ elif not ospath.isfile(f"{path}{filename}") and not filename.lower().endswith(tuple(EXTENSION_FILTER)):
self.__download_file(file_id, path, filename, mime_type)
- if self.is_cancelled:
+ if self.__is_cancelled:
break
@retry(wait=wait_exponential(multiplier=2, min=3, max=6), stop=stop_after_attempt(3),
@@ -833,18 +810,18 @@ class GoogleDriveHelper:
filename = filename.replace('/', '')
if len(filename.encode()) > 255:
ext = ospath.splitext(filename)[1]
- filename = filename[:245] + ext
+ filename = f"{filename[:245]}{ext}"
if self.name.endswith(ext):
self.name = filename
- fh = FileIO(f"{path}{filename}", 'wb')
+ fh = FileIO(f"{path}/{filename}", 'wb')
downloader = MediaIoBaseDownload(fh, request, chunksize=50 * 1024 * 1024)
done = False
while not done:
- if self.is_cancelled:
+ if self.__is_cancelled:
fh.close()
break
try:
- self.dstatus, done = downloader.next_chunk()
+ self.__status, done = downloader.next_chunk()
except HttpError as err:
if err.resp.get('content-type', '').startswith('application/json'):
reason = jsnloads(err.content).get('error').get('errors')[0].get('reason')
@@ -855,7 +832,7 @@ class GoogleDriveHelper:
raise err
if USE_SERVICE_ACCOUNTS:
if self.__sa_count == len(listdir("accounts")) or self.__sa_count > 50:
- self.is_cancelled = True
+ self.__is_cancelled = True
raise err
else:
self.__switchServiceAccount()
@@ -864,22 +841,15 @@ class GoogleDriveHelper:
else:
LOGGER.error(f"Got: {reason}")
raise err
- self._file_downloaded_bytes = 0
-
- def _on_download_progress(self):
- if self.dstatus is not None:
- chunk_size = self.dstatus.total_size * self.dstatus.progress() - self._file_downloaded_bytes
- self._file_downloaded_bytes = self.dstatus.total_size * self.dstatus.progress()
- self.downloaded_bytes += chunk_size
- self.dtotal_time += self.update_interval
+ self._file_processed_bytes = 0
def cancel_download(self):
- self.is_cancelled = True
- if self.is_downloading:
+ self.__is_cancelled = True
+ if self.__is_downloading:
LOGGER.info(f"Cancelling Download: {self.name}")
self.__listener.onDownloadError('Download stopped by user!')
- elif self.is_cloning:
+ elif self.__is_cloning:
LOGGER.info(f"Cancelling Clone: {self.name}")
- elif self.is_uploading:
+ elif self.__is_uploading:
LOGGER.info(f"Cancelling Upload: {self.name}")
self.__listener.onUploadError('your upload has been stopped and uploaded data has been deleted!')
diff --git a/bot/helper/mirror_utils/upload_utils/pyrogramEngine.py b/bot/helper/mirror_utils/upload_utils/pyrogramEngine.py
index bad7f9f9..abcb321c 100644
--- a/bot/helper/mirror_utils/upload_utils/pyrogramEngine.py
+++ b/bot/helper/mirror_utils/upload_utils/pyrogramEngine.py
@@ -1,16 +1,16 @@
-from logging import getLogger, WARNING
+from logging import getLogger, ERROR
from os import remove as osremove, walk, path as ospath, rename as osrename
from time import time, sleep
from pyrogram.errors import FloodWait, RPCError
from PIL import Image
from threading import RLock
-from bot import DOWNLOAD_DIR, AS_DOCUMENT, AS_DOC_USERS, AS_MEDIA_USERS, CUSTOM_FILENAME, EXTENSION_FILTER, app
-from bot.helper.ext_utils.fs_utils import take_ss, get_media_info, get_path_size
+from bot import AS_DOCUMENT, AS_DOC_USERS, AS_MEDIA_USERS, CUSTOM_FILENAME, EXTENSION_FILTER, app, LEECH_SPLIT_SIZE
+from bot.helper.ext_utils.fs_utils import take_ss, get_media_info, clean_unwanted
from bot.helper.ext_utils.bot_utils import get_readable_file_size
LOGGER = getLogger(__name__)
-getLogger("pyrogram").setLevel(WARNING)
+getLogger("pyrogram").setLevel(ERROR)
VIDEO_SUFFIXES = ("MKV", "MP4", "MOV", "WMV", "3GP", "MPG", "WEBM", "AVI", "FLV", "M4V", "GIF")
AUDIO_SUFFIXES = ("MP3", "M4A", "M4B", "FLAC", "WAV", "AIF", "OGG", "AAC", "DTS", "MID", "AMR", "MKA")
@@ -19,11 +19,12 @@ IMAGE_SUFFIXES = ("JPG", "JPX", "PNG", "CR2", "TIF", "BMP", "JXR", "PSD", "ICO",
class TgUploader:
- def __init__(self, name=None, listener=None):
+ def __init__(self, name=None, path=None, size=0, listener=None):
self.name = name
self.uploaded_bytes = 0
self._last_uploaded = 0
self.__listener = listener
+ self.__path = path
self.__start_time = time()
self.__total_files = 0
self.__is_cancelled = False
@@ -34,16 +35,17 @@ class TgUploader:
self.__resource_lock = RLock()
self.__is_corrupted = False
self.__sent_msg = app.get_messages(self.__listener.message.chat.id, self.__listener.uid)
+ self.__size = size
self.__user_settings()
- def upload(self):
- path = f"{DOWNLOAD_DIR}{self.__listener.uid}"
- size = get_readable_file_size(get_path_size(path))
- for dirpath, subdir, files in sorted(walk(path)):
+ def upload(self, o_files):
+ for dirpath, subdir, files in sorted(walk(self.__path)):
for file_ in sorted(files):
+ if file_ in o_files:
+ continue
if not file_.lower().endswith(tuple(EXTENSION_FILTER)):
- self.__total_files += 1
up_path = ospath.join(dirpath, file_)
+ self.__total_files += 1
if ospath.getsize(up_path) == 0:
LOGGER.error(f"{up_path} size is zero, telegram don't upload zero size files")
self.__corrupted += 1
@@ -55,9 +57,12 @@ class TgUploader:
self.__msgs_dict[self.__sent_msg.link] = file_
self._last_uploaded = 0
sleep(1)
+ if self.__listener.seed and not self.__listener.newDir:
+ clean_unwanted(self.__path)
if self.__total_files <= self.__corrupted:
return self.__listener.onUploadError('Files Corrupted. Check logs')
LOGGER.info(f"Leech Completed: {self.name}")
+ size = get_readable_file_size(self.__size)
self.__listener.onUploadComplete(None, size, self.__msgs_dict, self.__total_files, self.__corrupted, self.name)
def __upload_file(self, up_path, file_, dirpath):
@@ -149,8 +154,12 @@ class TgUploader:
self.__is_corrupted = True
if self.__thumb is None and thumb is not None and ospath.lexists(thumb):
osremove(thumb)
- if not self.__is_cancelled:
- osremove(up_path)
+ if not self.__is_cancelled and \
+ (not self.__listener.seed or self.__listener.newDir or dirpath.endswith("splited_files_mltb")):
+ try:
+ osremove(up_path)
+ except:
+ pass
def __upload_progress(self, current, total):
if self.__is_cancelled:
diff --git a/bot/helper/telegram_helper/bot_commands.py b/bot/helper/telegram_helper/bot_commands.py
index 518d9324..78b6adb9 100644
--- a/bot/helper/telegram_helper/bot_commands.py
+++ b/bot/helper/telegram_helper/bot_commands.py
@@ -4,9 +4,25 @@ from bot import CMD_INDEX
class _BotCommands:
def __init__(self):
self.StartCommand = f'start{CMD_INDEX}'
- self.MirrorCommand = f'mirror{CMD_INDEX}'
- self.UnzipMirrorCommand = f'unzipmirror{CMD_INDEX}'
- self.ZipMirrorCommand = f'zipmirror{CMD_INDEX}'
+ self.MirrorCommand = (f'mirror{CMD_INDEX}', f'm{CMD_INDEX}')
+ self.UnzipMirrorCommand = (f'unzipmirror{CMD_INDEX}', f'uzm{CMD_INDEX}')
+ self.ZipMirrorCommand = (f'zipmirror{CMD_INDEX}', f'zm{CMD_INDEX}')
+ self.QbMirrorCommand = (f'qbmirror{CMD_INDEX}', f'qm{CMD_INDEX}')
+ self.QbUnzipMirrorCommand = (f'qbunzipmirror{CMD_INDEX}', f'quzm{CMD_INDEX}')
+ self.QbZipMirrorCommand = (f'qbzipmirror{CMD_INDEX}', f'qzm{CMD_INDEX}')
+ self.YtdlCommand = (f'ytdl{CMD_INDEX}', f'y{CMD_INDEX}')
+ self.YtdlZipCommand = (f'ytdlzip{CMD_INDEX}', f'yz{CMD_INDEX}')
+ self.LeechCommand = (f'leech{CMD_INDEX}', f'l{CMD_INDEX}')
+ self.UnzipLeechCommand = (f'unzipleech{CMD_INDEX}', f'uzl{CMD_INDEX}')
+ self.ZipLeechCommand = (f'zipleech{CMD_INDEX}', f'zl{CMD_INDEX}')
+ self.QbLeechCommand = (f'qbleech{CMD_INDEX}', f'ql{CMD_INDEX}')
+ self.QbUnzipLeechCommand = (f'qbunzipleech{CMD_INDEX}', f'quzl{CMD_INDEX}')
+ self.QbZipLeechCommand = (f'qbzipleech{CMD_INDEX}', f'qzl{CMD_INDEX}')
+ self.YtdlLeechCommand = (f'ytdlleech{CMD_INDEX}', f'yl{CMD_INDEX}')
+ self.YtdlZipLeechCommand = (f'ytdlzipleech{CMD_INDEX}', f'yzl{CMD_INDEX}')
+ self.CloneCommand = f'clone{CMD_INDEX}'
+ self.CountCommand = f'count{CMD_INDEX}'
+ self.DeleteCommand = f'del{CMD_INDEX}'
self.CancelMirror = f'cancel{CMD_INDEX}'
self.CancelAllCommand = f'cancelall{CMD_INDEX}'
self.ListCommand = f'list{CMD_INDEX}'
@@ -22,34 +38,17 @@ class _BotCommands:
self.StatsCommand = f'stats{CMD_INDEX}'
self.HelpCommand = f'help{CMD_INDEX}'
self.LogCommand = f'log{CMD_INDEX}'
- self.CloneCommand = f'clone{CMD_INDEX}'
- self.CountCommand = f'count{CMD_INDEX}'
- self.WatchCommand = f'watch{CMD_INDEX}'
- self.ZipWatchCommand = f'zipwatch{CMD_INDEX}'
- self.QbMirrorCommand = f'qbmirror{CMD_INDEX}'
- self.QbUnzipMirrorCommand = f'qbunzipmirror{CMD_INDEX}'
- self.QbZipMirrorCommand = f'qbzipmirror{CMD_INDEX}'
- self.DeleteCommand = f'del{CMD_INDEX}'
self.ShellCommand = f'shell{CMD_INDEX}'
- self.ExecHelpCommand = f'exechelp{CMD_INDEX}'
- self.LeechSetCommand = f'leechset{CMD_INDEX}'
- self.SetThumbCommand = f'setthumb{CMD_INDEX}'
- self.LeechCommand = f'leech{CMD_INDEX}'
- self.UnzipLeechCommand = f'unzipleech{CMD_INDEX}'
- self.ZipLeechCommand = f'zipleech{CMD_INDEX}'
- self.QbLeechCommand = f'qbleech{CMD_INDEX}'
- self.QbUnzipLeechCommand = f'qbunzipleech{CMD_INDEX}'
- self.QbZipLeechCommand = f'qbzipleech{CMD_INDEX}'
- self.LeechWatchCommand = f'leechwatch{CMD_INDEX}'
- self.LeechZipWatchCommand = f'leechzipwatch{CMD_INDEX}'
- self.BtSelectCommand = f'btsel{CMD_INDEX}'
- self.RssListCommand = f'rsslist{CMD_INDEX}'
- self.RssGetCommand = f'rssget{CMD_INDEX}'
- self.RssSubCommand = f'rsssub{CMD_INDEX}'
- self.RssUnSubCommand = f'rssunsub{CMD_INDEX}'
- self.RssSettingsCommand = f'rssset{CMD_INDEX}'
self.EvalCommand = f'eval{CMD_INDEX}'
self.ExecCommand = f'exec{CMD_INDEX}'
self.ClearLocalsCommand = f'clearlocals{CMD_INDEX}'
+ self.LeechSetCommand = f'leechset{CMD_INDEX}'
+ self.SetThumbCommand = f'setthumb{CMD_INDEX}'
+ self.BtSelectCommand = f'btsel{CMD_INDEX}'
+ self.RssListCommand = (f'rsslist{CMD_INDEX}', f'rl{CMD_INDEX}')
+ self.RssGetCommand = (f'rssget{CMD_INDEX}', f'rg{CMD_INDEX}')
+ self.RssSubCommand = (f'rsssub{CMD_INDEX}', f'rs{CMD_INDEX}')
+ self.RssUnSubCommand = (f'rssunsub{CMD_INDEX}', f'rus{CMD_INDEX}')
+ self.RssSettingsCommand = (f'rssset{CMD_INDEX}', f'rst{CMD_INDEX}')
BotCommands = _BotCommands()
diff --git a/bot/helper/telegram_helper/message_utils.py b/bot/helper/telegram_helper/message_utils.py
index fe89b904..5241c58c 100644
--- a/bot/helper/telegram_helper/message_utils.py
+++ b/bot/helper/telegram_helper/message_utils.py
@@ -85,6 +85,19 @@ def sendLogFile(bot, message: Message):
reply_to_message_id=message.message_id,
chat_id=message.chat_id)
+def sendFile(bot, message: Message, name: str, caption=""):
+ with open(name, 'rb') as f:
+ try:
+ bot.sendDocument(document=f, filename=f.name, reply_to_message_id=message.message_id,
+ caption=caption, parse_mode='HTMl',chat_id=message.chat_id)
+ except RetryAfter as r:
+ LOGGER.warning(str(r))
+ sleep(r.retry_after * 1.5)
+ return sendFile(bot, message, name, caption)
+ except Exception as e:
+ LOGGER.error(str(e))
+ return
+
def auto_delete_message(bot, cmd_message: Message, bot_message: Message):
if AUTO_DELETE_MESSAGE_DURATION != -1:
sleep(AUTO_DELETE_MESSAGE_DURATION)
diff --git a/bot/modules/eval.py b/bot/modules/eval.py
index 50331028..cc262914 100644
--- a/bot/modules/eval.py
+++ b/bot/modules/eval.py
@@ -112,22 +112,10 @@ def clear(update, context):
del namespaces[update.message.chat_id]
send("Cleared locals.", bot, update)
-def exechelp(update, context):
- help_string = f'''
-
Executor
-• {BotCommands.EvalCommand}
Run Python Code Line | Lines
-• {BotCommands.ExecCommand}
Run Commands In Exec
-• {BotCommands.ClearLocalsCommand}
Cleared locals
-'''
- sendMessage(help_string, context.bot, update.message)
-
-
EVAL_HANDLER = CommandHandler(BotCommands.EvalCommand, evaluate, filters=CustomFilters.owner_filter, run_async=True)
EXEC_HANDLER = CommandHandler(BotCommands.ExecCommand, execute, filters=CustomFilters.owner_filter, run_async=True)
CLEAR_HANDLER = CommandHandler(BotCommands.ClearLocalsCommand, clear, filters=CustomFilters.owner_filter, run_async=True)
-EXECHELP_HANDLER = CommandHandler(BotCommands.ExecHelpCommand, exechelp, filters=CustomFilters.owner_filter, run_async=True)
dispatcher.add_handler(EVAL_HANDLER)
dispatcher.add_handler(EXEC_HANDLER)
dispatcher.add_handler(CLEAR_HANDLER)
-dispatcher.add_handler(EXECHELP_HANDLER)
diff --git a/bot/modules/list.py b/bot/modules/list.py
index 0fa52f37..57293a82 100644
--- a/bot/modules/list.py
+++ b/bot/modules/list.py
@@ -1,13 +1,16 @@
from threading import Thread
from telegram import InlineKeyboardMarkup
from telegram.ext import CommandHandler, CallbackQueryHandler
+from time import time
+from os import remove
from bot import LOGGER, dispatcher
from bot.helper.mirror_utils.upload_utils.gdriveTools import GoogleDriveHelper
-from bot.helper.telegram_helper.message_utils import sendMessage, editMessage, sendMarkup
+from bot.helper.telegram_helper.message_utils import sendMessage, editMessage, sendMarkup, sendFile, deleteMessage
from bot.helper.telegram_helper.filters import CustomFilters
from bot.helper.telegram_helper.bot_commands import BotCommands
from bot.helper.telegram_helper import button_build
+from bot.helper.ext_utils.html_helper import hmtl_content
def list_buttons(update, context):
user_id = update.message.from_user.id
@@ -36,14 +39,19 @@ def select_type(update, context):
query.answer()
item_type = data[2]
editMessage(f"
Searching for {key}", msg)
- Thread(target=_list_drive, args=(key, msg, item_type)).start()
+ Thread(target=_list_drive, args=(context.bot, key, msg, item_type)).start()
-def _list_drive(key, bmsg, item_type):
+def _list_drive(bot, key, bmsg, item_type):
LOGGER.info(f"listing: {key}")
gdrive = GoogleDriveHelper()
- msg, button = gdrive.drive_list(key, isRecursive=True, itemType=item_type)
- if button:
- editMessage(msg, bmsg, button)
+ rmsg, msg = gdrive.drive_list(key, isRecursive=True, itemType=item_type)
+ if msg:
+ name = f'{key}_{time()}.html'
+ with open(name, 'w', encoding='utf-8') as f:
+ f.write(hmtl_content.replace('{fileName}', key).replace('{msg}', msg))
+ deleteMessage(bot, bmsg)
+ sendFile(bot, bmsg.reply_to_message, name, rmsg)
+ remove(name)
else:
editMessage(f'No result found for
{key}', bmsg)
diff --git a/bot/modules/listener.py b/bot/modules/listener.py
new file mode 100644
index 00000000..e9e80236
--- /dev/null
+++ b/bot/modules/listener.py
@@ -0,0 +1,317 @@
+from requests import utils as rutils
+from re import search as re_search
+from time import sleep
+from os import path as ospath, remove as osremove, listdir, walk
+from subprocess import Popen
+from html import escape
+from telegram import InlineKeyboardMarkup
+
+from bot import Interval, INDEX_URL, VIEW_LINK, aria2, DOWNLOAD_DIR, download_dict, download_dict_lock, \
+ LEECH_SPLIT_SIZE, LOGGER, DB_URI, INCOMPLETE_TASK_NOTIFIER, MAX_SPLIT_SIZE
+from bot.helper.ext_utils.fs_utils import get_base_name, get_path_size, split_file, clean_download, clean_target
+from bot.helper.ext_utils.exceptions import NotSupportedExtractionArchive
+from bot.helper.mirror_utils.status_utils.extract_status import ExtractStatus
+from bot.helper.mirror_utils.status_utils.zip_status import ZipStatus
+from bot.helper.mirror_utils.status_utils.split_status import SplitStatus
+from bot.helper.mirror_utils.status_utils.upload_status import UploadStatus
+from bot.helper.mirror_utils.status_utils.tg_upload_status import TgUploadStatus
+from bot.helper.mirror_utils.upload_utils.gdriveTools import GoogleDriveHelper
+from bot.helper.mirror_utils.upload_utils.pyrogramEngine import TgUploader
+from bot.helper.telegram_helper.message_utils import sendMessage, sendMarkup, delete_all_messages, update_all_messages
+from bot.helper.telegram_helper.button_build import ButtonMaker
+from bot.helper.ext_utils.db_handler import DbManger
+
+
+class MirrorLeechListener:
+ def __init__(self, bot, message, isZip=False, extract=False, isQbit=False, isLeech=False, pswd=None, tag=None, select=False, seed=False):
+ self.bot = bot
+ self.message = message
+ self.uid = self.message.message_id
+ self.extract = extract
+ self.isZip = isZip
+ self.isQbit = isQbit
+ self.isLeech = isLeech
+ self.pswd = pswd
+ self.tag = tag
+ self.seed = seed
+ self.newDir = ""
+ self.dir = f"{DOWNLOAD_DIR}{self.uid}"
+ self.select = select
+ self.isPrivate = self.message.chat.type in ['private', 'group']
+ self.suproc = None
+
+ def clean(self):
+ try:
+ Interval[0].cancel()
+ Interval.clear()
+ aria2.purge()
+ delete_all_messages()
+ except:
+ pass
+
+ def onDownloadStart(self):
+ if not self.isPrivate and INCOMPLETE_TASK_NOTIFIER and DB_URI is not None:
+ DbManger().add_incomplete_task(self.message.chat.id, self.message.link, self.tag)
+
+ def onDownloadComplete(self):
+ with download_dict_lock:
+ download = download_dict[self.uid]
+ name = str(download.name()).replace('/', '')
+ gid = download.gid()
+ LOGGER.info(f"Download completed: {name}")
+ if name == "None" or self.isQbit or not ospath.exists(f"{self.dir}/{name}"):
+ name = listdir(f"{self.dir}")[-1]
+ m_path = f'{self.dir}/{name}'
+ size = get_path_size(m_path)
+ if self.isZip:
+ if self.seed and self.isLeech:
+ self.newDir = f"{self.dir}10000"
+ path = f"{self.newDir}/{name}.zip"
+ else:
+ path = f"{m_path}.zip"
+ with download_dict_lock:
+ download_dict[self.uid] = ZipStatus(name, size, gid, self)
+ if self.pswd is not None:
+ if self.isLeech and int(size) > LEECH_SPLIT_SIZE:
+ LOGGER.info(f'Zip: orig_path: {m_path}, zip_path: {path}.0*')
+ self.suproc = Popen(["7z", f"-v{LEECH_SPLIT_SIZE}b", "a", "-mx=0", f"-p{self.pswd}", path, m_path])
+ else:
+ LOGGER.info(f'Zip: orig_path: {m_path}, zip_path: {path}')
+ self.suproc = Popen(["7z", "a", "-mx=0", f"-p{self.pswd}", path, m_path])
+ elif self.isLeech and int(size) > LEECH_SPLIT_SIZE:
+ LOGGER.info(f'Zip: orig_path: {m_path}, zip_path: {path}.0*')
+ self.suproc = Popen(["7z", f"-v{LEECH_SPLIT_SIZE}b", "a", "-mx=0", path, m_path])
+ else:
+ LOGGER.info(f'Zip: orig_path: {m_path}, zip_path: {path}')
+ self.suproc = Popen(["7z", "a", "-mx=0", path, m_path])
+ self.suproc.wait()
+ if self.suproc.returncode == -9:
+ return
+ elif not self.seed:
+ clean_target(m_path)
+ elif self.extract:
+ try:
+ if ospath.isfile(m_path):
+ path = get_base_name(m_path)
+ LOGGER.info(f"Extracting: {name}")
+ with download_dict_lock:
+ download_dict[self.uid] = ExtractStatus(name, size, gid, self)
+ if ospath.isdir(m_path):
+ if self.seed:
+ self.newDir = f"{self.dir}10000"
+ path = f"{self.newDir}/{name}"
+ for dirpath, subdir, files in walk(m_path, topdown=False):
+ for file_ in files:
+ if re_search(r'\.part0*1\.rar$|\.7z\.0*1$|\.zip\.0*1$|\.zip$|\.7z$|^.(?!.*\.part\d+\.rar)(?=.*\.rar$)', file_):
+ m_path = ospath.join(dirpath, file_)
+ if self.seed:
+ t_path = dirpath.replace(self.dir, self.newDir)
+ else:
+ t_path = dirpath
+ if self.pswd is not None:
+ self.suproc = Popen(["7z", "x", f"-p{self.pswd}", m_path, f"-o{t_path}", "-aot"])
+ else:
+ self.suproc = Popen(["7z", "x", m_path, f"-o{t_path}", "-aot"])
+ self.suproc.wait()
+ if self.suproc.returncode == -9:
+ return
+ elif self.suproc.returncode != 0:
+ LOGGER.error('Unable to extract archive splits!')
+ if not self.seed and self.suproc is not None and self.suproc.returncode == 0:
+ for file_ in files:
+ if re_search(r'\.r\d+$|\.7z\.\d+$|\.z\d+$|\.zip\.\d+$|\.zip$|\.rar$|\.7z$', file_):
+ del_path = ospath.join(dirpath, file_)
+ try:
+ osremove(del_path)
+ except:
+ return
+ else:
+ if self.seed and self.isLeech:
+ self.newDir = f"{self.dir}10000"
+ path = path.replace(self.dir, self.newDir)
+ if self.pswd is not None:
+ self.suproc = Popen(["7z", "x", f"-p{self.pswd}", m_path, f"-o{path}", "-aot"])
+ else:
+ self.suproc = Popen(["7z", "x", m_path, f"-o{path}", "-aot"])
+ self.suproc.wait()
+ if self.suproc.returncode == -9:
+ return
+ elif self.suproc.returncode == 0:
+ LOGGER.info(f"Extracted Path: {path}")
+ if not self.seed:
+ try:
+ osremove(m_path)
+ except:
+ return
+ else:
+ LOGGER.error('Unable to extract archive! Uploading anyway')
+ self.newDir = ""
+ path = m_path
+ except NotSupportedExtractionArchive:
+ LOGGER.info("Not any valid archive, uploading file as it is.")
+ self.newDir = ""
+ path = m_path
+ else:
+ path = m_path
+ up_dir, up_name = path.rsplit('/', 1)
+ size = get_path_size(up_dir)
+ if self.isLeech:
+ m_size = []
+ o_files = []
+ if not self.isZip:
+ checked = False
+ for dirpath, subdir, files in walk(up_dir, topdown=False):
+ for file_ in files:
+ f_path = ospath.join(dirpath, file_)
+ f_size = ospath.getsize(f_path)
+ if f_size > LEECH_SPLIT_SIZE:
+ if not checked:
+ checked = True
+ with download_dict_lock:
+ download_dict[self.uid] = SplitStatus(up_name, size, gid, self)
+ LOGGER.info(f"Splitting: {up_name}")
+ res = split_file(f_path, f_size, file_, dirpath, LEECH_SPLIT_SIZE, self)
+ if not res:
+ return
+ if res == "errored":
+ if f_size <= MAX_SPLIT_SIZE:
+ continue
+ else:
+ try:
+ osremove(f_path)
+ except:
+ return
+ elif not self.seed or self.newDir:
+ try:
+ osremove(f_path)
+ except:
+ return
+ elif self.seed and res != "errored":
+ m_size.append(f_size)
+ o_files.append(file_)
+
+ size = get_path_size(up_dir)
+ for s in m_size:
+ size = size - s
+ LOGGER.info(f"Leech Name: {up_name}")
+ tg = TgUploader(up_name, up_dir, size, self)
+ tg_upload_status = TgUploadStatus(tg, size, gid, self)
+ with download_dict_lock:
+ download_dict[self.uid] = tg_upload_status
+ update_all_messages()
+ tg.upload(o_files)
+ else:
+ up_path = f'{up_dir}/{up_name}'
+ size = get_path_size(up_path)
+ LOGGER.info(f"Upload Name: {up_name}")
+ drive = GoogleDriveHelper(up_name, up_dir, size, self)
+ upload_status = UploadStatus(drive, size, gid, self)
+ with download_dict_lock:
+ download_dict[self.uid] = upload_status
+ update_all_messages()
+ drive.upload(up_name)
+
+ def onUploadComplete(self, link: str, size, files, folders, typ, name):
+ if not self.isPrivate and INCOMPLETE_TASK_NOTIFIER and DB_URI is not None:
+ DbManger().rm_complete_task(self.message.link)
+ msg = f"
Name: {escape(name)}\n\n
Size: {size}"
+ if self.isLeech:
+ msg += f'\n
Total Files: {folders}'
+ if typ != 0:
+ msg += f'\n
Corrupted Files: {typ}'
+ msg += f'\n
cc: {self.tag}\n\n'
+ if not files:
+ sendMessage(msg, self.bot, self.message)
+ else:
+ fmsg = ''
+ for index, (link, name) in enumerate(files.items(), start=1):
+ fmsg += f"{index}.
{name}\n"
+ if len(fmsg.encode() + msg.encode()) > 4000:
+ sendMessage(msg + fmsg, self.bot, self.message)
+ sleep(1)
+ fmsg = ''
+ if fmsg != '':
+ sendMessage(msg + fmsg, self.bot, self.message)
+ if self.seed:
+ if self.newDir:
+ clean_target(self.newDir)
+ return
+ else:
+ msg += f'\n\n
Type: {typ}'
+ if typ == "Folder":
+ msg += f'\n
SubFolders: {folders}'
+ msg += f'\n
Files: {files}'
+ msg += f'\n\n
cc: {self.tag}'
+ buttons = ButtonMaker()
+ buttons.buildbutton("☁️ Drive Link", link)
+ LOGGER.info(f'Done Uploading {name}')
+ if INDEX_URL is not None:
+ url_path = rutils.quote(f'{name}')
+ share_url = f'{INDEX_URL}/{url_path}'
+ if typ == "Folder":
+ share_url += '/'
+ buttons.buildbutton("⚡ Index Link", share_url)
+ else:
+ buttons.buildbutton("⚡ Index Link", share_url)
+ if VIEW_LINK:
+ share_urls = f'{INDEX_URL}/{url_path}?a=view'
+ buttons.buildbutton("🌐 View Link", share_urls)
+ sendMarkup(msg, self.bot, self.message, InlineKeyboardMarkup(buttons.build_menu(2)))
+ if self.seed:
+ if self.isZip:
+ clean_target(f"{self.dir}/{name}")
+ elif self.newDir:
+ clean_target(self.newDir)
+ return
+ clean_download(self.dir)
+ with download_dict_lock:
+ try:
+ del download_dict[self.uid]
+ except Exception as e:
+ LOGGER.error(str(e))
+ count = len(download_dict)
+ if count == 0:
+ self.clean()
+ else:
+ update_all_messages()
+
+ def onDownloadError(self, error):
+ error = error.replace('<', ' ').replace('>', ' ')
+ clean_download(self.dir)
+ if self.newDir:
+ clean_download(self.newDir)
+ with download_dict_lock:
+ try:
+ del download_dict[self.uid]
+ except Exception as e:
+ LOGGER.error(str(e))
+ count = len(download_dict)
+ msg = f"{self.tag} your download has been stopped due to: {error}"
+ sendMessage(msg, self.bot, self.message)
+ if count == 0:
+ self.clean()
+ else:
+ update_all_messages()
+
+ if not self.isPrivate and INCOMPLETE_TASK_NOTIFIER and DB_URI is not None:
+ DbManger().rm_complete_task(self.message.link)
+
+ def onUploadError(self, error):
+ e_str = error.replace('<', '').replace('>', '')
+ clean_download(self.dir)
+ if self.newDir:
+ clean_download(self.newDir)
+ with download_dict_lock:
+ try:
+ del download_dict[self.uid]
+ except Exception as e:
+ LOGGER.error(str(e))
+ count = len(download_dict)
+ sendMessage(f"{self.tag} {e_str}", self.bot, self.message)
+ if count == 0:
+ self.clean()
+ else:
+ update_all_messages()
+
+ if not self.isPrivate and INCOMPLETE_TASK_NOTIFIER and DB_URI is not None:
+ DbManger().rm_complete_task(self.message.link)
diff --git a/bot/modules/mirror.py b/bot/modules/mirror.py
deleted file mode 100644
index fe153ee4..00000000
--- a/bot/modules/mirror.py
+++ /dev/null
@@ -1,546 +0,0 @@
-from base64 import b64encode
-from requests import utils as rutils, get as rget
-from re import match as re_match, search as re_search, split as re_split
-from time import sleep, time
-from os import path as ospath, remove as osremove, listdir, walk
-from shutil import rmtree
-from threading import Thread
-from subprocess import Popen
-from html import escape
-from telegram.ext import CommandHandler
-from telegram import InlineKeyboardMarkup
-
-from bot import Interval, INDEX_URL, VIEW_LINK, aria2, QB_SEED, dispatcher, DOWNLOAD_DIR, \
- download_dict, download_dict_lock, LEECH_SPLIT_SIZE, LOGGER, DB_URI, INCOMPLETE_TASK_NOTIFIER
-from bot.helper.ext_utils.bot_utils import is_url, is_magnet, is_mega_link, is_gdrive_link, get_content_type
-from bot.helper.ext_utils.fs_utils import get_base_name, get_path_size, split_file, clean_download
-from bot.helper.ext_utils.exceptions import DirectDownloadLinkException, NotSupportedExtractionArchive
-from bot.helper.mirror_utils.download_utils.aria2_download import add_aria2c_download
-from bot.helper.mirror_utils.download_utils.gd_downloader import add_gd_download
-from bot.helper.mirror_utils.download_utils.qbit_downloader import QbDownloader
-from bot.helper.mirror_utils.download_utils.mega_downloader import add_mega_download
-from bot.helper.mirror_utils.download_utils.direct_link_generator import direct_link_generator
-from bot.helper.mirror_utils.download_utils.telegram_downloader import TelegramDownloadHelper
-from bot.helper.mirror_utils.status_utils.extract_status import ExtractStatus
-from bot.helper.mirror_utils.status_utils.zip_status import ZipStatus
-from bot.helper.mirror_utils.status_utils.split_status import SplitStatus
-from bot.helper.mirror_utils.status_utils.upload_status import UploadStatus
-from bot.helper.mirror_utils.status_utils.tg_upload_status import TgUploadStatus
-from bot.helper.mirror_utils.upload_utils.gdriveTools import GoogleDriveHelper
-from bot.helper.mirror_utils.upload_utils.pyrogramEngine import TgUploader
-from bot.helper.telegram_helper.bot_commands import BotCommands
-from bot.helper.telegram_helper.filters import CustomFilters
-from bot.helper.telegram_helper.message_utils import sendMessage, sendMarkup, delete_all_messages, update_all_messages
-from bot.helper.telegram_helper.button_build import ButtonMaker
-from bot.helper.ext_utils.db_handler import DbManger
-
-
-class MirrorListener:
- def __init__(self, bot, message, isZip=False, extract=False, isQbit=False, isLeech=False, pswd=None, tag=None, select=False, seed=False):
- self.bot = bot
- self.message = message
- self.uid = self.message.message_id
- self.extract = extract
- self.isZip = isZip
- self.isQbit = isQbit
- self.isLeech = isLeech
- self.pswd = pswd
- self.tag = tag
- self.seed = any([seed, QB_SEED])
- self.select = select
- self.isPrivate = self.message.chat.type in ['private', 'group']
- self.suproc = None
-
- def clean(self):
- try:
- Interval[0].cancel()
- Interval.clear()
- aria2.purge()
- delete_all_messages()
- except:
- pass
-
- def onDownloadStart(self):
- if not self.isPrivate and INCOMPLETE_TASK_NOTIFIER and DB_URI is not None:
- DbManger().add_incomplete_task(self.message.chat.id, self.message.link, self.tag)
-
- def onDownloadComplete(self):
- with download_dict_lock:
- LOGGER.info(f"Download completed: {download_dict[self.uid].name()}")
- download = download_dict[self.uid]
- name = str(download.name()).replace('/', '')
- gid = download.gid()
- if name == "None" or self.isQbit or not ospath.exists(f'{DOWNLOAD_DIR}{self.uid}/{name}'):
- name = listdir(f'{DOWNLOAD_DIR}{self.uid}')[-1]
- m_path = f'{DOWNLOAD_DIR}{self.uid}/{name}'
- size = get_path_size(m_path)
- if self.isZip:
- path = m_path + ".zip"
- with download_dict_lock:
- download_dict[self.uid] = ZipStatus(name, size, gid, self)
- if self.pswd is not None:
- if self.isLeech and int(size) > LEECH_SPLIT_SIZE:
- LOGGER.info(f'Zip: orig_path: {m_path}, zip_path: {path}.0*')
- self.suproc = Popen(["7z", f"-v{LEECH_SPLIT_SIZE}b", "a", "-mx=0", f"-p{self.pswd}", path, m_path])
- else:
- LOGGER.info(f'Zip: orig_path: {m_path}, zip_path: {path}')
- self.suproc = Popen(["7z", "a", "-mx=0", f"-p{self.pswd}", path, m_path])
- elif self.isLeech and int(size) > LEECH_SPLIT_SIZE:
- LOGGER.info(f'Zip: orig_path: {m_path}, zip_path: {path}.0*')
- self.suproc = Popen(["7z", f"-v{LEECH_SPLIT_SIZE}b", "a", "-mx=0", path, m_path])
- else:
- LOGGER.info(f'Zip: orig_path: {m_path}, zip_path: {path}')
- self.suproc = Popen(["7z", "a", "-mx=0", path, m_path])
- self.suproc.wait()
- if self.suproc.returncode == -9:
- return
- elif self.suproc.returncode != 0:
- LOGGER.error('An error occurred while zipping! Uploading anyway')
- path = f'{DOWNLOAD_DIR}{self.uid}/{name}'
- if self.suproc.returncode == 0 and (not self.isQbit or not self.seed or self.isLeech):
- try:
- rmtree(m_path)
- except:
- osremove(m_path)
- elif self.extract:
- try:
- if ospath.isfile(m_path):
- path = get_base_name(m_path)
- LOGGER.info(f"Extracting: {name}")
- with download_dict_lock:
- download_dict[self.uid] = ExtractStatus(name, size, gid, self)
- if ospath.isdir(m_path):
- for dirpath, subdir, files in walk(m_path, topdown=False):
- for file_ in files:
- if file_.endswith((".zip", ".7z")) or re_search(r'\.part0*1\.rar$|\.7z\.0*1$|\.zip\.0*1$', file_) \
- or (file_.endswith(".rar") and not re_search(r'\.part\d+\.rar$', file_)):
- m_path = ospath.join(dirpath, file_)
- if self.pswd is not None:
- self.suproc = Popen(["7z", "x", f"-p{self.pswd}", m_path, f"-o{dirpath}", "-aot"])
- else:
- self.suproc = Popen(["7z", "x", m_path, f"-o{dirpath}", "-aot"])
- self.suproc.wait()
- if self.suproc.returncode == -9:
- return
- elif self.suproc.returncode != 0:
- LOGGER.error('Unable to extract archive splits! Uploading anyway')
- if self.suproc is not None and self.suproc.returncode == 0:
- for file_ in files:
- if file_.endswith((".rar", ".zip", ".7z")) or \
- re_search(r'\.r\d+$|\.7z\.\d+$|\.z\d+$|\.zip\.\d+$', file_):
- del_path = ospath.join(dirpath, file_)
- osremove(del_path)
- path = f'{DOWNLOAD_DIR}{self.uid}/{name}'
- else:
- if self.pswd is not None:
- self.suproc = Popen(["bash", "pextract", m_path, self.pswd])
- else:
- self.suproc = Popen(["bash", "extract", m_path])
- self.suproc.wait()
- if self.suproc.returncode == -9:
- return
- elif self.suproc.returncode == 0:
- LOGGER.info(f"Extracted Path: {path}")
- osremove(m_path)
- else:
- LOGGER.error('Unable to extract archive! Uploading anyway')
- path = f'{DOWNLOAD_DIR}{self.uid}/{name}'
- except NotSupportedExtractionArchive:
- LOGGER.info("Not any valid archive, uploading file as it is.")
- path = f'{DOWNLOAD_DIR}{self.uid}/{name}'
- else:
- path = f'{DOWNLOAD_DIR}{self.uid}/{name}'
- up_name = path.rsplit('/', 1)[-1]
- if self.isLeech and not self.isZip:
- checked = False
- for dirpath, subdir, files in walk(f'{DOWNLOAD_DIR}{self.uid}', topdown=False):
- for file_ in files:
- f_path = ospath.join(dirpath, file_)
- f_size = ospath.getsize(f_path)
- if int(f_size) > LEECH_SPLIT_SIZE:
- if not checked:
- checked = True
- with download_dict_lock:
- download_dict[self.uid] = SplitStatus(up_name, size, gid, self)
- LOGGER.info(f"Splitting: {up_name}")
- res = split_file(f_path, f_size, file_, dirpath, LEECH_SPLIT_SIZE, self)
- if not res:
- return
- osremove(f_path)
- if self.isLeech:
- size = get_path_size(f'{DOWNLOAD_DIR}{self.uid}')
- LOGGER.info(f"Leech Name: {up_name}")
- tg = TgUploader(up_name, self)
- tg_upload_status = TgUploadStatus(tg, size, gid, self)
- with download_dict_lock:
- download_dict[self.uid] = tg_upload_status
- update_all_messages()
- tg.upload()
- else:
- up_path = f'{DOWNLOAD_DIR}{self.uid}/{up_name}'
- size = get_path_size(up_path)
- LOGGER.info(f"Upload Name: {up_name}")
- drive = GoogleDriveHelper(up_name, self)
- upload_status = UploadStatus(drive, size, gid, self)
- with download_dict_lock:
- download_dict[self.uid] = upload_status
- update_all_messages()
- drive.upload(up_name)
-
- def onDownloadError(self, error):
- error = error.replace('<', ' ').replace('>', ' ')
- clean_download(f'{DOWNLOAD_DIR}{self.uid}')
- with download_dict_lock:
- try:
- del download_dict[self.uid]
- except Exception as e:
- LOGGER.error(str(e))
- count = len(download_dict)
- msg = f"{self.tag} your download has been stopped due to: {error}"
- sendMessage(msg, self.bot, self.message)
- if count == 0:
- self.clean()
- else:
- update_all_messages()
-
- if not self.isPrivate and INCOMPLETE_TASK_NOTIFIER and DB_URI is not None:
- DbManger().rm_complete_task(self.message.link)
-
- def onUploadComplete(self, link: str, size, files, folders, typ, name: str):
- if not self.isPrivate and INCOMPLETE_TASK_NOTIFIER and DB_URI is not None:
- DbManger().rm_complete_task(self.message.link)
- msg = f"
Name: {escape(name)}\n\n
Size: {size}"
- if self.isLeech:
- msg += f'\n
Total Files: {folders}'
- if typ != 0:
- msg += f'\n
Corrupted Files: {typ}'
- msg += f'\n
cc: {self.tag}\n\n'
- if not files:
- sendMessage(msg, self.bot, self.message)
- else:
- fmsg = ''
- for index, (link, name) in enumerate(files.items(), start=1):
- fmsg += f"{index}.
{name}\n"
- if len(fmsg.encode() + msg.encode()) > 4000:
- sendMessage(msg + fmsg, self.bot, self.message)
- sleep(1)
- fmsg = ''
- if fmsg != '':
- sendMessage(msg + fmsg, self.bot, self.message)
- else:
- msg += f'\n\n
Type: {typ}'
- if ospath.isdir(f'{DOWNLOAD_DIR}{self.uid}/{name}'):
- msg += f'\n
SubFolders: {folders}'
- msg += f'\n
Files: {files}'
- msg += f'\n\n
cc: {self.tag}'
- buttons = ButtonMaker()
- buttons.buildbutton("☁️ Drive Link", link)
- LOGGER.info(f'Done Uploading {name}')
- if INDEX_URL is not None:
- url_path = rutils.quote(f'{name}')
- share_url = f'{INDEX_URL}/{url_path}'
- if ospath.isdir(f'{DOWNLOAD_DIR}/{self.uid}/{name}'):
- share_url += '/'
- buttons.buildbutton("⚡ Index Link", share_url)
- else:
- buttons.buildbutton("⚡ Index Link", share_url)
- if VIEW_LINK:
- share_urls = f'{INDEX_URL}/{url_path}?a=view'
- buttons.buildbutton("🌐 View Link", share_urls)
- sendMarkup(msg, self.bot, self.message, InlineKeyboardMarkup(buttons.build_menu(2)))
- if self.isQbit and self.seed and not self.extract:
- if self.isZip:
- try:
- osremove(f'{DOWNLOAD_DIR}{self.uid}/{name}')
- except:
- pass
- return
- clean_download(f'{DOWNLOAD_DIR}{self.uid}')
- with download_dict_lock:
- try:
- del download_dict[self.uid]
- except Exception as e:
- LOGGER.error(str(e))
- count = len(download_dict)
- if count == 0:
- self.clean()
- else:
- update_all_messages()
-
- def onUploadError(self, error):
- e_str = error.replace('<', '').replace('>', '')
- clean_download(f'{DOWNLOAD_DIR}{self.uid}')
- with download_dict_lock:
- try:
- del download_dict[self.uid]
- except Exception as e:
- LOGGER.error(str(e))
- count = len(download_dict)
- sendMessage(f"{self.tag} {e_str}", self.bot, self.message)
- if count == 0:
- self.clean()
- else:
- update_all_messages()
-
- if not self.isPrivate and INCOMPLETE_TASK_NOTIFIER and DB_URI is not None:
- DbManger().rm_complete_task(self.message.link)
-
-def _mirror(bot, message, isZip=False, extract=False, isQbit=False, isLeech=False, pswd=None, multi=0, select=False, seed=False):
- mesg = message.text.split('\n')
- message_args = mesg[0].split(maxsplit=1)
- name_args = mesg[0].split('|', maxsplit=1)
- index = 1
-
- if len(message_args) > 1:
- args = mesg[0].split(maxsplit=3)
- if "s" in [x.strip() for x in args]:
- select = True
- index += 1
- if "d" in [x.strip() for x in args]:
- seed = True
- index += 1
- message_args = mesg[0].split(maxsplit=index)
- if len(message_args) > index:
- link = message_args[index].strip()
- if link.isdigit():
- multi = int(link)
- link = ''
- elif link.startswith(("|", "pswd:")):
- link = ''
- else:
- link = ''
- else:
- link = ''
-
- if len(name_args) > 1:
- name = name_args[1]
- name = name.split(' pswd:')[0]
- name = name.strip()
- else:
- name = ''
-
- link = re_split(r"pswd:|\|", link)[0]
- link = link.strip()
-
- pswd_arg = mesg[0].split(' pswd: ')
- if len(pswd_arg) > 1:
- pswd = pswd_arg[1]
-
- if message.from_user.username:
- tag = f"@{message.from_user.username}"
- else:
- tag = message.from_user.mention_html(message.from_user.first_name)
-
- reply_to = message.reply_to_message
- if reply_to is not None:
- file = None
- media_array = [reply_to.document, reply_to.video, reply_to.audio]
- for i in media_array:
- if i is not None:
- file = i
- break
-
- if not reply_to.from_user.is_bot:
- if reply_to.from_user.username:
- tag = f"@{reply_to.from_user.username}"
- else:
- tag = reply_to.from_user.mention_html(reply_to.from_user.first_name)
-
- if not is_url(link) and not is_magnet(link) or len(link) == 0:
- if file is None:
- reply_text = reply_to.text.split(maxsplit=1)[0].strip()
- if is_url(reply_text) or is_magnet(reply_text):
- link = reply_text
- elif file.mime_type != "application/x-bittorrent" and not isQbit:
- listener = MirrorListener(bot, message, isZip, extract, isQbit, isLeech, pswd, tag)
- Thread(target=TelegramDownloadHelper(listener).add_download, args=(message, f'{DOWNLOAD_DIR}{listener.uid}/', name)).start()
- if multi > 1:
- sleep(4)
- nextmsg = type('nextmsg', (object, ), {'chat_id': message.chat_id, 'message_id': message.reply_to_message.message_id + 1})
- nextmsg = sendMessage(message_args[0], bot, nextmsg)
- nextmsg.from_user.id = message.from_user.id
- multi -= 1
- sleep(4)
- Thread(target=_mirror, args=(bot, nextmsg, isZip, extract, isQbit, isLeech, pswd, multi)).start()
- return
- else:
- link = file.get_file().file_path
-
- if not is_url(link) and not is_magnet(link) and not ospath.exists(link):
- help_msg = "
Send link along with command line:"
- if isQbit:
- help_msg += "\n\n
Bittorrent selection:"
- help_msg += "\n
/cmd s {link} or by replying to {file/link}"
- help_msg += "\n\n
Qbittorrent seed:"
- help_msg += "\n
/qbcmd d {link} or by replying to {file/link}. "
- help_msg += "Sure you can use seed and select perfix together with qbcmds."
- help_msg += "\n\n
Multi links only by replying to first link/file:"
- help_msg += "\n
/command 10(number of links/files)"
- else:
- help_msg += "\n
/cmd {link} |newname pswd: xx [zip/unzip]"
- help_msg += "\n\n
By replying to link/file:"
- help_msg += "\n
/cmd |newname pswd: xx [zip/unzip]"
- help_msg += "\n\n
Direct link authorization:"
- help_msg += "\n
/cmd {link} |newname pswd: xx\nusername\npassword"
- help_msg += "\n\n
Bittorrent selection:"
- help_msg += "\n
/cmd s {link} or by replying to {file/link}"
- help_msg += "\n\n
Multi links only by replying to first link/file:"
- help_msg += "\n
/command 10(number of links/files)"
- return sendMessage(help_msg, bot, message)
-
- LOGGER.info(link)
-
- if not is_mega_link(link) and not isQbit and not is_magnet(link) \
- and not is_gdrive_link(link) and not link.endswith('.torrent'):
- content_type = get_content_type(link)
- if content_type is None or re_match(r'text/html|text/plain', content_type):
- try:
- link = direct_link_generator(link)
- LOGGER.info(f"Generated link: {link}")
- except DirectDownloadLinkException as e:
- LOGGER.info(str(e))
- if str(e).startswith('ERROR:'):
- return sendMessage(str(e), bot, message)
- elif isQbit and not is_magnet(link):
- if link.endswith('.torrent') or "https://api.telegram.org/file/" in link:
- content_type = None
- else:
- content_type = get_content_type(link)
- if content_type is None or re_match(r'application/x-bittorrent|application/octet-stream', content_type):
- try:
- resp = rget(link, timeout=10, headers = {'user-agent': 'Wget/1.12'})
- if resp.status_code == 200:
- file_name = str(time()).replace(".", "") + ".torrent"
- with open(file_name, "wb") as t:
- t.write(resp.content)
- link = str(file_name)
- else:
- return sendMessage(f"{tag} ERROR: link got HTTP response: {resp.status_code}", bot, message)
- except Exception as e:
- error = str(e).replace('<', ' ').replace('>', ' ')
- if error.startswith('No connection adapters were found for'):
- link = error.split("'")[1]
- else:
- LOGGER.error(str(e))
- return sendMessage(tag + " " + error, bot, message)
- else:
- msg = "Qb commands for torrents only. if you are trying to dowload torrent then report."
- return sendMessage(msg, bot, message)
-
-
- listener = MirrorListener(bot, message, isZip, extract, isQbit, isLeech, pswd, tag, select, seed)
-
- if is_gdrive_link(link):
- if not isZip and not extract and not isLeech:
- gmsg = f"Use /{BotCommands.CloneCommand} to clone Google Drive file/folder\n\n"
- gmsg += f"Use /{BotCommands.ZipMirrorCommand} to make zip of Google Drive folder\n\n"
- gmsg += f"Use /{BotCommands.UnzipMirrorCommand} to extracts Google Drive archive folder/file"
- sendMessage(gmsg, bot, message)
- else:
- Thread(target=add_gd_download, args=(link, listener, name)).start()
- elif is_mega_link(link):
- Thread(target=add_mega_download, args=(link, f'{DOWNLOAD_DIR}{listener.uid}/', listener, name)).start()
- elif isQbit and (is_magnet(link) or ospath.exists(link)):
- Thread(target=QbDownloader(listener).add_qb_torrent, args=(link, f'{DOWNLOAD_DIR}{listener.uid}', select)).start()
- else:
- if len(mesg) > 1:
- try:
- ussr = mesg[1]
- except:
- ussr = ''
- try:
- pssw = mesg[2]
- except:
- pssw = ''
- auth = f"{ussr}:{pssw}"
- auth = "Basic " + b64encode(auth.encode()).decode('ascii')
- else:
- auth = ''
- Thread(target=add_aria2c_download, args=(link, f'{DOWNLOAD_DIR}{listener.uid}', listener, name, auth, select)).start()
-
- if multi > 1:
- sleep(4)
- nextmsg = type('nextmsg', (object, ), {'chat_id': message.chat_id, 'message_id': message.reply_to_message.message_id + 1})
- msg = message_args[0]
- if len(mesg) > 2:
- msg += '\n' + mesg[1] + '\n' + mesg[2]
- nextmsg = sendMessage(msg, bot, nextmsg)
- nextmsg.from_user.id = message.from_user.id
- multi -= 1
- sleep(4)
- Thread(target=_mirror, args=(bot, nextmsg, isZip, extract, isQbit, isLeech, pswd, multi, select, seed)).start()
-
-
-def mirror(update, context):
- _mirror(context.bot, update.message)
-
-def unzip_mirror(update, context):
- _mirror(context.bot, update.message, extract=True)
-
-def zip_mirror(update, context):
- _mirror(context.bot, update.message, True)
-
-def qb_mirror(update, context):
- _mirror(context.bot, update.message, isQbit=True)
-
-def qb_unzip_mirror(update, context):
- _mirror(context.bot, update.message, extract=True, isQbit=True)
-
-def qb_zip_mirror(update, context):
- _mirror(context.bot, update.message, True, isQbit=True)
-
-def leech(update, context):
- _mirror(context.bot, update.message, isLeech=True)
-
-def unzip_leech(update, context):
- _mirror(context.bot, update.message, extract=True, isLeech=True)
-
-def zip_leech(update, context):
- _mirror(context.bot, update.message, True, isLeech=True)
-
-def qb_leech(update, context):
- _mirror(context.bot, update.message, isQbit=True, isLeech=True)
-
-def qb_unzip_leech(update, context):
- _mirror(context.bot, update.message, extract=True, isQbit=True, isLeech=True)
-
-def qb_zip_leech(update, context):
- _mirror(context.bot, update.message, True, isQbit=True, isLeech=True)
-
-mirror_handler = CommandHandler(BotCommands.MirrorCommand, mirror,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-unzip_mirror_handler = CommandHandler(BotCommands.UnzipMirrorCommand, unzip_mirror,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-zip_mirror_handler = CommandHandler(BotCommands.ZipMirrorCommand, zip_mirror,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-qb_mirror_handler = CommandHandler(BotCommands.QbMirrorCommand, qb_mirror,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-qb_unzip_mirror_handler = CommandHandler(BotCommands.QbUnzipMirrorCommand, qb_unzip_mirror,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-qb_zip_mirror_handler = CommandHandler(BotCommands.QbZipMirrorCommand, qb_zip_mirror,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-leech_handler = CommandHandler(BotCommands.LeechCommand, leech,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-unzip_leech_handler = CommandHandler(BotCommands.UnzipLeechCommand, unzip_leech,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-zip_leech_handler = CommandHandler(BotCommands.ZipLeechCommand, zip_leech,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-qb_leech_handler = CommandHandler(BotCommands.QbLeechCommand, qb_leech,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-qb_unzip_leech_handler = CommandHandler(BotCommands.QbUnzipLeechCommand, qb_unzip_leech,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-qb_zip_leech_handler = CommandHandler(BotCommands.QbZipLeechCommand, qb_zip_leech,
- filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-
-dispatcher.add_handler(mirror_handler)
-dispatcher.add_handler(unzip_mirror_handler)
-dispatcher.add_handler(zip_mirror_handler)
-dispatcher.add_handler(qb_mirror_handler)
-dispatcher.add_handler(qb_unzip_mirror_handler)
-dispatcher.add_handler(qb_zip_mirror_handler)
-dispatcher.add_handler(leech_handler)
-dispatcher.add_handler(unzip_leech_handler)
-dispatcher.add_handler(zip_leech_handler)
-dispatcher.add_handler(qb_leech_handler)
-dispatcher.add_handler(qb_unzip_leech_handler)
-dispatcher.add_handler(qb_zip_leech_handler)
diff --git a/bot/modules/mirror_leech.py b/bot/modules/mirror_leech.py
new file mode 100644
index 00000000..609490e6
--- /dev/null
+++ b/bot/modules/mirror_leech.py
@@ -0,0 +1,292 @@
+from base64 import b64encode
+from re import match as re_match, split as re_split
+from time import sleep
+from os import path as ospath
+from threading import Thread
+from telegram.ext import CommandHandler
+
+from bot import dispatcher, DOWNLOAD_DIR, LOGGER
+from bot.helper.ext_utils.bot_utils import is_url, is_magnet, is_mega_link, is_gdrive_link, get_content_type
+from bot.helper.ext_utils.exceptions import DirectDownloadLinkException
+from bot.helper.mirror_utils.download_utils.aria2_download import add_aria2c_download
+from bot.helper.mirror_utils.download_utils.gd_downloader import add_gd_download
+from bot.helper.mirror_utils.download_utils.qbit_downloader import QbDownloader
+from bot.helper.mirror_utils.download_utils.mega_downloader import add_mega_download
+from bot.helper.mirror_utils.download_utils.direct_link_generator import direct_link_generator
+from bot.helper.mirror_utils.download_utils.telegram_downloader import TelegramDownloadHelper
+from bot.helper.telegram_helper.bot_commands import BotCommands
+from bot.helper.telegram_helper.filters import CustomFilters
+from bot.helper.telegram_helper.message_utils import sendMessage
+from .listener import MirrorLeechListener
+
+
+def _mirror_leech(bot, message, isZip=False, extract=False, isQbit=False, isLeech=False, multi=0):
+ mesg = message.text.split('\n')
+ message_args = mesg[0].split(maxsplit=1)
+ name_args = mesg[0].split('|', maxsplit=1)
+ index = 1
+ ratio = None
+ seed_time = None
+ select = False
+ seed = False
+
+ if len(message_args) > 1:
+ args = mesg[0].split(maxsplit=3)
+ for x in args:
+ x = x.strip()
+ if x == 's':
+ select = True
+ index += 1
+ elif x == 'd':
+ seed = True
+ index += 1
+ elif x.startswith('d:'):
+ seed = True
+ index += 1
+ dargs = x.split(':')
+ ratio = dargs[1] if dargs[1] else None
+ if len(dargs) == 3:
+ seed_time = dargs[2] if dargs[2] else None
+ message_args = mesg[0].split(maxsplit=index)
+ if len(message_args) > index:
+ link = message_args[index].strip()
+ if link.isdigit():
+ if multi == 0:
+ multi = int(link)
+ link = ''
+ elif link.startswith(("|", "pswd:")):
+ link = ''
+ else:
+ link = ''
+ else:
+ link = ''
+
+ if len(name_args) > 1:
+ name = name_args[1]
+ name = name.split(' pswd:')[0]
+ name = name.strip()
+ else:
+ name = ''
+
+ link = re_split(r"pswd:|\|", link)[0]
+ link = link.strip()
+
+ pswd_arg = mesg[0].split(' pswd: ')
+ if len(pswd_arg) > 1:
+ pswd = pswd_arg[1]
+ else:
+ pswd = None
+
+ if message.from_user.username:
+ tag = f"@{message.from_user.username}"
+ else:
+ tag = message.from_user.mention_html(message.from_user.first_name)
+
+ reply_to = message.reply_to_message
+ if reply_to is not None:
+ file_ = next((i for i in [reply_to.document, reply_to.video, reply_to.audio, reply_to.photo] if i), None)
+ if not reply_to.from_user.is_bot:
+ if reply_to.from_user.username:
+ tag = f"@{reply_to.from_user.username}"
+ else:
+ tag = reply_to.from_user.mention_html(reply_to.from_user.first_name)
+ if len(link) == 0 or not is_url(link) and not is_magnet(link):
+ if file_ is None:
+ reply_text = reply_to.text.split(maxsplit=1)[0].strip()
+ if is_url(reply_text) or is_magnet(reply_text):
+ link = reply_to.text.strip()
+ elif isinstance(file_, list):
+ link = file_[-1].get_file().file_path
+ elif not isQbit and file_.mime_type != "application/x-bittorrent":
+ listener = MirrorLeechListener(bot, message, isZip, extract, isQbit, isLeech, pswd, tag)
+ Thread(target=TelegramDownloadHelper(listener).add_download, args=(message, f'{DOWNLOAD_DIR}{listener.uid}/', name)).start()
+ if multi > 1:
+ sleep(4)
+ nextmsg = type('nextmsg', (object, ), {'chat_id': message.chat_id, 'message_id': message.reply_to_message.message_id + 1})
+ nextmsg = sendMessage(message.text, bot, nextmsg)
+ nextmsg.from_user.id = message.from_user.id
+ multi -= 1
+ sleep(4)
+ Thread(target=_mirror_leech, args=(bot, nextmsg, isZip, extract, isQbit, isLeech, multi)).start()
+ return
+ else:
+ link = file_.get_file().file_path
+
+ if not is_url(link) and not is_magnet(link) and not ospath.exists(link):
+ help_msg = "
Send link along with command line:"
+ if isQbit:
+ help_msg += "\n\n
Bittorrent selection:"
+ help_msg += "\n
/cmd s {link} or by replying to {file/link}"
+ help_msg += "\n\n
Qbittorrent seed:"
+ help_msg += "\n
/qbcmd d {link} or by replying to {file/link}.\n"
+ help_msg += "To specify ratio and seed time. Ex: d:0.7:10 (ratio and time) or d:0.7 "
+ help_msg += "(only ratio) or d::10 (only time) where time in minutes"
+ help_msg += "\n\n
Multi links only by replying to first link/file:"
+ help_msg += "\n
/command 10(number of links/files)"
+ else:
+ help_msg += "\n
/cmd {link} |newname pswd: xx [zip/unzip]"
+ help_msg += "\n\n
By replying to link/file:"
+ help_msg += "\n
/cmd |newname pswd: xx [zip/unzip]"
+ help_msg += "\n\n
Direct link authorization:"
+ help_msg += "\n
/cmd {link} |newname pswd: xx\nusername\npassword"
+ help_msg += "\n\n
Bittorrent selection:"
+ help_msg += "\n
/cmd s {link} or by replying to {file/link}"
+ help_msg += "\n\n
Bittorrent seed:"
+ help_msg += "\n
/qbcmd d {link} or by replying to {file/link}.\n"
+ help_msg += "To specify ratio and seed time. Ex: d:0.7:10 (ratio and time) or d:0.7 "
+ help_msg += "(only ratio) or d::10 (only time) where time in minutes"
+ help_msg += "\n\n
Multi links only by replying to first link/file:"
+ help_msg += "\n
/command 10(number of links/files)"
+ return sendMessage(help_msg, bot, message)
+
+ LOGGER.info(link)
+
+ if not is_mega_link(link) and not isQbit and not is_magnet(link) \
+ and not is_gdrive_link(link) and not link.endswith('.torrent'):
+ content_type = get_content_type(link)
+ if content_type is None or re_match(r'text/html|text/plain', content_type):
+ try:
+ link = direct_link_generator(link)
+ LOGGER.info(f"Generated link: {link}")
+ except DirectDownloadLinkException as e:
+ LOGGER.info(str(e))
+ if str(e).startswith('ERROR:'):
+ return sendMessage(str(e), bot, message)
+ elif isQbit and not is_magnet(link):
+ if link.endswith('.torrent') or "https://api.telegram.org/file/" in link:
+ content_type = None
+ else:
+ content_type = get_content_type(link)
+ if content_type is None or re_match(r'application/x-bittorrent|application/octet-stream', content_type):
+ try:
+ resp = rget(link, timeout=10, headers = {'user-agent': 'Wget/1.12'})
+ if resp.status_code == 200:
+ file_name = str(time()).replace(".", "") + ".torrent"
+ with open(file_name, "wb") as t:
+ t.write(resp.content)
+ link = str(file_name)
+ else:
+ return sendMessage(f"{tag} ERROR: link got HTTP response: {resp.status_code}", bot, message)
+ except Exception as e:
+ error = str(e).replace('<', ' ').replace('>', ' ')
+ if error.startswith('No connection adapters were found for'):
+ link = error.split("'")[1]
+ else:
+ LOGGER.error(str(e))
+ return sendMessage(tag + " " + error, bot, message)
+ else:
+ msg = "Qb commands for torrents only. if you are trying to dowload torrent then report."
+ return sendMessage(msg, bot, message)
+
+ listener = MirrorLeechListener(bot, message, isZip, extract, isQbit, isLeech, pswd, tag, select, seed)
+
+ if is_gdrive_link(link):
+ if not isZip and not extract and not isLeech:
+ gmsg = f"Use /{BotCommands.CloneCommand} to clone Google Drive file/folder\n\n"
+ gmsg += f"Use /{BotCommands.ZipMirrorCommand[0]} to make zip of Google Drive folder\n\n"
+ gmsg += f"Use /{BotCommands.UnzipMirrorCommand[0]} to extracts Google Drive archive folder/file"
+ sendMessage(gmsg, bot, message)
+ else:
+ Thread(target=add_gd_download, args=(link, f'{DOWNLOAD_DIR}{listener.uid}', listener, name)).start()
+ elif is_mega_link(link):
+ Thread(target=add_mega_download, args=(link, f'{DOWNLOAD_DIR}{listener.uid}/', listener, name)).start()
+ elif isQbit and (is_magnet(link) or ospath.exists(link)):
+ Thread(target=QbDownloader(listener).add_qb_torrent, args=(link, f'{DOWNLOAD_DIR}{listener.uid}',
+ select, ratio, seed_time)).start()
+ else:
+ if len(mesg) > 1:
+ ussr = mesg[1]
+ if len(mesg) > 2:
+ pssw = mesg[2]
+ else:
+ pssw = ''
+ auth = f"{ussr}:{pssw}"
+ auth = "Basic " + b64encode(auth.encode()).decode('ascii')
+ else:
+ auth = ''
+ Thread(target=add_aria2c_download, args=(link, f'{DOWNLOAD_DIR}{listener.uid}', listener, name,
+ auth, select, ratio, seed_time)).start()
+
+ if multi > 1:
+ sleep(4)
+ nextmsg = type('nextmsg', (object, ), {'chat_id': message.chat_id, 'message_id': message.reply_to_message.message_id + 1})
+ nextmsg = sendMessage(message.text, bot, nextmsg)
+ nextmsg.from_user.id = message.from_user.id
+ multi -= 1
+ sleep(4)
+ Thread(target=_mirror_leech, args=(bot, nextmsg, isZip, extract, isQbit, isLeech, multi)).start()
+
+
+def mirror(update, context):
+ _mirror_leech(context.bot, update.message)
+
+def unzip_mirror(update, context):
+ _mirror_leech(context.bot, update.message, extract=True)
+
+def zip_mirror(update, context):
+ _mirror_leech(context.bot, update.message, True)
+
+def qb_mirror(update, context):
+ _mirror_leech(context.bot, update.message, isQbit=True)
+
+def qb_unzip_mirror(update, context):
+ _mirror_leech(context.bot, update.message, extract=True, isQbit=True)
+
+def qb_zip_mirror(update, context):
+ _mirror_leech(context.bot, update.message, True, isQbit=True)
+
+def leech(update, context):
+ _mirror_leech(context.bot, update.message, isLeech=True)
+
+def unzip_leech(update, context):
+ _mirror_leech(context.bot, update.message, extract=True, isLeech=True)
+
+def zip_leech(update, context):
+ _mirror_leech(context.bot, update.message, True, isLeech=True)
+
+def qb_leech(update, context):
+ _mirror_leech(context.bot, update.message, isQbit=True, isLeech=True)
+
+def qb_unzip_leech(update, context):
+ _mirror_leech(context.bot, update.message, extract=True, isQbit=True, isLeech=True)
+
+def qb_zip_leech(update, context):
+ _mirror_leech(context.bot, update.message, True, isQbit=True, isLeech=True)
+
+mirror_handler = CommandHandler(BotCommands.MirrorCommand, mirror,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+unzip_mirror_handler = CommandHandler(BotCommands.UnzipMirrorCommand, unzip_mirror,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+zip_mirror_handler = CommandHandler(BotCommands.ZipMirrorCommand, zip_mirror,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+qb_mirror_handler = CommandHandler(BotCommands.QbMirrorCommand, qb_mirror,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+qb_unzip_mirror_handler = CommandHandler(BotCommands.QbUnzipMirrorCommand, qb_unzip_mirror,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+qb_zip_mirror_handler = CommandHandler(BotCommands.QbZipMirrorCommand, qb_zip_mirror,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+leech_handler = CommandHandler(BotCommands.LeechCommand, leech,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+unzip_leech_handler = CommandHandler(BotCommands.UnzipLeechCommand, unzip_leech,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+zip_leech_handler = CommandHandler(BotCommands.ZipLeechCommand, zip_leech,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+qb_leech_handler = CommandHandler(BotCommands.QbLeechCommand, qb_leech,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+qb_unzip_leech_handler = CommandHandler(BotCommands.QbUnzipLeechCommand, qb_unzip_leech,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+qb_zip_leech_handler = CommandHandler(BotCommands.QbZipLeechCommand, qb_zip_leech,
+ filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
+
+dispatcher.add_handler(mirror_handler)
+dispatcher.add_handler(unzip_mirror_handler)
+dispatcher.add_handler(zip_mirror_handler)
+dispatcher.add_handler(qb_mirror_handler)
+dispatcher.add_handler(qb_unzip_mirror_handler)
+dispatcher.add_handler(qb_zip_mirror_handler)
+dispatcher.add_handler(leech_handler)
+dispatcher.add_handler(unzip_leech_handler)
+dispatcher.add_handler(zip_leech_handler)
+dispatcher.add_handler(qb_leech_handler)
+dispatcher.add_handler(qb_unzip_leech_handler)
+dispatcher.add_handler(qb_zip_leech_handler)
diff --git a/bot/modules/search.py b/bot/modules/search.py
index 325ca6cf..159f92e5 100644
--- a/bot/modules/search.py
+++ b/bot/modules/search.py
@@ -1,18 +1,19 @@
from requests import get as rget
-from time import sleep
+from time import time
from threading import Thread
from html import escape
from urllib.parse import quote
from telegram import InlineKeyboardMarkup
from telegram.ext import CommandHandler, CallbackQueryHandler
+from os import remove
from bot import dispatcher, LOGGER, SEARCH_API_LINK, SEARCH_PLUGINS, get_client, SEARCH_LIMIT
-from bot.helper.ext_utils.telegraph_helper import telegraph
-from bot.helper.telegram_helper.message_utils import editMessage, sendMessage, sendMarkup
+from bot.helper.telegram_helper.message_utils import editMessage, sendMessage, sendMarkup, deleteMessage, sendFile
from bot.helper.telegram_helper.filters import CustomFilters
from bot.helper.telegram_helper.bot_commands import BotCommands
from bot.helper.ext_utils.bot_utils import get_readable_file_size
from bot.helper.telegram_helper import button_build
+from bot.helper.ext_utils.html_helper import html_template
if SEARCH_PLUGINS is not None:
PLUGINS = []
@@ -44,8 +45,6 @@ SITES = {
"all": "All"
}
-TELEGRAPH_LIMIT = 300
-
def torser(update, context):
user_id = update.message.from_user.id
@@ -109,12 +108,12 @@ def torserbut(update, context):
editMessage(f"
Searching for {key}\nTorrent Site:- {SITES.get(site)}", message)
else:
editMessage(f"
Searching for {key}\nTorrent Site:- {site.capitalize()}", message)
- Thread(target=_search, args=(key, site, message, method)).start()
+ Thread(target=_search, args=(context.bot, key, site, message, method)).start()
else:
query.answer()
editMessage("Search has been canceled!", message)
-def _search(key, site, message, method):
+def _search(bot, key, site, message, method):
if method.startswith('api'):
if method == 'apisearch':
LOGGER.info(f"API Searching: {key} from {site}")
@@ -139,20 +138,20 @@ def _search(key, site, message, method):
search_results = resp.json()
if "error" in search_results.keys():
return editMessage(f"No result found for
{key}\nTorrent Site:-
{SITES.get(site)}", message)
- msg = f"
Found {min(search_results['total'], TELEGRAPH_LIMIT)}"
+ cap = f"
Found {search_results['total']}"
if method == 'apitrend':
- msg += f"
trending result(s)\nTorrent Site:- {SITES.get(site)}"
+ cap += f"
trending results\nTorrent Site:- {SITES.get(site)}"
elif method == 'apirecent':
- msg += f"
recent result(s)\nTorrent Site:- {SITES.get(site)}"
+ cap += f"
recent results\nTorrent Site:- {SITES.get(site)}"
else:
- msg += f"
result(s) for {key}\nTorrent Site:- {SITES.get(site)}"
+ cap += f"
results for {key}\nTorrent Site:- {SITES.get(site)}"
search_results = search_results['data']
except Exception as e:
return editMessage(str(e), message)
else:
LOGGER.info(f"PLUGINS Searching: {key} from {site}")
client = get_client()
- search = client.search_start(pattern=str(key), plugins=str(site), category='all')
+ search = client.search_start(pattern=key, plugins=site, category='all')
search_id = search.id
while True:
result_status = client.search_status(search_id=search_id)
@@ -164,78 +163,69 @@ def _search(key, site, message, method):
total_results = dict_search_results.total
if total_results == 0:
return editMessage(f"No result found for
{key}\nTorrent Site:-
{site.capitalize()}", message)
- msg = f"
Found {min(total_results, TELEGRAPH_LIMIT)}"
- msg += f"
result(s) for {key}\nTorrent Site:- {site.capitalize()}"
- link = _getResult(search_results, key, message, method)
- buttons = button_build.ButtonMaker()
- buttons.buildbutton("🔎 VIEW", link)
- button = InlineKeyboardMarkup(buttons.build_menu(1))
- editMessage(msg, message, button)
+ cap = f"
Found {total_results}"
+ cap += f"
results for {key}\nTorrent Site:- {site.capitalize()}"
+ hmsg = _getResult(search_results, key, message, method)
+ name = f"{method}_{key}_{site}_{time()}.html"
+ with open(name, "w", encoding='utf-8') as f:
+ f.write(html_template.replace('{msg}', hmsg).replace('{title}', f'{method}_{key}_{site}'))
+ deleteMessage(bot, message)
+ sendFile(bot, message.reply_to_message, name, cap)
+ remove(name)
if not method.startswith('api'):
client.search_delete(search_id=search_id)
def _getResult(search_results, key, message, method):
- telegraph_content = []
if method == 'apirecent':
- msg = "
API Recent Results
"
+ msg = '
API Recent Results
'
elif method == 'apisearch':
- msg = f"
API Search Result(s) For {key}
"
+ msg = f'
API Search Results For {key}
'
elif method == 'apitrend':
- msg = "
API Trending Results
"
+ msg = '
API Trending Results
'
else:
- msg = f"
PLUGINS Search Result(s) For {key}
"
- for index, result in enumerate(search_results, start=1):
+ msg = f'
PLUGINS Search Results For {key}
'
+ for result in search_results:
+ msg += '
'
if method.startswith('api'):
if 'name' in result.keys():
- msg += f"{escape(result['name'])}
"
+ msg += f""
if 'torrents' in result.keys():
for subres in result['torrents']:
- msg += f"Quality: {subres['quality']} | Type: {subres['type']} | Size: {subres['size']}
"
+ msg += f"Quality: {subres['quality']} | "
+ mag += f"Type: {subres['type']} | Size: {subres['size']}"
if 'torrent' in subres.keys():
- msg += f"Direct Link
"
+ msg += "Direct Link"
elif 'magnet' in subres.keys():
- msg += f"Share Magnet to Telegram
"
+ msg += "Share Magnet to Telegram"
msg += '
'
else:
- msg += f"Size: {result['size']}
"
+ msg += f"Size: {result['size']}"
try:
- msg += f"Seeders: {result['seeders']} | Leechers: {result['leechers']}
"
+ msg += f"Seeders: {result['seeders']} | "
+ msg += f"Leechers: {result['leechers']}"
except:
pass
if 'torrent' in result.keys():
- msg += f"Direct Link
"
+ msg += f"Direct Link"
elif 'magnet' in result.keys():
- msg += f"Share Magnet to Telegram
"
+ msg += f"Share Magnet to Telegram"
else:
- msg += f"{escape(result.fileName)}
"
- msg += f"Size: {get_readable_file_size(result.fileSize)}
"
- msg += f"Seeders: {result.nbSeeders} | Leechers: {result.nbLeechers}
"
+ msg += f""
+ msg += f"Size: {get_readable_file_size(result.fileSize)}"
+ msg += f"Seeders: {result.nbSeeders} | "
+ msg += f"Leechers: {result.nbLeechers}"
link = result.fileUrl
if link.startswith('magnet:'):
- msg += f"Share Magnet to Telegram
"
+ msg += f"Share Magnet to Telegram"
else:
- msg += f"Direct Link
"
-
- if len(msg.encode('utf-8')) > 39000:
- telegraph_content.append(msg)
- msg = ""
-
- if index == TELEGRAPH_LIMIT:
- break
-
- if msg != "":
- telegraph_content.append(msg)
-
- editMessage(f"Creating {len(telegraph_content)} Telegraph pages.", message)
- path = [telegraph.create_page(
- title='Mirror-leech-bot Torrent Search',
- content=content
- )["path"] for content in telegraph_content]
- sleep(0.5)
- if len(path) > 1:
- editMessage(f"Editing {len(telegraph_content)} Telegraph pages.", message)
- telegraph.edit_telegraph(path, telegraph_content)
- return f"https://telegra.ph/{path[0]}"
+ msg += f"Direct Link"
+ msg += ''
+ return msg
def _api_buttons(user_id, method):
buttons = button_build.ButtonMaker()
diff --git a/bot/modules/watch.py b/bot/modules/ytdlp.py
similarity index 90%
rename from bot/modules/watch.py
rename to bot/modules/ytdlp.py
index 614e8ac8..19de7c68 100644
--- a/bot/modules/watch.py
+++ b/bot/modules/ytdlp.py
@@ -11,11 +11,11 @@ from bot.helper.ext_utils.bot_utils import get_readable_file_size, is_url
from bot.helper.mirror_utils.download_utils.youtube_dl_download_helper import YoutubeDLHelper
from bot.helper.telegram_helper.bot_commands import BotCommands
from bot.helper.telegram_helper.filters import CustomFilters
-from .mirror import MirrorListener
+from .listener import MirrorLeechListener
listener_dict = {}
-def _watch(bot, message, isZip=False, isLeech=False, multi=0):
+def _ytdl(bot, message, isZip=False, isLeech=False, multi=0):
mssg = message.text
user_id = message.from_user.id
msg_id = message.message_id
@@ -24,7 +24,8 @@ def _watch(bot, message, isZip=False, isLeech=False, multi=0):
if len(link) > 1:
link = link[1].strip()
if link.strip().isdigit():
- multi = int(link)
+ if multi == 0:
+ multi = int(link)
link = ''
elif link.strip().startswith(("|", "pswd:", "args:")):
link = ''
@@ -82,7 +83,7 @@ def _watch(bot, message, isZip=False, isLeech=False, multi=0):
help_msg += "\n\nCheck all arguments from this
FILE."
return sendMessage(help_msg, bot, message)
- listener = MirrorListener(bot, message, isZip, isLeech=isLeech, pswd=pswd, tag=tag)
+ listener = MirrorLeechListener(bot, message, isZip, isLeech=isLeech, pswd=pswd, tag=tag)
buttons = button_build.ButtonMaker()
best_video = "bv*+ba/b"
best_audio = "ba/b"
@@ -159,11 +160,11 @@ def _watch(bot, message, isZip=False, isLeech=False, multi=0):
if multi > 1:
sleep(4)
nextmsg = type('nextmsg', (object, ), {'chat_id': message.chat_id, 'message_id': message.reply_to_message.message_id + 1})
- nextmsg = sendMessage(mssg.split(' ')[0], bot, nextmsg)
+ nextmsg = sendMessage(mssg, bot, nextmsg)
nextmsg.from_user.id = message.from_user.id
multi -= 1
sleep(4)
- Thread(target=_watch, args=(bot, nextmsg, isZip, isLeech, multi)).start()
+ Thread(target=_ytdl, args=(bot, nextmsg, isZip, isLeech, multi)).start()
def _qual_subbuttons(task_id, qual, msg):
buttons = button_build.ButtonMaker()
@@ -271,30 +272,30 @@ def _auto_cancel(msg, msg_id):
except:
pass
-def watch(update, context):
- _watch(context.bot, update.message)
+def ytdl(update, context):
+ _ytdl(context.bot, update.message)
-def watchZip(update, context):
- _watch(context.bot, update.message, True)
+def ytdlZip(update, context):
+ _ytdl(context.bot, update.message, True)
-def leechWatch(update, context):
- _watch(context.bot, update.message, isLeech=True)
+def ytdlleech(update, context):
+ _ytdl(context.bot, update.message, isLeech=True)
-def leechWatchZip(update, context):
- _watch(context.bot, update.message, True, True)
+def ytdlZipleech(update, context):
+ _ytdl(context.bot, update.message, True, True)
-watch_handler = CommandHandler(BotCommands.WatchCommand, watch,
+ytdl_handler = CommandHandler(BotCommands.YtdlCommand, ytdl,
filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-zip_watch_handler = CommandHandler(BotCommands.ZipWatchCommand, watchZip,
+ytdl_zip_handler = CommandHandler(BotCommands.YtdlZipCommand, ytdlZip,
filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-leech_watch_handler = CommandHandler(BotCommands.LeechWatchCommand, leechWatch,
+ytdl_leech_handler = CommandHandler(BotCommands.YtdlLeechCommand, ytdlleech,
filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
-leech_zip_watch_handler = CommandHandler(BotCommands.LeechZipWatchCommand, leechWatchZip,
+ytdl_zip_leech_handler = CommandHandler(BotCommands.YtdlZipLeechCommand, ytdlZipleech,
filters=CustomFilters.authorized_chat | CustomFilters.authorized_user, run_async=True)
quality_handler = CallbackQueryHandler(select_format, pattern="qu", run_async=True)
-dispatcher.add_handler(watch_handler)
-dispatcher.add_handler(zip_watch_handler)
-dispatcher.add_handler(leech_watch_handler)
-dispatcher.add_handler(leech_zip_watch_handler)
+dispatcher.add_handler(ytdl_handler)
+dispatcher.add_handler(ytdl_zip_handler)
+dispatcher.add_handler(ytdl_leech_handler)
+dispatcher.add_handler(ytdl_zip_leech_handler)
dispatcher.add_handler(quality_handler)
diff --git a/extract b/extract
deleted file mode 100755
index 7500a407..00000000
--- a/extract
+++ /dev/null
@@ -1,199 +0,0 @@
-#!/bin/bash
-
-if [ $# -lt 1 ]; then
- echo "Usage: $(basename $0) FILES"
- exit 1
-fi
-
-extract() {
- arg="$1"
- cd "$(dirname "$arg")" || exit
- case "$arg" in
- *.tar.bz2)
- tar xjf "$arg" --one-top-level
- local code=$?
- ;;
- *.tar.gz)
- tar xzf "$arg" --one-top-level
- local code=$?
- ;;
- *.bz2)
- bunzip2 "$arg"
- local code=$?
- ;;
- *.gz)
- gunzip "$arg"
- local code=$?
- ;;
- *.tar)
- tar xf "$arg" --one-top-level
- local code=$?
- ;;
- *.tbz2)
- (tar xjf "$arg" --one-top-level)
- local code=$?
- ;;
- *.tgz)
- tar xzf "$arg" --one-top-level
- local code=$?
- ;;
- *.tar.xz)
- a_dir=$(expr "$arg" : '\(.*\).tar.xz')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.zip)
- a_dir=$(expr "$arg" : '\(.*\).zip')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.7z)
- a_dir=$(expr "$arg" : '\(.*\).7z')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.Z)
- uncompress "$arg"
- local code=$?
- ;;
- *.rar)
- a_dir=$(expr "$arg" : '\(.*\).rar')
- mkdir "$a_dir"
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.iso)
- a_dir=$(expr "$arg" : '\(.*\).iso')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.wim)
- a_dir=$(expr "$arg" : '\(.*\).wim')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.cab)
- a_dir=$(expr "$arg" : '\(.*\).cab')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.apm)
- a_dir=$(expr "$arg" : '\(.*\).apm')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.arj)
- a_dir=$(expr "$arg" : '\(.*\).arj')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.chm)
- a_dir=$(expr "$arg" : '\(.*\).chm')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.cpio)
- a_dir=$(expr "$arg" : '\(.*\).cpio')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.cramfs)
- a_dir=$(expr "$arg" : '\(.*\).cramfs')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.deb)
- a_dir=$(expr "$arg" : '\(.*\).deb')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.dmg)
- a_dir=$(expr "$arg" : '\(.*\).dmg')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.fat)
- a_dir=$(expr "$arg" : '\(.*\).fat')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.hfs)
- a_dir=$(expr "$arg" : '\(.*\).hfs')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.lzh)
- a_dir=$(expr "$arg" : '\(.*\).lzh')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.lzma)
- a_dir=$(expr "$arg" : '\(.*\).lzma')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.lzma2)
- a_dir=$(expr "$arg" : '\(.*\).lzma2')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.mbr)
- a_dir=$(expr "$arg" : '\(.*\).mbr')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.msi)
- a_dir=$(expr "$arg" : '\(.*\).msi')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.mslz)
- a_dir=$(expr "$arg" : '\(.*\).mslz')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.nsis)
- a_dir=$(expr "$arg" : '\(.*\).nsis')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.ntfs)
- a_dir=$(expr "$arg" : '\(.*\).ntfs')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.rpm)
- a_dir=$(expr "$arg" : '\(.*\).rpm')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.squashfs)
- a_dir=$(expr "$arg" : '\(.*\).squashfs')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.udf)
- a_dir=$(expr "$arg" : '\(.*\).udf')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.vhd)
- a_dir=$(expr "$arg" : '\(.*\).vhd')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *.xar)
- a_dir=$(expr "$arg" : '\(.*\).xar')
- 7z x "$arg" -o"$a_dir"
- local code=$?
- ;;
- *)
- echo "'$arg' cannot be extracted via extract()" 1>&2
- exit 1
- ;;
- esac
- cd - || exit $?
- exit $code
-}
-
-extract "$1"
diff --git a/pextract b/pextract
deleted file mode 100644
index 7c53e08a..00000000
--- a/pextract
+++ /dev/null
@@ -1,200 +0,0 @@
-#!/bin/bash
-
-if [ $# -lt 1 ]; then
- echo "Usage: $(basename $0) FILES"
- exit 1
-fi
-
-extract() {
- arg="$1"
- pswd="$2"
- cd "$(dirname "$arg")" || exit
- case "$arg" in
- *.tar.bz2)
- tar xjf "$arg" --one-top-level
- local code=$?
- ;;
- *.tar.gz)
- tar xzf "$arg" --one-top-level
- local code=$?
- ;;
- *.bz2)
- bunzip2 "$arg"
- local code=$?
- ;;
- *.gz)
- gunzip "$arg"
- local code=$?
- ;;
- *.tar)
- tar xf "$arg" --one-top-level
- local code=$?
- ;;
- *.tbz2)
- (tar xjf "$arg" --one-top-level)
- local code=$?
- ;;
- *.tgz)
- tar xzf "$arg" --one-top-level
- local code=$?
- ;;
- *.tar.xz)
- a_dir=$(expr "$arg" : '\(.*\).tar.xz')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.zip)
- a_dir=$(expr "$arg" : '\(.*\).zip')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.7z)
- a_dir=$(expr "$arg" : '\(.*\).7z')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.Z)
- uncompress "$arg"
- local code=$?
- ;;
- *.rar)
- a_dir=$(expr "$arg" : '\(.*\).rar')
- mkdir "$a_dir"
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.iso)
- a_dir=$(expr "$arg" : '\(.*\).iso')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.wim)
- a_dir=$(expr "$arg" : '\(.*\).wim')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.cab)
- a_dir=$(expr "$arg" : '\(.*\).cab')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.apm)
- a_dir=$(expr "$arg" : '\(.*\).apm')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.arj)
- a_dir=$(expr "$arg" : '\(.*\).arj')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.chm)
- a_dir=$(expr "$arg" : '\(.*\).chm')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.cpio)
- a_dir=$(expr "$arg" : '\(.*\).cpio')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.cramfs)
- a_dir=$(expr "$arg" : '\(.*\).cramfs')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.deb)
- a_dir=$(expr "$arg" : '\(.*\).deb')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.dmg)
- a_dir=$(expr "$arg" : '\(.*\).dmg')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.fat)
- a_dir=$(expr "$arg" : '\(.*\).fat')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.hfs)
- a_dir=$(expr "$arg" : '\(.*\).hfs')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.lzh)
- a_dir=$(expr "$arg" : '\(.*\).lzh')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.lzma)
- a_dir=$(expr "$arg" : '\(.*\).lzma')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.lzma2)
- a_dir=$(expr "$arg" : '\(.*\).lzma2')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.mbr)
- a_dir=$(expr "$arg" : '\(.*\).mbr')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.msi)
- a_dir=$(expr "$arg" : '\(.*\).msi')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.mslz)
- a_dir=$(expr "$arg" : '\(.*\).mslz')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.nsis)
- a_dir=$(expr "$arg" : '\(.*\).nsis')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.ntfs)
- a_dir=$(expr "$arg" : '\(.*\).ntfs')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.rpm)
- a_dir=$(expr "$arg" : '\(.*\).rpm')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.squashfs)
- a_dir=$(expr "$arg" : '\(.*\).squashfs')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.udf)
- a_dir=$(expr "$arg" : '\(.*\).udf')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.vhd)
- a_dir=$(expr "$arg" : '\(.*\).vhd')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *.xar)
- a_dir=$(expr "$arg" : '\(.*\).xar')
- 7z x "$arg" -o"$a_dir" -p"$pswd"
- local code=$?
- ;;
- *)
- echo "'$arg' cannot be extracted via extract()" 1>&2
- exit 1
- ;;
- esac
- cd - || exit $?
- exit $code
-}
-
-extract "$1" "$2"
diff --git a/qBittorrent/config/qBittorrent.conf b/qBittorrent/config/qBittorrent.conf
index ebb5e677..eeb73910 100644
--- a/qBittorrent/config/qBittorrent.conf
+++ b/qBittorrent/config/qBittorrent.conf
@@ -4,7 +4,8 @@ Accepted=true
[BitTorrent]
Session\AsyncIOThreadsCount=16
Session\MultiConnectionsPerIp=true
-Session\SlowTorrentsDownloadRate=50
+Session\SlowTorrentsDownloadRate=2
+Session\SlowTorrentsUploadRate=2
Session\SlowTorrentsInactivityTimer=600
Session\GlobalMaxSeedingMinutes=-1
diff --git a/web/wserver.py b/web/wserver.py
index f483c0da..7eff7658 100644
--- a/web/wserver.py
+++ b/web/wserver.py
@@ -8,6 +8,8 @@ from web.nodes import make_tree
app = Flask(__name__)
+aria2 = ariaAPI(ariaClient(host="http://localhost", port=6800, secret=""))
+
basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
handlers=[FileHandler('log.txt'), StreamHandler()],
level=INFO)
@@ -711,7 +713,6 @@ def list_torrent_contents(id_):
cont = make_tree(res)
client.auth_log_out()
else:
- aria2 = ariaAPI(ariaClient(host="http://localhost", port=6800, secret=""))
res = aria2.client.get_files(id_)
cont = make_tree(res, True)
return page.replace("{My_content}", cont[0]).replace("{form_url}", f"/app/files/{id_}?pin_code={pincode}")
@@ -764,7 +765,6 @@ def set_priority(id_):
resume = resume.strip(",")
- aria2 = ariaAPI(ariaClient(host="http://localhost", port=6800, secret=""))
res = aria2.client.change_option(id_, {'select-file': resume})
if res == "OK":
LOGGER.info(f"Verified! Gid: {id_}")