83 lines
2.4 KiB
Python
83 lines
2.4 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")
|
|
|
|
@dataclass
|
|
class CLData():
|
|
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()
|
|
root_dir = os.path.dirname(os.getcwd())
|
|
proj_dir = root_dir + '/undercover/'
|
|
output_dir = proj_dir + 'outputs/'
|
|
|
|
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(open(proj_dir + "/writing_templates/base.tex", "r").read())
|
|
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]")
|
|
threading.Timer(60 * 5, cleanup, [output_dir + unique_id]).start()
|
|
|
|
return (send_from_directory(
|
|
output_dir,
|
|
unique_id + ".pdf",
|
|
attachment_filename=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 + ".*"])
|