| | """ |
| | Script for extracting audio (16-bit, mono, 22000 Hz) from video file. |
| | """ |
| |
|
| | import os |
| | import argparse |
| | import subprocess |
| |
|
| |
|
| | def parse_args(): |
| | """ |
| | Create python script parameters. |
| | |
| | Returns |
| | ------- |
| | ArgumentParser |
| | Resulted args. |
| | """ |
| | parser = argparse.ArgumentParser( |
| | description="Extract audio from video file", |
| | formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
| | parser.add_argument( |
| | "--in-video", |
| | type=str, |
| | required=True, |
| | help="path to input video file or directory") |
| | parser.add_argument( |
| | "--out-audio", |
| | type=str, |
| | help="path to output audio file") |
| |
|
| | args = parser.parse_args() |
| | return args |
| |
|
| |
|
| | def extract_audio(in_video, |
| | out_audio): |
| | """ |
| | Real extract audio from video file. |
| | |
| | Parameters |
| | ---------- |
| | in_video : str |
| | Path to input video file. |
| | out_audio : str |
| | Path to output audio file. |
| | """ |
| | if not out_audio: |
| | file_stem, _ = os.path.splitext(in_video) |
| | out_audio = file_stem + ".wav" |
| | |
| | |
| | |
| | command = "ffmpeg -i {in_video} -vn -acodec pcm_s16le -ac 1 -ar 16000 {out_audio}" |
| | subprocess.call([command.format(in_video=in_video, out_audio=out_audio)], shell=True) |
| |
|
| |
|
| | def main(): |
| | """ |
| | Main body of script. |
| | """ |
| | args = parse_args() |
| | in_video = os.path.expanduser(args.in_video) |
| | if not os.path.exists(in_video): |
| | raise Exception("Input file/directory doesn't exist: {}".format(in_video)) |
| | if os.path.isfile(in_video): |
| | extract_audio( |
| | in_video=in_video, |
| | out_audio=args.out_audio) |
| | else: |
| | video_file_paths = [] |
| | for file_name in os.listdir(in_video): |
| | if not os.path.isfile(os.path.join(in_video, file_name)): |
| | continue |
| | _, file_ext = os.path.splitext(file_name) |
| | if file_ext.lower() in (".mp4", ".mkv", ".avi"): |
| | video_file_path = os.path.join(in_video, file_name) |
| | video_file_paths.append(video_file_path) |
| | video_file_paths = sorted(video_file_paths) |
| | for video_file_path in video_file_paths: |
| | extract_audio( |
| | in_video=video_file_path, |
| | out_audio="") |
| |
|
| |
|
| | if __name__ == "__main__": |
| | main() |
| |
|
