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

Refactoring line count

Browse Source
This commit is contained in:
Jamie Hardt
2021-06-03 12:24:31 -07:00
parent 5fb4c389f4
commit 55324a0f82
2 changed files with 46 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ptulsconv/commands.py
View File

@@ -93,9 +93,9 @@ def create_adr_reports(lines: List[ADRLine], tc_display_format: TimecodeFormat):
print_status_style("Creating ADR Report") print_status_style("Creating ADR Report")
output_summary(lines, tc_display_format=tc_display_format) output_summary(lines, tc_display_format=tc_display_format)
# print_status_style("Creating Line Count") print_status_style("Creating Line Count")
# output_line_count(lines) output_line_count(lines, reel_list=['R1', 'R2', 'R3', 'R4', 'R5', 'R6', 'R7'])
#
print_status_style("Creating Supervisor Logs directory and reports") print_status_style("Creating Supervisor Logs directory and reports")
os.makedirs("Supervisor Logs", exist_ok=True) os.makedirs("Supervisor Logs", exist_ok=True)
os.chdir("Supervisor Logs") os.chdir("Supervisor Logs")

81
ptulsconv/pdf/line_count.py
View File

@@ -1,3 +1,5 @@
from typing import List, Optional
from reportlab.pdfbase import pdfmetrics from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont from reportlab.pdfbase.ttfonts import TTFont
@@ -9,17 +11,19 @@ from reportlab.platypus import Table, Paragraph
from reportlab.lib.styles import getSampleStyleSheet from reportlab.lib.styles import getSampleStyleSheet
from .__init__ import time_format, make_doc_template from .__init__ import time_format, make_doc_template
from ..broadcast_timecode import TimecodeFormat
from ..docparser.adr_entity import ADRLine
def build_columns(lines, show_priorities=False, include_omitted=False): def build_columns(lines: List[ADRLine], reel_list: Optional[List[str]], show_priorities=False, include_omitted=False):
columns = list() columns = list()
reel_numbers = sorted(set([x['Reel'] for x in lines if 'Reel' in x.keys()])) reel_numbers = reel_list or sorted(set([x.reel for x in lines if x.reel is not None]))
num_column_width = 0.375 * inch num_column_width = 15. / 32. * inch
columns.append({ columns.append({
'heading': '#', 'heading': '#',
'value_getter': lambda recs: recs[0]['Character Number'], 'value_getter': lambda recs: recs[0].character_id,
'value_getter2': lambda recs: "", 'value_getter2': lambda recs: "",
'style_getter': lambda col_index: [], 'style_getter': lambda col_index: [],
'width': 0.375 * inch, 'width': 0.375 * inch,
@@ -28,8 +32,8 @@ def build_columns(lines, show_priorities=False, include_omitted=False):
columns.append({ columns.append({
'heading': 'Role', 'heading': 'Role',
'value_getter': lambda recs: recs[0]['Character Name'], 'value_getter': lambda recs: recs[0].character_name,
'value_getter2': lambda recs: recs[0].get('Actor Name', ""), 'value_getter2': lambda recs: recs[0].actor_name or "",
'style_getter': lambda col_index: [('LINEAFTER', (col_index, 0), (col_index, -1), 1.0, colors.black)], 'style_getter': lambda col_index: [('LINEAFTER', (col_index, 0), (col_index, -1), 1.0, colors.black)],
'width': 1.75 * inch, 'width': 1.75 * inch,
'summarize': False 'summarize': False
@@ -48,9 +52,9 @@ def build_columns(lines, show_priorities=False, include_omitted=False):
for n in reel_numbers: for n in reel_numbers:
columns.append({ columns.append({
'heading': n, 'heading': n,
'value_getter': lambda recs, n1=n: len([r for r in recs if r['Reel'] == n1]), 'value_getter': lambda recs, n1=n: len([r for r in recs if r.reel == n1]),
'value_getter2': lambda recs, n1=n: time_format(sum([r.get('Time Budget Mins', 0.) for r in recs 'value_getter2': lambda recs, n1=n: time_format(sum([r.time_budget_mins or 0. for r in recs
if r['Reel'] == n1])), if r.reel == n1])),
'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER'), 'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER'),
('LINEAFTER', (col_index, 0), (col_index, -1), .5, colors.gray)], ('LINEAFTER', (col_index, 0), (col_index, -1), .5, colors.gray)],
'width': num_column_width 'width': num_column_width
@@ -60,27 +64,27 @@ def build_columns(lines, show_priorities=False, include_omitted=False):
for n in range(1, 6,): for n in range(1, 6,):
columns.append({ columns.append({
'heading': 'P%i' % n, 'heading': 'P%i' % n,
'value_getter': lambda recs: len([r for r in recs if r.get('Priority', None) == n]), 'value_getter': lambda recs: len([r for r in recs if r.priority == n]),
'value_getter2': lambda recs: time_format(sum([r.get('Time Budget Mins', 0.) 'value_getter2': lambda recs: time_format(sum([r.time_budget_mins or 0.
for r in recs if r.get('Priority', None) == n])), for r in recs if r.priority == n])),
'style_getter': lambda col_index: [], 'style_getter': lambda col_index: [],
'width': num_column_width 'width': num_column_width
}) })
columns.append({ columns.append({
'heading': '>P5', 'heading': '>P5',
'value_getter': lambda recs: len([r for r in recs if r.get('Priority', 5) > 5]), 'value_getter': lambda recs: len([r for r in recs if (r.priority or 5) > 5]),
'value_getter2': lambda recs: time_format(sum([r.get('Time Budget Mins', 0.) 'value_getter2': lambda recs: time_format(sum([r.time_budget_mins or 0.
for r in recs if r.get('Priority', 5) > 5])), for r in recs if (r.priority or 5) > 5])),
'style_getter': lambda col_index: [], 'style_getter': lambda col_index: [],
'width': num_column_width 'width': num_column_width
}) })
columns.append({ columns.append({
'heading': 'TV', 'heading': 'TV',
'value_getter': lambda recs: len([r for r in recs if 'TV' in r.keys()]), 'value_getter': lambda recs: len([r for r in recs if r.tv]),
'value_getter2': lambda recs: time_format(sum([r.get('Time Budget Mins', 0.) 'value_getter2': lambda recs: time_format(sum([r.time_budget_mins or 0.
for r in recs if 'TV' in r.keys()])), for r in recs if r.tv])),
'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER'), 'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER'),
('LINEBEFORE', (col_index, 0), (col_index, -1), 1., colors.black), ('LINEBEFORE', (col_index, 0), (col_index, -1), 1., colors.black),
('LINEAFTER', (col_index, 0), (col_index, -1), .5, colors.gray)], ('LINEAFTER', (col_index, 0), (col_index, -1), .5, colors.gray)],
@@ -89,9 +93,9 @@ def build_columns(lines, show_priorities=False, include_omitted=False):
columns.append({ columns.append({
'heading': 'Opt', 'heading': 'Opt',
'value_getter': lambda recs: len([r for r in recs if 'Optional' in r.keys()]), 'value_getter': lambda recs: len([r for r in recs if r.optional]),
'value_getter2': lambda recs: time_format(sum([r.get('Time Budget Mins', 0.) 'value_getter2': lambda recs: time_format(sum([r.time_budget_mins or 0.
for r in recs if 'Optional' in r.keys()])), for r in recs if r.optional])),
'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER'), 'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER'),
('LINEAFTER', (col_index, 0), (col_index, -1), .5, colors.gray)], ('LINEAFTER', (col_index, 0), (col_index, -1), .5, colors.gray)],
'width': num_column_width 'width': num_column_width
@@ -99,9 +103,9 @@ def build_columns(lines, show_priorities=False, include_omitted=False):
columns.append({ columns.append({
'heading': 'Eff', 'heading': 'Eff',
'value_getter': lambda recs: len([r for r in recs if 'Effort' in r.keys()]), 'value_getter': lambda recs: len([r for r in recs if r.effort]),
'value_getter2': lambda recs: time_format(sum([r.get('Time Budget Mins',0.) 'value_getter2': lambda recs: time_format(sum([r.time_budget_mins or 0.
for r in recs if 'Effort' in r.keys()])), for r in recs if r.effort])),
'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER')], 'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER')],
'width': num_column_width 'width': num_column_width
}) })
@@ -109,18 +113,18 @@ def build_columns(lines, show_priorities=False, include_omitted=False):
if include_omitted: if include_omitted:
columns.append({ columns.append({
'heading': 'Omit', 'heading': 'Omit',
'value_getter': lambda recs: len([r for r in recs if 'Omitted' in r.keys()]), 'value_getter': lambda recs: len([r for r in recs if r.omitted]),
'value_getter2': lambda recs: time_format(sum([r.get('Time Budget Mins', 0.) 'value_getter2': lambda recs: time_format(sum([r.time_budget_mins or 0.
for r in recs if 'Omitted' in r.keys()])), for r in recs if r.omitted])),
'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER')], 'style_getter': lambda col_index: [('ALIGN', (col_index, 0), (col_index, -1), 'CENTER')],
'width': num_column_width 'width': num_column_width
}) })
columns.append({ columns.append({
'heading': 'Total', 'heading': 'Total',
'value_getter': lambda recs: len([r for r in recs if 'Omitted' not in r.keys()]), 'value_getter': lambda recs: len([r for r in recs if not r.omitted]),
'value_getter2': lambda recs: time_format(sum([r.get('Time Budget Mins', 0.) 'value_getter2': lambda recs: time_format(sum([r.time_budget_mins or 0.
for r in recs if 'Omitted' not in r.keys()]), zero_str=None), for r in recs if not r.omitted]), zero_str=None),
'style_getter': lambda col_index: [('LINEBEFORE', (col_index, 0), (col_index, -1), 1.0, colors.black), 'style_getter': lambda col_index: [('LINEBEFORE', (col_index, 0), (col_index, -1), 1.0, colors.black),
('ALIGN', (col_index, 0), (col_index, -1), 'CENTER')], ('ALIGN', (col_index, 0), (col_index, -1), 'CENTER')],
'width': 0.5 * inch 'width': 0.5 * inch
@@ -129,13 +133,12 @@ def build_columns(lines, show_priorities=False, include_omitted=False):
return columns return columns
def populate_columns(lines, columns, include_omitted, page_size): def populate_columns(lines: List[ADRLine], columns, include_omitted, page_size):
data = list() data = list()
styles = list() styles = list()
columns_widths = list() columns_widths = list()
sorted_character_numbers = sorted(set([x['Character Number'] for x in lines sorted_character_numbers = sorted(set([x.character_id for x in lines]),
if 'Character Number' in x.keys()]),
key=lambda x: str(x)) key=lambda x: str(x))
# construct column styles # construct column styles
@@ -147,10 +150,10 @@ def populate_columns(lines, columns, include_omitted, page_size):
data.append(list(map(lambda x: x['heading'], columns))) data.append(list(map(lambda x: x['heading'], columns)))
if not include_omitted: if not include_omitted:
lines = [x for x in lines if 'Omitted' not in x.keys()] lines = [x for x in lines if not x.omitted]
for n in sorted_character_numbers: for n in sorted_character_numbers:
char_records = list([x for x in lines if x['Character Number'] == n]) char_records = [x for x in lines if x.character_id == n]
row_data = list() row_data = list()
row_data2 = list() row_data2 = list()
for col in columns: for col in columns:
@@ -187,8 +190,10 @@ def populate_columns(lines, columns, include_omitted, page_size):
def build_header(column_widths): def build_header(column_widths):
pass pass
def output_report(lines, include_omitted=False, page_size=portrait(letter)):
columns = build_columns(lines, include_omitted) def output_report(lines: List[ADRLine], reel_list: List[str], include_omitted=False,
page_size=portrait(letter)):
columns = build_columns(lines, include_omitted=include_omitted, reel_list=reel_list)
data, style, columns_widths = populate_columns(lines, columns, include_omitted, page_size) data, style, columns_widths = populate_columns(lines, columns, include_omitted, page_size)
style.append(('FONTNAME', (0, 0), (-1, -1), "Futura")) style.append(('FONTNAME', (0, 0), (-1, -1), "Futura"))
@@ -198,7 +203,7 @@ def output_report(lines, include_omitted=False, page_size=portrait(letter)):
pdfmetrics.registerFont(TTFont('Futura', 'Futura.ttc')) pdfmetrics.registerFont(TTFont('Futura', 'Futura.ttc'))
title = "%s Line Count" % (lines[0]['Title']) title = "%s Line Count" % lines[0].title
filename = title + '.pdf' filename = title + '.pdf'
doc = make_doc_template(page_size=page_size, filename=filename, doc = make_doc_template(page_size=page_size, filename=filename,
document_title=title, document_title=title,