#!/usr/bin/env python
from functools import partial
from multiprocessing import Pool
from transcribe import transcribe_to_file
import argparse
import os
import logging


logging.basicConfig(level=logging.INFO, format='%(message)s')
logger = logging.getLogger(__file__)

DEFAULT_WORKER_COUNT = 20

parser = argparse.ArgumentParser(description='Transcribe multiple audio files in parallel using Google Speech API')
parser.add_argument('in_filenames', nargs='+', help='Input filename(s)')
parser.add_argument('--keep-suffix', action='store_true',
    help='Don\'t strip filename suffix when generating metadata .json output filename')
parser.add_argument('--workers', default=DEFAULT_WORKER_COUNT,
    help='Number of workers (default {})'.format(DEFAULT_WORKER_COUNT))


def transcribe_one(in_filename, keep_suffix=False):
    if keep_suffix:
        base_filename = in_filename
    else:
        base_filename = os.path.splitext(in_filename)[0]
    out_filename = '{}.json'.format(base_filename)
    logger.info('Starting: {} -> {}'.format(in_filename, out_filename))
    with open(out_filename, 'w') as out_file:
        transcribe_to_file(in_filename, out_file, as_json=True)
    logger.info('Finished: {} -> {}'.format(in_filename, out_filename))


def transcribe_many(in_filenames, keep_suffix=False, worker_count=DEFAULT_WORKER_COUNT):
    pool = Pool(processes=worker_count)
    func = partial(transcribe_one, keep_suffix=keep_suffix)
    pool.map_async(func, in_filenames).get(99999999)


if __name__ == '__main__':
    args = parser.parse_args()
    transcribe_many(args.in_filenames, args.keep_suffix)