view piecrust/sources/prose.py @ 380:f33712c4cfab

routing: Fix bugs with matching URLs with correct route but missing metadata. When matching a route like `/foo/%slug%` against an URL like `/foo`, the route will (correctly) return a match, but it will be completely missing the `slug` metadata, resulting in problems elsewhere. This change makes it so that any missing route metadata will be filled in with an empty string. And because this means generated URLs may differ from the incoming URL when using trailing slashes (`/foo/` _vs._ `/foo`), we make the assert in the chef server handle those discrepancies.
author Ludovic Chabant <ludovic@chabant.com>
date Sun, 10 May 2015 00:34:21 -0700
parents 4b1019bb2533
children 65db6df28120
line wrap: on
line source

import os
import os.path
import copy
import logging
from piecrust.sources.base import MODE_CREATING, MODE_PARSING
from piecrust.sources.default import DefaultPageSource


logger = logging.getLogger(__name__)


class ProseSource(DefaultPageSource):
    SOURCE_NAME = 'prose'

    def __init__(self, app, name, config):
        super(ProseSource, self).__init__(app, name, config)
        self.config_recipe = config.get('config', {})

    def _populateMetadata(self, rel_path, metadata, mode=None):
        metadata['config'] = self._makeConfig(rel_path, mode)

    def _makeConfig(self, rel_path, mode):
        c = copy.deepcopy(self.config_recipe)
        if c.get('title') == '%first_line%' and mode != MODE_CREATING:
            path = os.path.join(self.fs_endpoint_path, rel_path)
            try:
                c['title'] = get_first_line(path)
            except OSError:
                if mode == MODE_PARSING:
                    raise
        return c


def get_first_line(path):
    with open(path, 'r') as f:
        while True:
            l = f.readline()
            if not l:
                break
            l = l.strip()
            if not l:
                continue
            return l
    return None