75 lines
2.2 KiB
Python
Executable File
75 lines
2.2 KiB
Python
Executable File
import click
|
|
from click import ClickException
|
|
|
|
from app.config import BEGIN_ANSWERS_TOKEN
|
|
from app.generate import generate_answers, generate_problems
|
|
from app.pdf import make_pdf
|
|
|
|
|
|
@click.command()
|
|
@click.argument("bits", type=click.INT)
|
|
@click.argument("num-problems", type=click.INT)
|
|
@click.option("--pdf", default=False, is_flag=True)
|
|
@click.option("--silent", default=False, is_flag=True)
|
|
@click.option("--include-answers", default=True, is_flag=True)
|
|
@click.option("--output-filepath")
|
|
@click.option("--num-columns", default=1)
|
|
@click.option("--font-size", default=26)
|
|
def main(
|
|
bits: int,
|
|
num_problems: int,
|
|
pdf: bool = False,
|
|
silent: bool = False,
|
|
include_answers: bool = True,
|
|
output_filepath: str = None,
|
|
num_columns: int = 1,
|
|
font_size: int = 26,
|
|
) -> None:
|
|
|
|
validate_args(pdf, silent, output_filepath)
|
|
|
|
answers, problems = make_problems_and_answers(bits, num_problems)
|
|
|
|
if pdf:
|
|
silent = True
|
|
make_pdf(
|
|
problems=problems,
|
|
answers=answers,
|
|
output_path=output_filepath or "problems.pdf",
|
|
include_answers=include_answers,
|
|
num_columns=num_columns,
|
|
font_size=font_size,
|
|
)
|
|
|
|
if not silent:
|
|
if include_answers:
|
|
click.echo(BEGIN_ANSWERS_TOKEN.join((problems, answers)))
|
|
else:
|
|
click.echo(problems)
|
|
|
|
|
|
def validate_args(pdf, silent, output_filepath):
|
|
if pdf and silent:
|
|
raise ClickException(
|
|
"please specify either `pdf` or `silent`, not both (otherwise there "
|
|
"won't be any outcome of running the app!"
|
|
)
|
|
|
|
if pdf and output_filepath and not output_filepath.endswith("pdf"):
|
|
raise ClickException("Please include an output filepath ending in '.pdf'")
|
|
|
|
|
|
def make_problems_and_answers(bits, num_problems):
|
|
problems = generate_problems(bits, num_problems)
|
|
answers = generate_answers(problems)
|
|
problems_string = "\n\n".join(
|
|
[f"{index:2}) {p} =" for index, p in enumerate(problems, start=1)]
|
|
)
|
|
answers_string = "\n\n".join(
|
|
[
|
|
f"{index:2}) {problem} = {answer}"
|
|
for index, (problem, answer) in enumerate(zip(problems, answers), start=1)
|
|
]
|
|
)
|
|
return answers_string, problems_string
|