useful python scripts

python
import subprocess
import sys

from tqdm import tqdm


class AudioExtractor:
    """
    Extract audio from a video file with a progress bar.

    Args:
        input_file (str): The input video file.
        output_file (str): The output audio file.
        codec (str, optional): The audio codec to use. Default is 'aac'.
        bitrate (str, optional): The audio bitrate. Default is '192k'.
        ffmpeg_loglevel (str, optional): Desired FFmpeg log level. Default is 'info'.
    """

    def __init__(self, input_file, output_file="", codec='aac', bitrate='192k', ffmpeg_loglevel="info"):
        self.input_file = input_file
        self.output_file = output_file
        self.codec = codec
        self.bitrate = bitrate
        self.ffmpeg_loglevel = ffmpeg_loglevel
        self._ffmpeg_args = [
            'ffmpeg', '-y', '-i', self.input_file,
            '-vn', '-acodec', self.codec, '-b:a', self.bitrate, self.output_file,
            '-hide_banner', '-loglevel', self.ffmpeg_loglevel, '-progress', 'pipe:1', '-nostats'
        ]
        self._duration_secs = self._get_duration()

    def _get_duration(self):
        result = subprocess.run(
            ['ffprobe', '-v', 'error', '-show_entries', 'format=duration', '-of', 'default=noprint_wrappers=1:nokey=1', self.input_file],
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True
        )
        return float(result.stdout.strip())

    def _update_progress_bar(self, line, progress_bar):
        """Update the progress bar based on FFmpeg output."""
        if "out_time_ms=" in line:
            time_str = line.split('=')[1].strip()
            seconds_processed = int(time_str) / 1_000_000
            # Ensure that we do not exceed the total progress
            progress_bar.update(min(seconds_processed - progress_bar.n, progress_bar.total - progress_bar.n))

    def run(self):
        """Run the FFmpeg process and display the progress."""
        if self._duration_secs is None:
            print("Cannot determine video duration. Exiting.")
            return

        with subprocess.Popen(self._ffmpeg_args, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) as process:
            progress_bar = tqdm(total=self._duration_secs, unit='s', desc="Extracting audio", dynamic_ncols=True)

            try:
                for line in process.stdout:
                    self._update_progress_bar(line, progress_bar)
            except KeyboardInterrupt:
                progress_bar.close()
                print("[KeyboardInterrupt] FFmpeg process killed.")
                process.kill()
                sys.exit()
            except Exception as e:
                progress_bar.close()
                print(f"Error during FFmpeg execution: {e}")
                process.kill()
                sys.exit()

            process.wait()  # Ensure the process completes
            progress_bar.close()
            if process.returncode != 0:
                print(f"FFmpeg process failed with return code {process.returncode}.")
            else:
                print("Audio extraction completed successfully.")

python
from .ffmpeg import AudioExtractor

if __name__ == "__main__":
    extractor = AudioExtractor("./vim-in-100-seconds.mp4", "./output_audio.m4a")
    extractor.run()