compose

#!/usr/bin/python3

from jinja2 import Environment, select_autoescape, FileSystemLoader
from argparse import ArgumentParser
import os
import yaml
import pdfkit
import subprocess
from importlib import util as importutil


arg_parser = ArgumentParser()

arg_parser.add_argument(
    dest="data",
    help="CV data file name"
)
arg_parser.add_argument(
    "--width",
    dest="width",
    help="Page width in cm",
    default=28,
    type=float
)
arg_parser.add_argument(
    "--height",
    dest="height",
    help="Page height in cm",
    default=29.7,
    type=float
)
arg_parser.add_argument(
    "--template",
    dest="template",
    help="Template to use",
    default="default"
)

arg_parser.add_argument(
    "-T",
    dest="mTop",
    help="Margin top",
    default="0",
    type=str
)

arg_parser.add_argument(
    "-B",
    dest="mBottom",
    help="Margin bottom",
    default="0",
    type=str
)

arg_parser.add_argument(
    "-L",
    dest="mLeft",
    help="Margin left",
    default="0",
    type=str
)

arg_parser.add_argument(
    "-R",
    dest="mRight",
    help="Margin right",
    default="0",
    type=str
)

arg_parser.add_argument(
    "--grayscale",
    help="Convert to grayscale",
    dest="grayscale",
    action="store_true"
)

args = arg_parser.parse_args()


DIR_PATH = os.path.dirname(os.path.abspath(__file__))
OUT_PATH = os.path.join(DIR_PATH, "out")
TEMPLATES_FOLDER = os.path.join(DIR_PATH, "templates", args.template)
DATAPARSER_MODULE = os.path.join(TEMPLATES_FOLDER, "dataparser.py")
STYLES_FILE = os.path.join(TEMPLATES_FOLDER, "index.scss")
HTML_IN_FILE = os.path.join(TEMPLATES_FOLDER, "build/index.html")
CSS_IN_FILE = os.path.join(TEMPLATES_FOLDER, "build/index.css")
LOC_DIR = os.path.join(TEMPLATES_FOLDER, "loc")
PDF_OUT_FILE = os.path.join(OUT_PATH, "cv.pdf")
DATA_DIR = os.path.join(DIR_PATH, "data")
# using path relative to template/<name>/build/index.html
PHOTO_DIR = "../../../data/photo"

DEFAULT_LOC = "en.yml"

data_filename = "{}.yml".format(args.data)

with open(os.path.join(DATA_DIR, data_filename), "rb") as datastream:
    data = yaml.safe_load(datastream)

# importing custom dataparser module
dataparser_spec = importutil.spec_from_file_location(
    ".dataparser",
    DATAPARSER_MODULE
)
dataparser = importutil.module_from_spec(dataparser_spec)
dataparser_spec.loader.exec_module(dataparser)

# parsing data
data = dataparser.parse(data)

if data.get('photo'):
    data['photo'] = os.path.join(PHOTO_DIR, data['photo'])

locale_filename = "{}.yml".format(data.get("loc", "en"))

try:
    with open(os.path.join(LOC_DIR, locale_filename), "rb") as loc:
        data['loc'] = yaml.safe_load(loc)
except FileNotFoundError:
    with open(os.path.join(LOC_DIR, DEFAULT_LOC), "rb") as loc:
        data['loc'] = yaml.safe_load(loc)

env = Environment(
    loader=FileSystemLoader(TEMPLATES_FOLDER),
    autoescape=select_autoescape(["html", "scss"])
)

main = env.get_template("index.html")

with open(HTML_IN_FILE, "w", encoding="utf-8") as f:
    f.write(main.render(**data))

subprocess.call(["npx", "node-sass", STYLES_FILE, CSS_IN_FILE])
subprocess.call(["npx", "postcss", CSS_IN_FILE, "-u", "autoprefixer", "-r"])

options = {
    'page-width': "{}cm".format(args.width),
    'page-height': "{}cm".format(args.height),
    'margin-top': args.mTop,
    'margin-right': args.mRight,
    'margin-bottom': args.mBottom,
    'margin-left': args.mLeft,
    'encoding': "UTF-8",
}

if args.grayscale:
    options['grayscale'] = ''

pdfkit.from_file(HTML_IN_FILE, PDF_OUT_FILE, options=options)

print("Template rendered successfuly")