X-Git-Url: https://git.subgeniuskitty.com/cmless/.git/blobdiff_plain/d955cfad358f87bc92b6f3cb2d1741e5dbb61cc5..78e2ec48966bc0c60db77c4e7a46f866c89c5351:/bin/misc.py

diff --git a/bin/misc.py b/bin/misc.py
index 821ef34..a58d5be 100644
--- a/bin/misc.py
+++ b/bin/misc.py
@@ -2,14 +2,13 @@
 # See LICENSE file for copyright and license details.
 
 # Python imports
-import sys, subprocess
+import os, sys, subprocess, configparser
 
 # CMless imports
-import debug, config
+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):
+    """Open the text file at 'path' and return the contents as a string."""
     try:
         with open(path) as f:
             contents = f.read()
@@ -18,9 +17,12 @@ def load_file(path):
         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):
+    """Execute external markup processor and return processed 'text'.
+
+    Markup processor is specified by config.markup_processor variable
+    and should accept markup on stdin while returning HTML on stdout.
+    """
     try:
         p = subprocess.Popen(config.markup_processor,
                 stdout=subprocess.PIPE,stdin=subprocess.PIPE)
@@ -30,3 +32,47 @@ def process_markup(text):
         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):
+    """Strip metadata from the beginning of 'content' and return whatever remains."""
+    keyword = template.add_delimiter("END_PAGE_METADATA")
+    index = content.find(keyword)
+    if index != -1:
+        return content[index+len(keyword):]
+    else:
+        return content
+
+def lookup_current_metadata(key):
+    """Wrapper for lookup_file_metadata() supplying the current REQUEST_URI as 'file'."""
+    file = config.site_data_prefix + os.environ['REQUEST_URI'] + config.markup_file_extension
+    return lookup_file_metadata(key,file)
+
+def lookup_file_metadata(key,file):
+    """Read 'key' metadata in 'file' and return corresponding value.
+
+    Assumes key is contained in the [DEFAULT] category.
+    Returns empty string if key is not present.
+    """
+    page_file = load_file(file)
+
+    # Extract just the metadata block so we can pass it to ConfigParser.
+    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:
+        # ConfigParser will accept an empty string so use that as fallback.
+        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:
+        # An empty string is technically a valid value for a key but we will
+        # overload it as also indicating 'key' wasn't present since a key
+        # without a value is functionally equivalent to no key at all for our
+        # purposes.
+        return ""