ResumeGenerator.py

import os
from jinja2 import Environment, FileSystemLoader
import json
import yaml
from datetime import datetime
import argparse
import re
from weasyprint import HTML, CSS


# Resume + Template Constants
CUSTOM_DATE_FORMAT = "%b %Y"
PATH_CAREER_CUP_DIRECTORY = "CareerCup"
PATH_CAREER_CUP_TEMPLATE = r"CareerCup/careerCupTemplate.html"

# Anon
anchorPattern = re.compile(r"<a([^>]*)>[^<]*</a>")


def dateformat(value: datetime, format=CUSTOM_DATE_FORMAT):
    return value.strftime(format)


class ResumeGenerator(object):
    def __init__(
        self,
        working_directory: str,
        dict_resume: dict,
        path_html_template: str,
        dict_anon=None,
    ):
        self.TEMPLATE_ENVIRONMENT = Environment(
            autoescape=False,
            loader=FileSystemLoader(working_directory),
            trim_blocks=False,
        )
        self.TEMPLATE_ENVIRONMENT.filters["dateformat"] = dateformat
        self.path_html_template = path_html_template
        self.dict_anon = dict_anon
        self.dict_resume = dict_resume

    def create_html_resume(self, output_html_name: str) -> str:
        with open(output_html_name, "w") as fp:
            html = self.TEMPLATE_ENVIRONMENT.get_template(
                self.path_html_template
            ).render(self.dict_resume)

            if self.dict_anon is not None:
                html = re.sub(anchorPattern, "", html)
                for old_string, new_string in self.dict_anon.items():
                    html = html.replace(old_string, new_string)

            fp.write(html)
        return output_html_name

    def create_pdf_from_html(self, output_name: str):
        html_path = self.create_html_resume(output_name + ".html")
        html = HTML(filename=html_path)
        css = CSS(string='@page { size: A3; margin: 1cm }')
        html.write_pdf(
            output_name + ".pdf",
            stylesheets=[css]
        )