From 226c645a6b6db7d6222697149889559df35202cb Mon Sep 17 00:00:00 2001 From: zevav Date: Wed, 20 Feb 2019 11:51:58 -0500 Subject: [PATCH] bug fixes and back to pprint for JSON output --- setup.py | 2 +- tatt/helpers.py | 5 +++-- tatt/transcribe.py | 4 ++-- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/setup.py b/setup.py index 53d30a4..e74afa4 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ with open('README.md') as file: setup( name="tatt", - version="0.943", + version="0.947", py_modules=['tatt'], url='https://github.com/zevaverbach/tatt', install_requires=[ diff --git a/tatt/helpers.py b/tatt/helpers.py index 5ba3bbd..c3bdfbe 100644 --- a/tatt/helpers.py +++ b/tatt/helpers.py @@ -25,17 +25,18 @@ def make_string_all_services(free_only=False): def get_job(job_name): - job = helpers.get_transcription_jobs_dict().get(name) + job = get_transcription_jobs_dict().get(job_name) if not job: raise exceptions.DoesntExistError if job['status'].lower() != 'completed': raise exceptions.NotAvailable(f'transcript status is {job["status"]}') + return job def get_transcript(job_name): job = get_job(job_name) service = get_service(job['service_name']) - return service.retrieve_transcript(name) + return service.retrieve_transcript(job_name) def get_service(service_name): diff --git a/tatt/transcribe.py b/tatt/transcribe.py index a081247..ac03997 100644 --- a/tatt/transcribe.py +++ b/tatt/transcribe.py @@ -19,14 +19,14 @@ def cli(): def get(name, save): """Downloads and/or saves completed transcript.""" try: - transcript = get_transcript(name) + transcript = helpers.get_transcript(name) except exceptions.DoesntExistError: raise click.ClickException(f'no such transcript {name}') except exceptions.NotAvailable as e: raise click.ClickException(str(e)) if not save: - click.echo(transcript) + pprint(transcript) else: with open(f'{name}.json', 'w') as fout: fout.write(json.dumps(transcript))