it's a click cli now, and PDF output is working as an option
This commit is contained in:
39
app/pdf.py
Normal file
39
app/pdf.py
Normal file
@@ -0,0 +1,39 @@
|
||||
import os
|
||||
import subprocess
|
||||
import tempfile
|
||||
|
||||
|
||||
def make_pdf(
|
||||
problems: str, answers: str, output_path: str, include_answers: bool
|
||||
) -> None:
|
||||
if include_answers:
|
||||
pdf_jobs = (("problems", problems), ("answers", answers))
|
||||
else:
|
||||
pdf_jobs = ("problems", problems)
|
||||
|
||||
for job_name, job in pdf_jobs:
|
||||
|
||||
if job_name == "answers":
|
||||
path, filename = os.path.split(output_path)
|
||||
new_filename = f"{filename.split('.pdf')[0]}-answers.pdf"
|
||||
output_path = os.path.join(path, new_filename)
|
||||
|
||||
with tempfile.NamedTemporaryFile(mode="w") as txt_file:
|
||||
txt_file.write(job)
|
||||
txt_file.flush()
|
||||
|
||||
command = (
|
||||
f"enscript --columns=4 --no-header --output=tempfile.ps {txt_file.name}"
|
||||
)
|
||||
print(f"command: '{command}'")
|
||||
|
||||
output = subprocess.check_output(command, shell=True)
|
||||
print(output)
|
||||
|
||||
command = f"ps2pdf tempfile.ps {output_path}"
|
||||
print(f"command: '{command}'")
|
||||
|
||||
output = subprocess.check_output(command, shell=True)
|
||||
print(output)
|
||||
print("made pdf", output_path)
|
||||
os.unlink("tempfile.ps")
|
||||
Reference in New Issue
Block a user