add /leech file

Author: SpEcHiDe

publicleechgroup/__init__.py

@@ -17,6 +17,7 @@
 
 # the logging things
 import logging
+import os
 from logging.handlers import RotatingFileHandler
 from time import time
 

publicleechgroup/helper_funcs/download_aria_p_n.py

@@ -213,7 +213,9 @@ async def call_apropriate_function(
     incoming_link,
     c_file_name,
     sent_message_to_update_tg_p,
-    is_zip
+    is_zip,
+    force_doc=False,
+    cfn=None
 ):
     if incoming_link.lower().startswith("magnet:"):
         sagtus, err_message = add_magnet(aria_instance, incoming_link, c_file_name)
@@ -263,7 +265,9 @@ async def call_apropriate_function(
         sent_message_to_update_tg_p,
         to_upload_file,
         user_id,
-        response
+        response,
+        force_doc=force_doc,
+        cfn=cfn
     )
     LOGGER.info(final_response)
     message_to_send = ""

publicleechgroup/helper_funcs/upload_to_tg.py

@@ -54,7 +54,9 @@ async def upload_to_tg(
     from_user,
     dict_contatining_uploaded_files,
     edit_media=False,
-    custom_caption=None
+    custom_caption=None,
+    force_doc=False,
+    cfn=None
 ):
     LOGGER.info(local_file_name)
     base_file_name = os.path.basename(local_file_name)
@@ -90,7 +92,9 @@ async def upload_to_tg(
                 from_user,
                 dict_contatining_uploaded_files,
                 edit_media,
-                caption_str
+                caption_str,
+                force_doc=force_doc,
+                cfn=cfn
             )
     else:
         if os.path.getsize(local_file_name) > TG_MAX_FILE_SIZE:
@@ -118,25 +122,40 @@ async def upload_to_tg(
                     message,
                     os.path.join(splitted_dir, le_file),
                     from_user,
-                    dict_contatining_uploaded_files
+                    dict_contatining_uploaded_files,
+                    force_doc=force_doc,
+                    cfn=cfn
                 )
         else:
             sent_message = await upload_single_file(
                 message,
                 local_file_name,
                 caption_str,
                 from_user,
-                edit_media
+                edit_media,
+                force_doc,
+                cfn
             )
             if sent_message is not None:
                 dict_contatining_uploaded_files[os.path.basename(local_file_name)] = sent_message.message_id
     # await message.delete()
     return dict_contatining_uploaded_files
 
 
-async def upload_single_file(message, local_file_name, caption_str, from_user, edit_media):
+async def upload_single_file(
+    message,
+    local_file_name,
+    caption_str,
+    from_user,
+    edit_media,
+    force_doc=False,
+    cfn=None
+):
     await asyncio.sleep(EDIT_SLEEP_TIME_OUT)
     sent_message = None
+    if cfn:
+        os.rename(local_file_name, cfn)
+        local_file_name = cfn
     start_time = time.time()
     #
     thumbnail_location = os.path.join(
@@ -151,7 +170,10 @@ async def upload_single_file(message, local_file_name, caption_str, from_user, e
         message_for_progress_display = await message.reply_text(
             "starting upload of {}".format(os.path.basename(local_file_name))
         )
-    if local_file_name.upper().endswith(("MKV", "MP4", "WEBM")):
+
+    if local_file_name.upper().endswith((
+        "M4V", "MP4", "MOV", "FLV", "WMV", "3GP", "MPEG", "WEBM", "MKV"
+    )) and not force_doc:
         metadata = extractMetadata(createParser(local_file_name))
         duration = 0
         if metadata.has("duration"):
@@ -230,7 +252,10 @@ async def upload_single_file(message, local_file_name, caption_str, from_user, e
             )
         if thumb is not None:
             os.remove(thumb)
-    elif local_file_name.upper().endswith(("MP3", "M4A", "M4B", "FLAC", "WAV")):
+
+    elif local_file_name.upper().endswith((
+        "MP3", "M4A", "M4B", "FLAC", "WAV", "AIF", "OGG", "AAC", "DTS"
+    )) and not force_doc:
         metadata = extractMetadata(createParser(local_file_name))
         duration = 0
         title = ""
@@ -285,6 +310,7 @@ async def upload_single_file(message, local_file_name, caption_str, from_user, e
             )
         if thumb is not None:
             os.remove(thumb)
+
     else:
         thumb_image_path = None
         if os.path.isfile(thumbnail_location):
@@ -330,5 +356,6 @@ async def upload_single_file(message, local_file_name, caption_str, from_user, e
 
     if message.message_id != message_for_progress_display.message_id:
         await message_for_progress_display.delete()
+
     os.remove(local_file_name)
     return sent_message

publicleechgroup/plugins/incoming_message_fn.py

@@ -77,7 +77,10 @@ async def leech_commandi_f(client, message):
         "checking",
         quote=True
     )
+    force_doc = False
     m_sgra = " ".join(message.command[1:])
+    if "file" in m_sgra.lower():
+        force_doc = True
     # get link from the incoming message
     dl_url, cf_name, _, _ = await extract_link(
         message.reply_to_message, "LEECH"
@@ -125,7 +128,9 @@ async def leech_commandi_f(client, message):
                 dl_url,
                 new_download_location,
                 m_,
-                is_zip
+                is_zip,
+                force_doc=force_doc,
+                cfn=cf_name
             )
             if not sagtus:
                 # if FAILED, display the error message