fixed bug in cli that didn't handle txt files (from google)

2019-03-08 21:44:55 -05:00
parent 8f63320be4
commit 0990f76e19
8 changed files with 20 additions and 5 deletions
--- a/transcript_processing/tpro.py
+++ b/transcript_processing/tpro.py
@@ -23,10 +23,15 @@ def cli(print_output,
        input_format,
        output_format):

-    json_data = json.load(transcript_data_path)
-    service = services[input_format]
+    transcript_data_file_handle = transcript_data_path

-    converter = service(json_data)
+    service = services[input_format]
+    if service.transcript_type == dict:
+        transcript_data = json.load(transcript_data_file_handle)
+    else:
+        transcript_data = transcript_data_file_handle.read()
+
+    converter = service(transcript_data)
    converter.convert()
    converter.save(output_path, output_format)