builder_utils.py

"""Module containing all utility functions used by the builder"""

import os
import json
import csv
import re

def prettify(text):
    text = re.sub('[^A-Za-z0-9-_. ]+', '', text)
    text = re.sub('[-_]+', ' ', text)
    text = text.title()

    return text

def slugify(text):
    text = re.sub('[^A-Za-z0-9-_. ]+', '', text)
    text = re.sub('[_. ]+', '-', text)
    text = text.lower()

    return text

def get_first_line(body):
    line = body.splitlines()[0]
    
    return line

def parse_md(file_path, file_contents):
    from builder_libs import markdown

    html_body = markdown.markdown(file_contents)

    file_name = os.path.split(file_path)[1]
    file_name = os.path.splitext(file_name)[0]

    #custom file name decoding specific to this site
    custom_name = file_name.split('-')
    custom_name.pop() #removing year
    custom_name.pop() #removing month
    custom_name.pop() #removing date
    custom_name = '-'.join(custom_name)
    
    dataset = [{
        '_file_name_': file_name,
        '_file_name_custom_': custom_name,
        '_body_md_': file_contents,
        '_body_html_': html_body
    }]

    return dataset

def parse_json(file_path, file_contents):
    dataset = json.loads(file_contents)

    return dataset

def parse_csv(file_path, file_contents):
    reader = csv.DictReader(file_contents.splitlines())

    dataset = []
    for row in reader:
        dataset.append(row)

    return dataset