Merge pull request #46 from XronTrix10/preview

Stable update for Video split support

Author: XronTrix10

colab_leecher/__main__.py

@@ -184,6 +184,10 @@ async def handle_options(client, callback_query):
     elif callback_query.data == "video":
         keyboard = InlineKeyboardMarkup(
             [
+                [
+                    InlineKeyboardButton("Split Videos", callback_data="split-true"),
+                    InlineKeyboardButton("Zip Videos", callback_data="split-false"),
+                ],
                 [
                     InlineKeyboardButton("Convert", callback_data="convert-true"),
                     InlineKeyboardButton(
@@ -202,7 +206,7 @@ async def handle_options(client, callback_query):
             ]
         )
         await callback_query.message.edit_text(
-            f"CHOOSE YOUR DESIRED OPTION ⚙️ »\n\n╭⌬ CONVERT » <code>{BOT.Setting.convert_video}</code>\n├⌬ OUTPUT FORMAT » <code>{BOT.Options.video_out}</code>\n╰⌬ OUTPUT QUALITY » <code>{BOT.Setting.convert_quality}</code>",
+            f"CHOOSE YOUR DESIRED OPTION ⚙️ »\n\n╭⌬ CONVERT » <code>{BOT.Setting.convert_video}</code>\n├⌬ SPLIT » <code>{BOT.Setting.split_video}</code>\n├⌬ OUTPUT FORMAT » <code>{BOT.Options.video_out}</code>\n╰⌬ OUTPUT QUALITY » <code>{BOT.Setting.convert_quality}</code>",
             reply_markup=keyboard,
         )
     elif callback_query.data == "caption":
@@ -269,6 +273,14 @@ async def handle_options(client, callback_query):
         await send_settings(
             client, callback_query.message, callback_query.message.id, False
         )
+    elif callback_query.data in ["split-true", "split-false"]:
+        BOT.Options.is_split = True if callback_query.data == "split-true" else False
+        BOT.Setting.split_video = (
+            "Split Videos" if callback_query.data == "split-true" else "Zip Videos"
+        )
+        await send_settings(
+            client, callback_query.message, callback_query.message.id, False
+        )
     elif callback_query.data in [
         "convert-true",
         "convert-false",

colab_leecher/utility/converters.py

@@ -2,6 +2,7 @@
 
 
 import os
+import json
 import GPUtil
 import shutil
 import logging
@@ -14,6 +15,7 @@
 from colab_leecher.utility.variables import BOT, MSG, BotTimes, Paths, Messages
 from colab_leecher.utility.helper import (
     getSize,
+    fileType,
     keyboard,
     multipartArchive,
     sizeUnit,
@@ -127,7 +129,12 @@ async def sizeChecker(file_path, remove: bool):
         ):
             await splitArchive(file_path, max_size)
         else:
-            await archive(file_path, True, remove)
+            f_type = fileType(file_path)
+            if f_type == "video" and BOT.Options.is_split:
+                # TODO: Store the size in a constant variable
+                await splitVideo(file_path, 2000, remove)
+            else:
+                await archive(file_path, True, remove)
             await sleep(2)
         return True
     else:
@@ -157,18 +164,19 @@ async def archive(path, is_split, remove: bool):
     else:
         cmd = f'7z a -mx=0 -tzip -p{BOT.Options.zip_pswd} {split} "{Paths.temp_zpath}/{name}.zip" {path}'
     proc = subprocess.Popen(cmd, shell=True)
-    total = sizeUnit(getSize(path))
+    total_size = getSize(path)
+    total_in_unit = sizeUnit(total_size)
     while proc.poll() is None:
         speed_string, eta, percentage = speedETA(
-            BotTimes.task_start, getSize(Paths.temp_zpath), getSize(path)
+            BotTimes.task_start, getSize(Paths.temp_zpath), total_size
         )
         await status_bar(
             Messages.status_head,
             speed_string,
             percentage,
             getTime(eta),
             sizeUnit(getSize(Paths.temp_zpath)),
-            total,
+            total_in_unit,
             "Xr-Zipp 🔒",
         )
         await sleep(1)
@@ -286,3 +294,55 @@ async def splitArchive(file_path, max_size):
             # Get next chunk
             chunk = f.read(max_size)
             i += 1  # Increment chunk counter
+
+
+async def splitVideo(file_path, max_size, remove: bool):
+    global Paths, BOT, MSG, Messages
+    _, filename = ospath.split(file_path)
+    just_name, extension = ospath.splitext(filename)
+
+    # FFmpeg command to get video information in JSON format
+    cmd = ["ffprobe", "-v", "quiet", "-print_format", "json", "-show_format", file_path]
+
+    bitrate = None
+    try:
+        # Run the command and get output
+        output = subprocess.check_output(cmd)
+        video_info = json.loads(output)
+        # Get bitrate in bits per second
+        bitrate = float(video_info["format"]["bit_rate"])
+    except subprocess.CalledProcessError:
+        logging.error("Error: Could not get video bitrate")
+        bitrate = 1000
+
+    # Convert target size from MB to bits
+    target_size_bits = max_size * 8 * 1024 * 1024
+
+    # Calculate duration in seconds
+    duration = int(target_size_bits / bitrate)
+
+    cmd = f'ffmpeg -i {file_path} -c copy -f segment -segment_time {duration} -reset_timestamps 1 "{Paths.temp_zpath}/{just_name}.part%03d{extension}"'
+
+    Messages.status_head = f"<b>✂️ SPLITTING » </b>\n\n<code>{filename}</code>\n"
+    BotTimes.task_start = datetime.now()
+
+    proc = subprocess.Popen(cmd, shell=True)
+    total_size = getSize(file_path)
+    total_in_unit = sizeUnit(total_size)
+    while proc.poll() is None:
+        speed_string, eta, percentage = speedETA(
+            BotTimes.task_start, getSize(Paths.temp_zpath), total_size
+        )
+        await status_bar(
+            Messages.status_head,
+            speed_string,
+            percentage,
+            getTime(eta),
+            sizeUnit(getSize(Paths.temp_zpath)),
+            total_in_unit,
+            "Xr-Split ✂️",
+        )
+        await sleep(1)
+
+    if remove:
+        os.remove(file_path)

colab_leecher/utility/helper.py

@@ -103,6 +103,9 @@ def fileType(file_path: str):
         ".ts": "video",
         ".m3u8": "video",
         ".webm": "video",
+        ".mpg": "video",
+        ".mpeg": "video",
+        ".mpeg4": "video",
         ".vob": "video",
         ".m4v": "video",
         ".mp3": "audio",
@@ -341,7 +344,7 @@ async def send_settings(client, message, msg_id, command: bool):
                 InlineKeyboardButton(
                     f"Set {up_mode.capitalize()}", callback_data=up_mode
                 ),
-                InlineKeyboardButton("Video Convert", callback_data="video"),
+                InlineKeyboardButton("Video Settings", callback_data="video"),
             ],
             [
                 InlineKeyboardButton("Caption Font", callback_data="caption"),
@@ -356,6 +359,7 @@ async def send_settings(client, message, msg_id, command: bool):
     )
     text = "**CURRENT BOT SETTINGS ⚙️ »**"
     text += f"\n\n╭⌬ UPLOAD » <i>{BOT.Setting.stream_upload}</i>"
+    text += f"\n├⌬ SPLIT » <i>{BOT.Setting.split_video}</i>"
     text += f"\n├⌬ CONVERT » <i>{BOT.Setting.convert_video}</i>"
     text += f"\n├⌬ CAPTION » <i>{BOT.Setting.caption}</i>"
     pr = "None" if BOT.Setting.prefix == "" else "Exists"

colab_leecher/utility/variables.py

@@ -12,16 +12,18 @@ class BOT:
     class Setting:
         stream_upload = "Media"
         convert_video = "Yes"
-        convert_quality = "High"
+        convert_quality = "Low"
         caption = "Monospace"
+        split_video = "Split Videos"
         prefix = ""
         suffix = ""
         thumbnail = False
 
     class Options:
         stream_upload = True
         convert_video = True
-        convert_quality = True
+        convert_quality = False
+        is_split = True
         caption = "code"
         video_out = "mp4"
         custom_name = ""