UnderCover/latty.py

# 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)
    print("Unique ID: " + unique)
    return unique

@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
        print("COM: '" + com + "'")
        result = subprocess.run(['bash', '-c', com], stdout=subprocess.PIPE, text=True)

        if result.returncode == 0:
            threading.Timer(60 * 30, 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:
            errors = result.stdout.split("\n")
            del errors[:13]
            del errors[-2:]
            return (None, errors)

def cleanup(unique):
    print(unique)
    subprocess.run(['bash', '-c', "rm " + unique + ".*"])