Added CSV output options

ptulsconv/commands.py

@@ -4,6 +4,8 @@ import os
 import sys
 from itertools import chain
 
+import csv
+
 import ptulsconv
 from .reporting import print_section_header_style, print_status_style, print_warning
 from .validations import *
@@ -53,14 +55,15 @@ adr_field_map = ((['Title', 'PT.Session.Name'], 'Title', str),
                  (['Movie.Start_Offset_Seconds'], 'Movie Seconds', float),
                  )
 
+def dump_csv(events, output=sys.stdout):
+    keys = set()
+    for e in events:
+        keys.update(e.keys())
 
-def dump_csv(events, keys=(), output=sys.stdout):
+    dump_keyed_csv(events, keys=keys, output=output)
-    import csv
 
-    if len(keys) == 0:
-        keys = ('Title', 'Cue Number', 'Character Name', 'Reel', 'Version', 'Line',
-                'Start', 'Finish', 'Reason', 'Note', 'TV', 'Version')
 
+def dump_keyed_csv(events, keys=(), output=sys.stdout):
     writer = csv.writer(output, dialect='excel')
     writer.writerow(keys)
 
@@ -92,6 +95,9 @@ def dump_field_map(field_map_name, output=sys.stdout):
 
 def normalize_record_keys_for_adr(records):
     for record in records['events']:
+        if 'ADR' not in record.keys():
+            continue
+
         for field in adr_field_map:
             spot_keys = field[0]
             output_key = field[1]
@@ -99,6 +105,7 @@ def normalize_record_keys_for_adr(records):
             for attempt_key in spot_keys:
                 if attempt_key in record.keys():
                     record[output_key] = field_type(record[attempt_key])
+                    break
 
     return records
 
@@ -184,6 +191,12 @@ def convert(input_file, output_format='fmpxml', start=None, end=None, select_ree
             os.chdir("Director Logs")
             output_summary(lines, by_character=True)
 
+            os.chdir("..")
+            print_status_style("Creating CSV outputs")
+            os.makedirs("CSV", exist_ok=True)
+            os.chdir("CSV")
+            output_adr_csv(lines)
+
             os.chdir("..")
             print_status_style("Creating Scripts directory and reports")
             os.makedirs("Talent Scripts", exist_ok=True)
@@ -197,3 +210,24 @@ def convert(input_file, output_format='fmpxml', start=None, end=None, select_ree
                 print_section_header_style("Performing XSL Translation")
                 print_status_style("Using builtin translation: %s" % xsl)
                 fmp_transformed_dump(parsed, input_file, xsl, output)
+
+
+def output_adr_csv(lines):
+    adr_keys = ('Title', 'Cue Number', 'Character Name', 'Reel', 'Version', 'Line',
+                'Start', 'Finish', 'Reason', 'Note', 'TV', 'Version')
+    reels = set([ln['Reel'] for ln in lines])
+    reels.add(None)
+    for n in [n['Character Number'] for n in lines]:
+        for reel in reels:
+            these_lines = [ln for ln in lines
+                           if ln['Character Number'] == n and
+                           ln.get('Reel', None) == reel]
+
+            if len(these_lines) == 0:
+                continue
+
+            outfile_name = "%s_%s_%s_%s.csv" % (these_lines[0]['Title'],
+                                                n, these_lines[0]['Character Name'], reel,)
+
+            with open(outfile_name, mode='w', newline='') as outfile:
+                dump_keyed_csv(these_lines, adr_keys, outfile)

ptulsconv/pdf/talent_sides.py

@@ -23,7 +23,7 @@ def output_report(lines):
         sorted(char_lines, key=lambda line: line['PT.Clip.Start_Seconds'])
 
         title = "%s (%s) %s ADR Script" % (char_lines[0]['Title'], char_lines[0]['Character Name'], n)
-        filename = "%s_%s_%s_ADR Script.pdf" % (lines[0]['Title'], n, lines[0]['Character Name'])
+        filename = "%s_%s_%s_ADR Script.pdf" % (char_lines[0]['Title'], n, char_lines[0]['Character Name'])
 
         doc = make_doc_template(page_size=letter, filename=filename, document_title=title,
                                 record=char_lines[0], document_header=char_lines[0]['Character Name'])
@@ -32,9 +32,9 @@ def output_report(lines):
 
         prompt_style = getSampleStyleSheet()['Normal']
         prompt_style.fontName = 'Futura'
-        prompt_style.fontSize = 18
+        prompt_style.fontSize = 18.
 
-        prompt_style.leading = 24
+        prompt_style.leading = 24.
         prompt_style.leftIndent = 1.5 * inch
         prompt_style.rightIndent = 1.5 * inch