105 lines
3.0 KiB
Python
105 lines
3.0 KiB
Python
# Copyright Sage Vaillancourt 2021
|
|
|
|
import os
|
|
import subprocess
|
|
from dataclasses import dataclass
|
|
|
|
from flask import send_from_directory
|
|
|
|
|
|
def get_unique():
|
|
import uuid
|
|
unique = str(uuid.uuid1().hex)
|
|
return unique
|
|
|
|
|
|
def get_datetime():
|
|
from datetime import datetime
|
|
now = datetime.now()
|
|
return now.strftime("%Y-%m-%d %H:%M:%S")
|
|
|
|
|
|
root_dir = os.path.dirname(os.getcwd())
|
|
proj_dir = root_dir + '/undercover/'
|
|
output_dir = proj_dir + 'outputs/'
|
|
base_tex_text = open(proj_dir + "/letter_templates/base.tex", "r").read()
|
|
|
|
|
|
@dataclass
|
|
class CLData:
|
|
selectedLetter: int # Metadata
|
|
|
|
username: str
|
|
company: str
|
|
jobAndPronoun: str
|
|
skillTypes: str
|
|
mySkills: str
|
|
closingText: str
|
|
body: str
|
|
|
|
def get_pairs(self):
|
|
return [
|
|
("username", self.username),
|
|
("company", self.company),
|
|
("jobAndPronoun", self.jobAndPronoun),
|
|
("skillTypes", self.skillTypes),
|
|
("mySkills", self.mySkills),
|
|
("closingText", self.closingText),
|
|
("body", self.body),
|
|
]
|
|
|
|
def generate_pdf(self):
|
|
import threading
|
|
unique_id = get_unique()
|
|
unique_file = output_dir + unique_id + ".tex"
|
|
f = open(unique_file, "w")
|
|
for pair in self.get_pairs():
|
|
f.write("\\def \\" + pair[0] + "{" + pair[1] + "}\n")
|
|
f.write(base_tex_text)
|
|
f.close()
|
|
|
|
output_arg = "-jobname=outputs/" + unique_id + " " + unique_file
|
|
com = "pdflatex -halt-on-error " + output_arg
|
|
build_text = "[" + get_datetime() + "] Building for " + unique_id + " "
|
|
result = subprocess.run(
|
|
['bash', '-c', com],
|
|
stdout=subprocess.PIPE,
|
|
text=True
|
|
)
|
|
|
|
if result.returncode == 0:
|
|
print(build_text + "[SUCCESS]")
|
|
compress_com = "gs -sDEVICE=pdfwrite -dCompatibilityLevel=1.5 -dNOPAUSE -dQUIET -dBATCH -dPrinted=false -sOutputFile=outputs/" + unique_id + ".compressed.pdf outputs/" + unique_id + ".pdf"
|
|
|
|
result = subprocess.run(
|
|
['bash', '-c', compress_com],
|
|
stdout=subprocess.PIPE,
|
|
text=True
|
|
)
|
|
|
|
threading.Timer(60 * 5, cleanup, [output_dir + unique_id]).start()
|
|
|
|
output_file = unique_id
|
|
if result.returncode == 0:
|
|
output_file += ".compressed.pdf"
|
|
else:
|
|
output_file += ".pdf"
|
|
|
|
return (send_from_directory(
|
|
output_dir,
|
|
output_file,
|
|
download_name=self.username.replace(" ", "") + "_CoverLetter.pdf",
|
|
as_attachment=True
|
|
), None)
|
|
else:
|
|
print(build_text + "[FAIL]")
|
|
# Collect output but delete boilerplate text
|
|
errors = list(map(str.strip, result.stdout.split("\n")))
|
|
del errors[:13]
|
|
del errors[-2:]
|
|
return None, errors
|
|
|
|
|
|
def cleanup(unique):
|
|
subprocess.run(['bash', '-c', "rm " + unique + ".*"])
|