summaryrefslogtreecommitdiff
path: root/forms
diff options
context:
space:
mode:
Diffstat (limited to 'forms')
-rwxr-xr-xforms/colour_chart.py75
1 files changed, 56 insertions, 19 deletions
diff --git a/forms/colour_chart.py b/forms/colour_chart.py
index bbdf021..7ec7ddd 100755
--- a/forms/colour_chart.py
+++ b/forms/colour_chart.py
@@ -9,10 +9,6 @@ import strictyaml
class Chart:
- COLUMNS=4
- ROWS=16
- INDEX_COLS=6
-
def __init__(self):
self.parser = self.get_parser()
self.args = self.parser.parse_args()
@@ -24,10 +20,40 @@ class Chart:
"chart_data",
type=argparse.FileType("r")
)
+ parser.add_argument(
+ "--output",
+ type=argparse.FileType("w"),
+ default=None,
+ help="Destination file",
+ )
+ parser.add_argument(
+ "--columns", "-c",
+ default=4,
+ type=int,
+ help="Number of columns per page",
+ )
+ parser.add_argument(
+ "--rows", "-r",
+ default=16,
+ type=int,
+ help="Number of rows per page",
+ )
+ parser.add_argument(
+ "--index-columns", "-i",
+ default=6,
+ type=int,
+ help="Number of columns per index page",
+ )
+ parser.add_argument(
+ "--background-fill", "-b",
+ action=argparse.BooleanOptionalAction,
+ default=True,
+ help="Fill the background with the target colour",
+ )
return parser
def get_arranged_cells(self):
- per_page = self.COLUMNS * self.ROWS
+ per_page = self.args.columns * self.args.rows
padded = self.data["colours"] + [{}] * (
per_page - len(self.data["colours"]) % per_page
)
@@ -37,8 +63,8 @@ class Chart:
]
cells = [
[
- [p[i + self.ROWS * j] for j in range(self.COLUMNS)]
- for i in range(len(p) // self.COLUMNS)
+ [p[i + self.args.rows * j] for j in range(self.args.columns)]
+ for i in range(len(p) // self.args.columns)
]
for p in pages
]
@@ -46,13 +72,13 @@ class Chart:
def get_index(self):
index = [
- (c["label"], i // self.ROWS)
+ (c["label"], i // self.args.rows)
for i, c in enumerate(self.data["colours"])
]
index.sort(key=operator.itemgetter(0))
return index
- def write_pdf(self, fname):
+ def write_pdf(self):
pdf = fpdf.FPDF(format="A5")
pdf.add_font(
family="FreeSans",
@@ -70,27 +96,33 @@ class Chart:
for p_num, page in enumerate(self.get_arranged_cells()):
pdf.add_page()
- pdf.set_font("FreeSerif", size=12)
+ pdf.set_font("FreeSerif", size=10)
+ cell_width = (148 - 20) / (self.args.columns * 2)
pdf.cell(64, 8, self.data["title"], align="C")
pdf.ln()
- pdf.set_font("FreeSans", "B", 12)
- for j in range(self.COLUMNS):
- pdf.cell(16, 8, str(j + p_num * self.COLUMNS),)
- pdf.cell(16, 8, "")
+ pdf.set_font("FreeSans", "B", 10)
+ for j in range(self.args.columns):
+ pdf.cell(cell_width, 8, str(j + p_num * self.args.columns),)
+ pdf.cell(cell_width, 8, "")
pdf.ln()
for row in page:
for cell in row:
if cell:
pdf.set_fill_color(255,255,255)
- if cell.get("bg"):
+ if self.args.background_fill and cell.get("bg"):
pdf.set_fill_color(*map(int, cell["bg"].split()))
- pdf.cell(16, 8, "", border=1, fill=True)
- pdf.cell(16, 8, cell["label"], )
+ pdf.cell(cell_width, 8, "", border=1, fill=True)
+ pdf.cell(cell_width, 8, cell["label"], )
pdf.ln(10)
+ pdf.add_page()
+ pdf.set_font("FreeSerif", size=10)
+ cell_width = (148 - 20) / (self.args.columns * 2)
+ pdf.cell(64, 8, self.data["title"], align="C")
+ pdf.ln()
pdf.set_font("FreeSerif", size=10)
with pdf.text_columns(
- ncols=self.INDEX_COLS,
+ ncols=self.args.index_columns,
balance=True,
gutter=4,
text_align="RIGHT",
@@ -103,8 +135,13 @@ class Chart:
cols.write(str(c[1]))
cols.ln()
+ fname = (
+ getattr(self.args.output, "name", None) or
+ self.args.chart_data.name.replace(".yaml", ".pdf")
+ )
+ print(fname)
pdf.output(fname)
if __name__ == "__main__":
- Chart().write_pdf("asd.pdf")
+ Chart().write_pdf()