root/hacks/trunk/enrico.py

#!/usr/bin/python
"""Source code file to www.enricomoretti.it

Generates a tree of HTML files and their navigation using Django's
templating system and Jason Orendorff path.py"""

import os 
import sys
from path import path
from django.conf import settings
# set up boilerplate Django settings
settings.configure(
    DEBUG=True, TEMPLATE_DEBUG=True, TEMPLATE_DIRS=('templates',))
from django.template import Context
from django.template.loader import get_template

__author__ = "Antonio Cavedoni <antonio@cavedoni.org>"  
__svnid__ = "$Id$"

IGNORED_DIRS = (
    '_private', '_vti_log', '2006', 'about', 'aspnet_client', 
    'bin', 'cgi-bin', 'images', 'mdb-database', 'styles', 
    'templates'
    )
IGNORED_FILES = (
    '.DS_Store', '.htaccess', '_vti_inf.html'
    )

def render_to_file(context, template, filename):
    """Render context to filename with template"""
    t = get_template(template)
    f = open(filename, 'w')
    f.write(t.render(context))
    f.close()

def deslugify(name):
    """Transform a directory name into a (somewhat) human-readable title"""
    bits = name.split('-')
    bits[0] = bits[0].capitalize()
    return " ".join(bits)

def get_absolute_path(name):
    """Given a relative directory path, make it absolute"""
    return name.replace('./', '/')

def has_contents(directory):
    """Check whether a folder has files (.jpg or .mov) or subdirectories"""
    return directory.files("*.jpg") or directory.files("*.mov") \
        or directory.dirs()

def htmlize(directory):
    """Return a link to a directory, if it has contents"""
    absolute_dir = get_absolute_path(directory)
    if has_contents(directory):
        return '<a href="%s/">%s</a><br>\n' % \
            (absolute_dir, deslugify(directory.name))
    else:
        return "%s<br>\n" % deslugify(directory.name)

def generate_menu(startpath, endpath=None):
    """Generate an HTML menu for a directory. If the parameter `endpath`
    is set, generate a tree from startpath to endpath"""

    # @@ I should probably just get rid of path.py
    _path = path(startpath)

    # default value of endpath is startpath
    if not endpath: 
        endpath = startpath

    output = []

    for directory in _path.walkdirs():
        # filter out unwanted dirs
        if directory.namebase in IGNORED_DIRS: continue

        # gather nesting information on the directories
        end_levels = len(endpath.split('/'))
        dir_levels = len(directory.split('/'))
        level_diff = end_levels - dir_levels

        # always print top-level directories
        if directory.parent == startpath:
            output.append(htmlize(directory))
        # all the directories that are not in the root
        elif dir_levels > 2:
            if directory.parent in endpath and level_diff >= -1:
                output.append("&nbsp; &nbsp; " * (dir_levels - 2))
                output.append(htmlize(directory))

    return " ".join(output)

def generate_sections(startpath):
    """Generate index pages and single files for a photo/movie gallery"""

    _path = path(startpath)
    dirs = [x for x in _path.walkdirs() if x.name not in IGNORED_DIRS]

    def generate_files(_files, ext):
        for i in range(len(_files)):
            try:
                next_link = _files[i+1].replace(ext, '.html')
            except IndexError:
                next_link = None
            try:
                prev_link = _files[i-1].replace(ext, '.html')
            except IndexError:
                prev_link = None
            absolute_dir = get_absolute_path(directory)
            c['next_link'] = "%s/%s" % \
                (absolute_dir, next_link)
            c['prev_link'] = "%s/%s" % \
                (absolute_dir, prev_link)
            c['src'] = "%s/%s" % \
                (absolute_dir, _files[i])

            outfile = "%s/%s" % \
                (directory, _files[i].replace(ext, '.html'))

            if i == 0:
                del c['prev_link']
                outfile = "%s/%s" % (directory, 'index.html')
            elif i == 1:
                c['prev_link'] = "./"
            elif i == (len(_files) - 1):
                del c['next_link']

            if ext == '.jpg':
                render_to_file(c, 'picture.tmpl', outfile)
            elif ext == '.mov':
                render_to_file(c, 'quicktime.tmpl', outfile)

    for directory in dirs:
        c = Context({'sections': generate_menu(startpath, directory)})
        render_to_file(c, 'picture.tmpl', 
                       "%s/%s" % (directory, 'index.html'))
        jpgs = [x.name for x in directory.files("*.jpg")]
        movs = [x.name for x in directory.files("*.mov")]
        generate_files(jpgs, ".jpg")
        generate_files(movs, ".mov")

if __name__ == "__main__":
    startpath = path('.')
    c = Context({'sections': generate_menu(startpath)})
    render_to_file(c, 'base.tmpl', "./index.html")
    render_to_file(c, 'about.tmpl', "./about/index.html")
    generate_sections(startpath)