StreamingCommunity 1.7.6__py3-none-any.whl → 1.9.1__py3-none-any.whl

StreamingCommunity/{Src/Lib → Lib}/FFmpeg/capture.py

@@ -1,170 +1,170 @@
-# 16.04.24
-
-import re
-import logging
-import threading
-import subprocess
-
-
-# Internal utilities
-from StreamingCommunity.Src.Util.console import console
-from StreamingCommunity.Src.Util.os import internet_manager
-
-
-# Variable
-terminate_flag = threading.Event()
-
-
-def capture_output(process: subprocess.Popen, description: str) -> None:
-    """
-    Function to capture and print output from a subprocess.
-
-    Parameters:
-        - process (subprocess.Popen): The subprocess whose output is captured.
-        - description (str): Description of the command being executed.
-    """
-    try:
-
-        # Variable to store the length of the longest progress string
-        max_length = 0
-
-        for line in iter(process.stdout.readline, ''):
-            try:
-                line = line.strip()
-                if not line:
-                    continue
-
-                logging.info(f"FFMPEG line: {line}")
-
-                # Capture only error
-                if "rror" in str(line):
-                    console.log(f"[red]FFMPEG: {str(line).strip()}")
-
-                # Check if termination is requested
-                if terminate_flag.is_set():
-                    break
-
-                if "size=" in line:
-                    try:
-
-                        # Parse the output line to extract relevant information
-                        data = parse_output_line(line)
-
-                        if 'q' in data:
-                            is_end = (float(data.get('q', -1.0)) == -1.0)
-                            size_key = 'Lsize' if is_end else 'size'
-                            byte_size = int(re.findall(r'\d+', data.get(size_key, '0'))[0]) * 1000
-                        else:
-                            byte_size = int(re.findall(r'\d+', data.get('size', '0'))[0]) * 1000
-
-
-                        # Construct the progress string with formatted output information
-                        progress_string = (f"[yellow][FFmpeg] [white][{description}[white]]: "
-                                           f"([green]'speed': [yellow]{data.get('speed', 'N/A')}[white], "
-                                           f"[green]'size': [yellow]{internet_manager.format_file_size(byte_size)}[white])")
-                        max_length = max(max_length, len(progress_string))
-
-                        # Print the progress string to the console, overwriting the previous line
-                        console.print(progress_string.ljust(max_length), end="\r")
-
-                    except Exception as e:
-                        logging.error(f"Error parsing output line: {line} - {e}")
-
-            except Exception as e:
-                logging.error(f"Error processing line from subprocess: {e}")
-
-    except Exception as e:
-        logging.error(f"Error in capture_output: {e}")
-
-    finally:
-        try:
-            terminate_process(process)
-        except Exception as e:
-            logging.error(f"Error terminating process: {e}")
-
-
-def parse_output_line(line: str) -> dict:
-    """
-    Function to parse the output line and extract relevant information.
-
-    Parameters:
-        - line (str): The output line to parse.
-
-    Returns:
-        dict: A dictionary containing parsed information.
-    """
-    try:
-
-        data = {}
-
-        # Split the line by whitespace and extract key-value pairs
-        parts = line.replace("  ", "").replace("= ", "=").split()
-
-        for part in parts:
-            key_value = part.split('=')
-
-            if len(key_value) == 2:
-                key = key_value[0]
-                value = key_value[1]
-                data[key] = value
-
-        return data
-    
-    except Exception as e:
-        logging.error(f"Error parsing line: {line} - {e}")
-        return {}
-
-
-def terminate_process(process):
-    """
-    Function to terminate a subprocess if it's still running.
-
-    Parameters:
-        - process (subprocess.Popen): The subprocess to terminate.
-    """
-    try:
-        if process.poll() is None:  # Check if the process is still running
-            process.kill()
-    except Exception as e:
-        logging.error(f"Failed to terminate process: {e}")
-
-
-def capture_ffmpeg_real_time(ffmpeg_command: list, description: str) -> None:
-    """
-    Function to capture real-time output from ffmpeg process.
-
-    Parameters:
-        - ffmpeg_command (list): The command to execute ffmpeg.
-        - description (str): Description of the command being executed.
-    """
-
-    global terminate_flag
-
-    # Clear the terminate_flag before starting a new capture
-    terminate_flag.clear()
-
-    try:
-
-        # Start the ffmpeg process with subprocess.Popen
-        process = subprocess.Popen(ffmpeg_command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)
-
-        # Start a thread to capture and print output
-        output_thread = threading.Thread(target=capture_output, args=(process, description))
-        output_thread.start()
-
-        try:
-            # Wait for ffmpeg process to complete
-            process.wait()
-
-        except KeyboardInterrupt:
-            logging.error("Terminating ffmpeg process...")
-
-        except Exception as e:
-            logging.error(f"Error in ffmpeg process: {e}")
-            
-        finally:
-            terminate_flag.set()  # Signal the output capture thread to terminate
-            output_thread.join()  # Wait for the output capture thread to complete
-
-    except Exception as e:
-        logging.error(f"Failed to start ffmpeg process: {e}")
+# 16.04.24
+
+import re
+import logging
+import threading
+import subprocess
+
+
+# Internal utilities
+from StreamingCommunity.Util.console import console
+from StreamingCommunity.Util.os import internet_manager
+
+
+# Variable
+terminate_flag = threading.Event()
+
+
+def capture_output(process: subprocess.Popen, description: str) -> None:
+    """
+    Function to capture and print output from a subprocess.
+
+    Parameters:
+        - process (subprocess.Popen): The subprocess whose output is captured.
+        - description (str): Description of the command being executed.
+    """
+    try:
+
+        # Variable to store the length of the longest progress string
+        max_length = 0
+
+        for line in iter(process.stdout.readline, ''):
+            try:
+                line = line.strip()
+                if not line:
+                    continue
+
+                logging.info(f"FFMPEG line: {line}")
+
+                # Capture only error
+                if "rror" in str(line):
+                    console.log(f"[red]FFMPEG: {str(line).strip()}")
+
+                # Check if termination is requested
+                if terminate_flag.is_set():
+                    break
+
+                if "size=" in line:
+                    try:
+
+                        # Parse the output line to extract relevant information
+                        data = parse_output_line(line)
+
+                        if 'q' in data:
+                            is_end = (float(data.get('q', -1.0)) == -1.0)
+                            size_key = 'Lsize' if is_end else 'size'
+                            byte_size = int(re.findall(r'\d+', data.get(size_key, '0'))[0]) * 1000
+                        else:
+                            byte_size = int(re.findall(r'\d+', data.get('size', '0'))[0]) * 1000
+
+
+                        # Construct the progress string with formatted output information
+                        progress_string = (f"[yellow][FFmpeg] [white][{description}[white]]: "
+                                           f"([green]'speed': [yellow]{data.get('speed', 'N/A')}[white], "
+                                           f"[green]'size': [yellow]{internet_manager.format_file_size(byte_size)}[white])")
+                        max_length = max(max_length, len(progress_string))
+
+                        # Print the progress string to the console, overwriting the previous line
+                        console.print(progress_string.ljust(max_length), end="\r")
+
+                    except Exception as e:
+                        logging.error(f"Error parsing output line: {line} - {e}")
+
+            except Exception as e:
+                logging.error(f"Error processing line from subprocess: {e}")
+
+    except Exception as e:
+        logging.error(f"Error in capture_output: {e}")
+
+    finally:
+        try:
+            terminate_process(process)
+        except Exception as e:
+            logging.error(f"Error terminating process: {e}")
+
+
+def parse_output_line(line: str) -> dict:
+    """
+    Function to parse the output line and extract relevant information.
+
+    Parameters:
+        - line (str): The output line to parse.
+
+    Returns:
+        dict: A dictionary containing parsed information.
+    """
+    try:
+
+        data = {}
+
+        # Split the line by whitespace and extract key-value pairs
+        parts = line.replace("  ", "").replace("= ", "=").split()
+
+        for part in parts:
+            key_value = part.split('=')
+
+            if len(key_value) == 2:
+                key = key_value[0]
+                value = key_value[1]
+                data[key] = value
+
+        return data
+    
+    except Exception as e:
+        logging.error(f"Error parsing line: {line} - {e}")
+        return {}
+
+
+def terminate_process(process):
+    """
+    Function to terminate a subprocess if it's still running.
+
+    Parameters:
+        - process (subprocess.Popen): The subprocess to terminate.
+    """
+    try:
+        if process.poll() is None:  # Check if the process is still running
+            process.kill()
+    except Exception as e:
+        logging.error(f"Failed to terminate process: {e}")
+
+
+def capture_ffmpeg_real_time(ffmpeg_command: list, description: str) -> None:
+    """
+    Function to capture real-time output from ffmpeg process.
+
+    Parameters:
+        - ffmpeg_command (list): The command to execute ffmpeg.
+        - description (str): Description of the command being executed.
+    """
+
+    global terminate_flag
+
+    # Clear the terminate_flag before starting a new capture
+    terminate_flag.clear()
+
+    try:
+
+        # Start the ffmpeg process with subprocess.Popen
+        process = subprocess.Popen(ffmpeg_command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)
+
+        # Start a thread to capture and print output
+        output_thread = threading.Thread(target=capture_output, args=(process, description))
+        output_thread.start()
+
+        try:
+            # Wait for ffmpeg process to complete
+            process.wait()
+
+        except KeyboardInterrupt:
+            logging.error("Terminating ffmpeg process...")
+
+        except Exception as e:
+            logging.error(f"Error in ffmpeg process: {e}")
+            
+        finally:
+            terminate_flag.set()  # Signal the output capture thread to terminate
+            output_thread.join()  # Wait for the output capture thread to complete
+
+    except Exception as e:
+        logging.error(f"Failed to start ffmpeg process: {e}")