convert_audio.py

# -*- coding: utf-8 -*-

import argparse
from lib.collection_utils import *
from lib.io_utils import *
from lib.processing_utils import *
from lib.video_utils import *
from multiprocessing import Pool
from multiprocessing.dummy import Pool as ThreadPool
import os
from pprint import pprint
import shutil
import subprocess
import sys
import tarfile

# input
parser = argparse.ArgumentParser()
parser.add_argument('-in', dest="INPUT_FILE", default="path/to/*.ext1", help="Input file pattern")
parser.add_argument('-filter', dest="FILTER", default="", help="Filter string if input file is csv file")
parser.add_argument('-dir', dest="MEDIA_DIRECTORY", default="", help="Media directory if input file is csv file")
parser.add_argument('-out', dest="OUTPUT_FILE", default="path/to/%s.ext2", help="Output file pattern")
parser.add_argument('-threads', dest="THREADS", default=3, type=int, help="Number of concurrent threads, -1 for all available")
parser.add_argument('-overwrite', dest="OVERWRITE", action="store_true", help="Overwrite existing wavs?")
parser.add_argument('-probe', dest="PROBE", action="store_true", help="Just output commands?")
a = parser.parse_args()

filenames = []
if a.INPUT_FILE.endswith(".csv"):
    fieldNames, files, fileCount = getFilesFromString(a)
    if len(a.FILTER) > 0:
        files = filterByQueryString(files, a.FILTER)
        fileCount = len(files)
        print("%s files after filtering" % fileCount)
    filenames = [f["filename"] for f in files]
else:
    filenames = getFilenames(a.INPUT_FILE)

if len(filenames) < 1:
    print("No files found")
    sys.exit()

makeDirectories(a.OUTPUT_FILE)

def convertFile(p):
    global a

    fileFrom, fileTo = p
    fileFrom = fileFrom.replace('\\', '/')

    if os.path.isfile(fileTo) and not a.OVERWRITE:
        return

    command = ['ffmpeg', '-hide_banner', '-loglevel', 'panic', '-y', '-i', fileFrom]
    extFrom = getFileExt(fileFrom)
    extTo = getFileExt(fileTo)

    if extTo == ".webm":
        if isVideoExtension(extFrom):
            command += ['-c:v', 'libvpx-vp9']
        command += ['-c:a', 'libopus']

    elif extTo == ".ogv":
        command += ['-c:v', 'libtheora', '-c:a', 'libvorbis']

    elif extTo == ".ogg":
        command += ['-c:a', 'libvorbis']

    command += [fileTo]

    if a.PROBE:
        print(" ".join(command))
    else:
        finished = subprocess.check_call(command)

props = []
for fn in filenames:
    basename = getBasename(fn)
    outfn = a.OUTPUT_FILE % basename
    props.append((fn, outfn))

pool = ThreadPool(getThreadCount(a.THREADS))
results = pool.map(convertFile, props)
pool.close()
pool.join()

print("Done.")