From e41d443c0cfeb67b0b519b35412c3c8559c94193 Mon Sep 17 00:00:00 2001
From: Anas <77075674+anasty17@users.noreply.github.com>
Date: Thu, 10 Jun 2021 04:27:47 +0300
Subject: [PATCH] stop duplicate mega and mega limit (#151)
* stop duplicate mega and mega limit
* fix msg not deleted
---
bot/__init__.py | 14 ++++++
.../download_utils/mega_downloader.py | 50 ++++++++++++++++---
bot/modules/clone.py | 3 ++
config_sample.env | 2 +
4 files changed, 63 insertions(+), 6 deletions(-)
diff --git a/bot/__init__.py b/bot/__init__.py
index 867663f..5f11211 100644
--- a/bot/__init__.py
+++ b/bot/__init__.py
@@ -187,6 +187,12 @@ try:
CLONE_LIMIT = None
except KeyError:
CLONE_LIMIT = None
+try:
+ MEGA_LIMIT = getConfig('MEGA_LIMIT')
+ if len(MEGA_LIMIT) == 0:
+ MEGA_LIMIT = None
+except KeyError:
+ MEGA_LIMIT = None
try:
BUTTON_FOUR_NAME = getConfig('BUTTON_FOUR_NAME')
BUTTON_FOUR_URL = getConfig('BUTTON_FOUR_URL')
@@ -219,6 +225,14 @@ try:
STOP_DUPLICATE_MIRROR = False
except KeyError:
STOP_DUPLICATE_MIRROR = False
+try:
+ STOP_DUPLICATE_MEGA = getConfig('STOP_DUPLICATE_MEGA')
+ if STOP_DUPLICATE_MEGA.lower() == 'true':
+ STOP_DUPLICATE_MEGA = True
+ else:
+ STOP_DUPLICATE_MEGA = False
+except KeyError:
+ STOP_DUPLICATE_MEGA = False
try:
VIEW_LINK = getConfig('VIEW_LINK')
if VIEW_LINK.lower() == 'true':
diff --git a/bot/helper/mirror_utils/download_utils/mega_downloader.py b/bot/helper/mirror_utils/download_utils/mega_downloader.py
index a0c8ee6..9c281b3 100644
--- a/bot/helper/mirror_utils/download_utils/mega_downloader.py
+++ b/bot/helper/mirror_utils/download_utils/mega_downloader.py
@@ -1,10 +1,12 @@
from bot import LOGGER, MEGA_API_KEY, download_dict_lock, download_dict, MEGA_EMAIL_ID, MEGA_PASSWORD
import threading
from mega import (MegaApi, MegaListener, MegaRequest, MegaTransfer, MegaError)
-from bot.helper.telegram_helper.message_utils import update_all_messages
+from bot.helper.telegram_helper.message_utils import update_all_messages, sendMarkup, deleteMessage, sendMessage
import os
-from bot.helper.ext_utils.bot_utils import new_thread, get_mega_link_type
+from bot.helper.ext_utils.bot_utils import new_thread, get_mega_link_type, get_readable_file_size
from bot.helper.mirror_utils.status_utils.mega_download_status import MegaDownloadStatus
+from bot.helper.mirror_utils.upload_utils.gdriveTools import GoogleDriveHelper
+from bot import MEGA_LIMIT, STOP_DUPLICATE_MEGA
import random
import string
@@ -151,9 +153,6 @@ class MegaDownloadHelper:
global listeners
mega_listener = MegaAppListener(executor.continue_event, listener)
listeners.append(mega_listener)
- with download_dict_lock:
- download_dict[listener.uid] = MegaDownloadStatus(mega_listener, listener)
- os.makedirs(path)
api.addListener(mega_listener)
if MEGA_EMAIL_ID is not None and MEGA_PASSWORD is not None:
executor.do(api.login, (MEGA_EMAIL_ID, MEGA_PASSWORD))
@@ -167,7 +166,46 @@ class MegaDownloadHelper:
folder_api.addListener(mega_listener)
executor.do(folder_api.loginToFolder, (mega_link,))
node = folder_api.authorizeNode(mega_listener.node)
-
+ if STOP_DUPLICATE_MEGA:
+ msg = sendMessage('Check the File/Folder if already in drive...', listener.bot, listener.update)
+ mname = node.getName()
+ if listener.isTar == True:
+ mname = mname + ".tar"
+ if listener.extract == True:
+ smsg = None
+ else:
+ gd = GoogleDriveHelper()
+ smsg, button = gd.drive_list(mname)
+ if smsg:
+ deleteMessage(listener.bot, msg)
+ msg1 = "File/Folder is already available in Drive.\nHere are the search results:"
+ sendMarkup(msg1, listener.bot, listener.update, button)
+ return
+ else:
+ deleteMessage(listener.bot, msg)
+ if MEGA_LIMIT is not None:
+ msg2 = sendMessage('Check the File/Folder size...', listener.bot, listener.update)
+ limit = MEGA_LIMIT
+ limit = limit.split(' ', maxsplit=1)
+ limitint = int(limit[0])
+ msg3 = f'Failed, Clone limit is {MEGA_LIMIT}.\nYour File/Folder size is {get_readable_file_size(api.getSize(node))}.'
+ if 'GB' in limit or 'gb' in limit:
+ if api.getSize(node) > limitint * 1024**3:
+ deleteMessage(listener.bot, msg2)
+ sendMessage(msg3, listener.bot, listener.update)
+ return
+ else:
+ deleteMessage(listener.bot, msg2)
+ elif 'TB' in limit or 'tb' in limit:
+ if api.getSize(node) > limitint * 1024**4:
+ deleteMessage(listener.bot, msg2)
+ sendMessage(msg3, listener.bot, listener.update)
+ return
+ else:
+ deleteMessage(listener.bot, msg2)
+ with download_dict_lock:
+ download_dict[listener.uid] = MegaDownloadStatus(mega_listener, listener)
+ os.makedirs(path)
gid = ''.join(random.SystemRandom().choices(string.ascii_letters + string.digits, k=8))
mega_listener.setValues(node.getName(), api.getSize(node), gid)
executor.do(api.startDownload,(node,path))
diff --git a/bot/modules/clone.py b/bot/modules/clone.py
index 7a18a78..ed34f7b 100644
--- a/bot/modules/clone.py
+++ b/bot/modules/clone.py
@@ -27,6 +27,9 @@ def cloneNode(update, context):
msg3 = "File/Folder is already available in Drive.\nHere are the search results:"
sendMarkup(msg3, context.bot, update, button)
return
+ else:
+ if CLONE_LIMIT is None:
+ deleteMessage(context.bot, msg1)
if CLONE_LIMIT is not None:
limit = CLONE_LIMIT
limit = limit.split(' ', maxsplit=1)
diff --git a/config_sample.env b/config_sample.env
index 888186b..0880d87 100644
--- a/config_sample.env
+++ b/config_sample.env
@@ -23,10 +23,12 @@ MEGA_PASSWORD = ""
BLOCK_MEGA_FOLDER = ""
BLOCK_MEGA_LINKS = ""
STOP_DUPLICATE_MIRROR = ""
+STOP_DUPLICATE_MEGA = ""
STOP_DUPLICATE_CLONE = ""
SHORTENER = ""
SHORTENER_API = ""
CLONE_LIMIT = "" #leave space between number and unit (TB or GB only)
+MEGA_LIMIT = ""
IMAGE_URL = "https://telegra.ph/file/db03910496f06094f1f7a.jpg"
# Change ENABLE_FILESIZE_LIMIT = True if you want to use MAX_TORRENT_SIZE
ENABLE_FILESIZE_LIMIT = false