md-pdf/md_pdf/build/context.py

from md_pdf.consts import TEMPLATES_DIR, STATIC_DIR, DATE_FORMAT, TIME_FORMAT, DATETIME_FORMAT, INTERNAL_TEMPLATES_DIR, INTERNAL_STATIC_DIR
from word_count import word_count
from md_pdf.utils import get_plain_text
from md_pdf import __version__
from dateutil import parser
import datetime
import os


EXTRA_CONTEXT = {
    'templates_dir': TEMPLATES_DIR,
    'static_dir': STATIC_DIR,
    'internal_templates_dir': INTERNAL_TEMPLATES_DIR,
    'internal_static_dir': INTERNAL_STATIC_DIR,
    'date': datetime.datetime.now().strftime(DATE_FORMAT),
    'time': datetime.datetime.now().strftime(TIME_FORMAT),
    'datetime': datetime.datetime.now().strftime(DATETIME_FORMAT),
    'mdp_version': __version__
}


def get_context(config: dict, content: str) -> dict:
    config = config.copy()
    if 'context' in config:
        context = config['context'].copy()
        del config['context']
    else:
        context = {}
    merged_context = dict(
        config,
        **EXTRA_CONTEXT,
        **context,
        **{
            'output_dir': os.path.abspath(config['output_dir']),
        }
    )
    if config.get('show_word_count'):
        merged_context['word_count'] = word_count(get_plain_text(content))
    if config.get('submission_date'):
        if type(config['submission_date']) in [datetime.date, datetime.datetime, datetime.time]:
            submission_date = config['submission_date']
        else:
            submission_date = parser.parse(config['submission_date'])
        merged_context['submission_date'] = submission_date.strftime(DATE_FORMAT)
    return merged_context