peg-this 3.0.2__py3-none-any.whl → 3.0.6__py3-none-any.whl

peg_this/utils/ffmpeg_utils.py

@@ -0,0 +1,134 @@
+
+import subprocess
+import logging
+import sys
+
+import ffmpeg
+from rich.console import Console
+from rich.progress import Progress, SpinnerColumn, BarColumn, TextColumn
+
+console = Console()
+
+
+def check_ffmpeg_ffprobe():
+    """
+    Checks if ffmpeg and ffprobe executables are available in the system's PATH.
+    ffmpeg-python requires this.
+    """
+    try:
+        # The library does this internally, but we can provide a clearer error message.
+        subprocess.check_call(['ffmpeg', '-version'], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        subprocess.check_call(['ffprobe', '-version'], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    except FileNotFoundError:
+        console.print("[bold red]Error: ffmpeg and ffprobe not found.[/bold red]")
+        if sys.platform == "win32":
+            console.print("You can install it using Chocolatey: [bold]choco install ffmpeg[/bold]")
+            console.print("Or Scoop: [bold]scoop install ffmpeg[/bold]")
+        elif sys.platform == "darwin":
+            console.print("You can install it using Homebrew: [bold]brew install ffmpeg[/bold]")
+        else:
+            console.print("You can install it using your system's package manager, e.g., [bold]sudo apt update && sudo apt install ffmpeg[/bold] on Debian/Ubuntu.")
+        console.print("Please ensure its location is in your system's PATH.")
+        sys.exit(1)
+
+
+def run_command(stream_spec, description="Processing...", show_progress=False):
+    """
+    Runs an ffmpeg command using ffmpeg-python.
+    - For simple commands, it runs directly.
+    - For commands with a progress bar, it generates the ffmpeg arguments,
+      runs them as a subprocess, and parses stderr to show progress,
+      mimicking the logic from the original script for accuracy.
+    """
+    console.print(f"[bold cyan]{description}[/bold cyan]")
+    
+    # Get the full command arguments from the ffmpeg-python stream object
+    args = stream_spec.get_args()
+    full_command = ['ffmpeg'] + args
+    logging.info(f"Executing command: {' '.join(full_command)}")
+
+    if not show_progress:
+        try:
+            # Use ffmpeg.run() for simple, non-progress tasks. It's cleaner.
+            out, err = ffmpeg.run(stream_spec, capture_stdout=True, capture_stderr=True, quiet=True)
+            logging.info("Command successful (no progress bar).")
+            return out.decode('utf-8')
+        except ffmpeg.Error as e:
+            error_message = e.stderr.decode('utf-8')
+            console.print("[bold red]An error occurred:[/bold red]")
+            console.print(error_message)
+            logging.error(f"ffmpeg error:{error_message}")
+            return None
+    else:
+        # For the progress bar, we must run ffmpeg as a subprocess and parse stderr.
+        duration = 0
+        try:
+            # Find the primary input file from the command arguments to probe it.
+            input_file_path = None
+            for i, arg in enumerate(full_command):
+                if arg == '-i' and i + 1 < len(full_command):
+                    # This is a robust way to find the first input file.
+                    input_file_path = full_command[i+1]
+                    break
+            
+            if input_file_path:
+                probe_info = ffmpeg.probe(input_file_path)
+                duration = float(probe_info['format']['duration'])
+            else:
+                logging.warning("Could not find input file in command to determine duration for progress bar.")
+
+        except (ffmpeg.Error, KeyError) as e:
+            console.print(f"[bold yellow]Warning: Could not determine video duration for progress bar.[/bold yellow]")
+            logging.warning(f"Could not probe for duration: {e}")
+
+        with Progress(
+            SpinnerColumn(),
+            TextColumn("[progress.description]{task.description}"),
+            BarColumn(),
+            TextColumn("[progress.percentage]{task.percentage:>3.0f}%"),
+            console=console,
+        ) as progress:
+            task = progress.add_task(description, total=100)
+            
+            # Run the command as a subprocess to capture stderr in real-time
+            process = subprocess.Popen(
+                full_command,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                universal_newlines=True,
+                encoding='utf-8'
+            )
+
+            for line in process.stderr:
+                logging.debug(f"ffmpeg stderr: {line.strip()}")
+                if "time=" in line and duration > 0:
+                    try:
+                        time_str = line.split("time=")[1].split(" ")[0].strip()
+                        h, m, s_parts = time_str.split(':')
+                        s = float(s_parts)
+                        elapsed_time = int(h) * 3600 + int(m) * 60 + s
+                        percent_complete = (elapsed_time / duration) * 100
+                        progress.update(task, completed=min(percent_complete, 100))
+                    except Exception:
+                        pass # Ignore any parsing errors
+
+            process.wait()
+            progress.update(task, completed=100)
+            
+            if process.returncode != 0:
+                # The error was already logged line-by-line, but we can add a final message.
+                log_file = logging.getLogger().handlers[0].baseFilename
+                console.print(f"[bold red]An error occurred during processing. Check {log_file} for details.[/bold red]")
+                return None
+        
+        logging.info("Command successful (with progress bar).")
+        return "Success"
+
+
+def has_audio_stream(file_path):
+    """Check if the media file has an audio stream."""
+    try:
+        probe = ffmpeg.probe(file_path, select_streams='a')
+        return 'streams' in probe and len(probe['streams']) > 0
+    except ffmpeg.Error:
+        return False

peg_this/utils/ui_utils.py

@@ -0,0 +1,43 @@
+
+import os
+from pathlib import Path
+
+import questionary
+from rich.console import Console
+
+try:
+    import tkinter as tk
+    from tkinter import filedialog
+except ImportError:
+    tk = None
+
+console = Console()
+
+
+def get_media_files():
+    """Scan the current directory for media files."""
+    media_extensions = [".mkv", ".mp4", ".avi", ".mov", ".webm", ".flv", ".wmv", ".mp3", ".flac", ".wav", ".ogg", ".gif"]
+    files = [f for f in os.listdir('.') if os.path.isfile(f) and Path(f).suffix.lower() in media_extensions]
+    return files
+
+
+def select_media_file():
+    """Display a menu to select a media file, or open a file picker."""
+    media_files = get_media_files()
+    if not media_files:
+        console.print("[bold yellow]No media files found in this directory.[/bold yellow]")
+        if tk and questionary.confirm("Would you like to select a file from another location?").ask():
+            root = tk.Tk()
+            root.withdraw()
+            file_path = filedialog.askopenfilename(
+                title="Select a media file",
+                filetypes=[("Media Files", "*.mkv *.mp4 *.avi *.mov *.webm *.flv *.wmv *.mp3 *.flac *.wav *.ogg *.gif"), ("All Files", "*.*")]
+            )
+            return file_path if file_path else None
+        return None
+
+    choices = media_files + [questionary.Separator(), "Back"]
+    file = questionary.select("Select a media file to process:", choices=choices, use_indicator=True).ask()
+    
+    # Return the absolute path to prevent "file not found" errors
+    return os.path.abspath(file) if file and file != "Back" else None

{peg_this-3.0.2.dist-info → peg_this-3.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: peg_this
-Version: 3.0.2
+Version: 3.0.6
 Summary: A powerful, intuitive command-line video editor suite, built on FFmpeg.
 Author-email: Hariharen S S <thisishariharen@gmail.com>
 Project-URL: Homepage, https://github.com/hariharen9/ffmpeg-this
@@ -26,6 +26,10 @@ Dynamic: license-file
 
 A powerful and user-friendly batch script for converting, manipulating, and inspecting media files using the power of FFmpeg. This script provides a simple command-line menu to perform common audio and video tasks without needing to memorize complex FFmpeg commands.
 
+
+<img src="/assets/peg.gif" width="720">
+
+
 ## ✨ Features
 
 - **Inspect Media Properties**: View detailed information about video and audio streams, including codecs, resolution, frame rate, bitrates, and more.
@@ -39,8 +43,27 @@ A powerful and user-friendly batch script for converting, manipulating, and insp
 
 
 ## 🚀 Usage
+### Prerequisite: Install FFmpeg
+
+> `peg_this` uses a library called `ffmpeg-python` which acts as a controller for the main FFmpeg program. It does not include FFmpeg itself. Therefore, you must have FFmpeg installed on your system and available in your terminal's PATH.
+
+For **macOS** users, the easiest way to install it is with [Homebrew](https://brew.sh/):
+```bash
+brew install ffmpeg
+```
+
+For **Windows** users, you can use a package manager like [Chocolatey](https://chocolatey.org/) or [Scoop](https://scoop.sh/):
+```bash
+# Using Chocolatey
+choco install ffmpeg
+
+# Using Scoop
+scoop install ffmpeg
+```
+
+For other systems, please see the official download page: **[ffmpeg.org/download.html](https://ffmpeg.org/download.html)**
 
-There are three ways to use `peg_this`:
+There are ***three*** ways to use `peg_this`:
 
 ### 1. Pip Install (Recommended)
 

peg_this-3.0.6.dist-info/RECORD

@@ -0,0 +1,19 @@
+peg_this/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+peg_this/peg_this.py,sha256=BqJutllqMHqE312WgPdnd3r9-msKKOfZp9ykV0ejUm8,3739
+peg_this/features/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+peg_this/features/audio.py,sha256=O0_lRjzolualxWVXiSiEke5vOpPGwkz0FbMQy5brO58,1533
+peg_this/features/batch.py,sha256=72yXjNfvg-SCxjtaacFzFudmZ8Yd7_rLpvJzbwB8UdA,5178
+peg_this/features/convert.py,sha256=KLtmN6PGhqIEmY3j40Ur9xqC-rZREu8UBM9Ura1-20o,3865
+peg_this/features/crop.py,sha256=bzlzfFM41OFX_3KEMNOAJu9XvM-6Mmf_AApJ1rN8oaU,3986
+peg_this/features/inspect.py,sha256=xtlmedpsMYQ-vLsnDNzUF59TiZDFLGMyL_wtd8PgiHg,2535
+peg_this/features/join.py,sha256=w_HUsuBv0g9nlOjVvdkeT9dDwxOEvdyvsYCwNocwl98,3414
+peg_this/features/trim.py,sha256=2ZWSNpl4DzSCPt6q7FZloQc-Rga5cX8IW69MfzAZysE,863
+peg_this/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+peg_this/utils/ffmpeg_utils.py,sha256=cxm5SyZYHw99dqaIiNZ9qhdltyvMwecs1FYHXjChwR8,5883
+peg_this/utils/ui_utils.py,sha256=cntxp8m7en_irFNEK5TcrPotYVZr-hzF_QdcXUIBA2Y,1584
+peg_this-3.0.6.dist-info/licenses/LICENSE,sha256=WL1MklYSco7KZvDjbf191tIKOxWQdekqda7dDJc6Wn8,1067
+peg_this-3.0.6.dist-info/METADATA,sha256=_bWnDbXC0vGoPDImzXVd05AyITf8XZIuvJK-uK7K_3c,4310
+peg_this-3.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+peg_this-3.0.6.dist-info/entry_points.txt,sha256=9GVTFuE1w_wgY-Tz3--wI5j52BAKrt4atphVD8ioHhQ,52
+peg_this-3.0.6.dist-info/top_level.txt,sha256=kSS5jZg3KN2kJqYZwMvQnI4gvlFxsUNzIm3QJsbKFdc,9
+peg_this-3.0.6.dist-info/RECORD,,

peg_this-3.0.2.dist-info/RECORD

@@ -1,8 +0,0 @@
-peg_this/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-peg_this/peg_this.py,sha256=JSxUYO5zjg0URu6SG3I1AWio_yQ8t8wKW9yqP-pRbUY,30228
-peg_this-3.0.2.dist-info/licenses/LICENSE,sha256=WL1MklYSco7KZvDjbf191tIKOxWQdekqda7dDJc6Wn8,1067
-peg_this-3.0.2.dist-info/METADATA,sha256=suW_MI8jSHxZXEKKOsrry7QnXoHyVZ0s6ZFljc-RKy8,3527
-peg_this-3.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-peg_this-3.0.2.dist-info/entry_points.txt,sha256=9GVTFuE1w_wgY-Tz3--wI5j52BAKrt4atphVD8ioHhQ,52
-peg_this-3.0.2.dist-info/top_level.txt,sha256=kSS5jZg3KN2kJqYZwMvQnI4gvlFxsUNzIm3QJsbKFdc,9
-peg_this-3.0.2.dist-info/RECORD,,