Added ability to record metadata at top of page data file.

[cmless] / bin / misc.py

# (c) 2017 Aaron Taylor <ataylor at subgeniuskitty dot com>
# See LICENSE file for copyright and license details.

# Python imports
import os, sys, subprocess, configparser

# CMless imports
import debug, config, template

# Accepts a string containing a filesystem path to a text file.
# Returns a string containing the contents of that file.
def load_file(path):
    try:
        with open(path) as f:
            contents = f.read()
    except:
        if debug.print_to_browser: print("Unable to open " + path + " for reading.")
        sys.exit("Unable to open " + path + " for reading.")
    return contents

# Accepts a string containing markup'ed text
# Returns a string containing HTML'ed text
def process_markup(text):
    try:
        p = subprocess.Popen(config.markup_processor,
                stdout=subprocess.PIPE,stdin=subprocess.PIPE)
    except:
        if debug.print_to_browser: print("Unable to open markup processor: " +
                config.markup_processor)
        sys.exit("Unable to open markup processor: " + config.markup_processor)
    text = p.communicate(text.encode('UTF-8'))[0]
    return(text.decode('UTF-8'))

def strip_page_metadata(content):
    keyword = template.add_delimiter("END_PAGE_METADATA")
    index = content.find(keyword)
    if index != -1:
        return content[index+len(keyword):]
    else:
        return content

def lookup_page_metadata(key):
    page_file = load_file(config.site_data_prefix + os.environ['REQUEST_URI'] +
            config.markup_file_extension)

    keyword_begin = template.add_delimiter("BEGIN_PAGE_METADATA")
    keyword_end = template.add_delimiter("END_PAGE_METADATA")
    index = page_file.find(keyword_end)
    if index != -1:
        page_file = page_file[len(keyword_begin):index]
    else:
        page_file = ""

    page_metadata = configparser.ConfigParser()
    page_metadata.read_string(page_file)

    if 'DEFAULT' in page_metadata and key in page_metadata['DEFAULT']:
        return page_metadata['DEFAULT'][key]
    else:
        return ""