jamie/ptulsconv
1
0
Fork 0
mirror of https://github.com/iluvcapra/ptulsconv.git synced 2025-12-31 08:50:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added CSV output

Browse Source
This commit is contained in:
Jamie Hardt
2021-05-25 18:37:45 -07:00
parent 9927488f1e
commit 20b84623ff
2 changed files with 30 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
ptulsconv/__main__.py
View File

@@ -29,8 +29,8 @@ def main():
parser.add_option_group(filter_opts)
parser.add_option('-f', '--format', dest='output_format', metavar='FMT',
choices=['fmpxml', 'json', 'full'], default='fmpxml',
help='Set output format, `fmpxml`, `json`, or `full`. Default '
choices=['fmpxml', 'json', 'adr', 'csv'], default='fmpxml',
help='Set output format, `fmpxml`, `json`, `csv`, or `adr`. Default '
'is `fmpxml`.')
parser.add_option('-x', '--xsl', dest='xslt', metavar='XML', default=None,
@@ -98,8 +98,11 @@ def main():
if options.xslt is not None:
output_format = 'fmpxml'
convert(input_file=args[1], output_format=output_format, start=options.in_time, end=options.out_time,
include_muted=options.include_muted, xsl=options.xslt, select_reel=options.select_reel,
convert(input_file=args[1], output_format=output_format,
start=options.in_time,
end=options.out_time,
include_muted=options.include_muted,
xsl=options.xslt, select_reel=options.select_reel,
progress=False, output=sys.stdout, log_output=sys.stderr,
warnings=options.warnings)
except FileNotFoundError as e:

28
ptulsconv/commands.py
View File

@@ -14,7 +14,6 @@ from ptulsconv.pdf.line_count import output_report as output_line_count
from ptulsconv.pdf.talent_sides import output_report as output_talent_sides
from ptulsconv.pdf.summary_log import output_report as output_summary
# field_map maps tags in the text export to fields in FMPXMLRESULT
# - tuple field 0 is a list of tags, the first tag with contents will be used as source
# - tuple field 1 is the field in FMPXMLRESULT
@@ -55,6 +54,24 @@ adr_field_map = ((['Title', 'PT.Session.Name'], 'Title', str),
)
def dump_csv(events, keys=(), output=sys.stdout):
import csv
if len(keys) == 0:
keys = ('Title', 'Cue Number', 'Character Name', 'Reel', 'Version', 'Line',
'Start', 'Finish', 'Reason', 'Note', 'TV', 'Version')
writer = csv.writer(output, dialect='excel')
writer.writerow(keys)
for event in events:
this_row = list()
for key in keys:
this_row.append(event.get(key, ""))
writer.writerow(this_row)
def dump_field_map(field_map_name, output=sys.stdout):
output.write("# Map of Tag fields to XML output columns\n")
output.write("# (in order of precedence)\n")
@@ -73,7 +90,7 @@ def dump_field_map(field_map_name, output=sys.stdout):
output.write("# %-27s-> %-20s | %-8s| %-7i\n" % (tag[:27], field[1][:20], field[2].__name__, n + 1))
def normalize_record_keys(records):
def normalize_record_keys_for_adr(records):
for record in records['events']:
for field in adr_field_map:
spot_keys = field[0]
@@ -123,7 +140,7 @@ def convert(input_file, output_format='fmpxml', start=None, end=None, select_ree
reel_xform = ptulsconv.transformations.SelectReel(reel_num=select_reel)
parsed = reel_xform.transform(parsed)
parsed = normalize_record_keys(parsed)
parsed = normalize_record_keys_for_adr(parsed)
if warnings:
for warning in chain(validate_unique_field(parsed, field='QN'),
@@ -137,14 +154,15 @@ def convert(input_file, output_format='fmpxml', start=None, end=None, select_ree
validate_unique_count(parsed, field='Title', count=1),
validate_unique_count(parsed, field='Spotting', count=1),
validate_unique_count(parsed, field='Supervisor', count=1)):
print_warning(warning.report_message())
if output_format == 'json':
json.dump(parsed, output)
elif output_format == 'adr':
elif output_format == 'csv':
dump_csv(parsed['events'])
elif output_format == 'adr':
lines = [e for e in parsed['events'] if 'ADR' in e.keys()]
print_section_header_style("Creating PDF Reports")