summaryrefslogtreecommitdiff
path: root/planet/shell/__init__.py
blob: a65b121b8bda0064328c1e45bc1ad9120e5b37f7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import planet
import os
import sys

logged_modes = []

def run(template_file, doc, mode='template'):
    """ select a template module based on file extension and execute it """
    log = planet.logger

    if mode == 'template':
        dirs = planet.config.template_directories()
    else:
        dirs = planet.config.filter_directories()
 
    # parse out "extra" options
    if template_file.find('?') < 0:
        extra_options = {}
    else:
        import cgi
        template_file, extra_options = template_file.split('?',1)
        extra_options = dict(cgi.parse_qsl(extra_options))

    # see if the template can be located
    for template_dir in dirs:
        template_resolved = os.path.join(template_dir, template_file)
        if os.path.exists(template_resolved): break
    else:
        log.error("Unable to locate %s %s", mode, template_file)
        if not mode in logged_modes:
            log.info("%s search path:", mode)
            for template_dir in dirs:
                log.info("    %s", os.path.realpath(template_dir))
            logged_modes.append(mode)
        return
    template_resolved = os.path.abspath(template_resolved)

    # Add shell directory to the path, if not already there
    shellpath = os.path.join(sys.path[0],'planet','shell')
    if shellpath not in sys.path:
        sys.path.append(shellpath)

    # Try loading module for processing this template, based on the extension
    base,ext = os.path.splitext(os.path.basename(template_resolved))
    module_name = ext[1:]
    try:
        try:
            module = __import__("_" + module_name)
        except:
            module = __import__(module_name)
    except Exception, inst:
        return log.error("Skipping %s '%s' after failing to load '%s': %s", 
            mode, template_resolved, module_name, inst)

    # Execute the shell module
    options = planet.config.template_options(template_file)
    if module_name == 'plugin': options['__file__'] = template_file
    options.update(extra_options)
    log.debug("Processing %s %s using %s", mode,
        os.path.realpath(template_resolved), module_name)
    if mode == 'filter':
        return module.run(template_resolved, doc, None, options)
    else:
        output_dir = planet.config.output_dir()
        output_file = os.path.join(output_dir, base)
        module.run(template_resolved, doc, output_file, options)
        return output_file