ssi_expander.py

#!/usr/bin/env python
'''
This script copies the files in one directory to another, expanding any SSI
<!-- #include --> statements it encounters along the way.

Usage:

  ./ssi_expander.py [source_directory] destination_directory

If source_directory is not specified, then the current directory is used.
'''

import os
import ssi
import sys

def process(source, dest):
  for dirpath, dirnames, filenames in os.walk(source):
    dest_dir = os.path.realpath(os.path.join(dest, os.path.relpath(dirpath, source)))
    os.mkdir(dest_dir)
    for filename in filenames:
      src_path = os.path.abspath(os.path.join(dirpath, filename))
      dest_path = os.path.join(dest_dir, filename)
      if not filename.endswith('.html'):
        os.symlink(src_path, dest_path)
      else:
        file(dest_path, 'w').write(ssi.InlineIncludes(src_path, "/%s" % os.path.relpath(src_path)))

    # ignore hidden directories
    for dirname in dirnames[:]:
      if dirname.startswith('.'):
        dirnames.remove(dirname)


if __name__ == '__main__':
  if len(sys.argv) == 2:
    source = '.'
    dest = sys.argv[1]
  elif len(sys.argv) == 3:
    source, dest = sys.argv[1:]
  else:
    sys.stderr.write('Usage: %s [source_directory] destination_directory\n' % sys.argv[0])
    sys.exit(1)

  process(source, dest)