root/branches/demo-site/configure.py

#!/usr/bin/env python

import os
import sys
import random
import string
from datetime import datetime

PROGRAM_NAME = 'configure.py'
PROGRAM_VERSION = 0.1

BASE_PATH = os.path.abspath(os.path.split(__file__)[0])
SRC_PATH = os.path.join(BASE_PATH, 'local_settings.py.dist')
DIST_PATH = os.path.join(BASE_PATH, 'local_settings.py')

ACCEPT_CHARS = 'abcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*(-_=+)'

def _program_name():
    return '%s %s' % (PROGRAM_NAME, PROGRAM_VERSION)

def _failed_message(msg=None):
    msg = msg or ''
    print '\033[41;37m Error: %s \033[00m' % msg
    sys.exit(1)

def _success_message(msg=None):
    msg = msg or ''
    print '\033[36m%s\033[00m' % msg

def configure():
    """
    generate local settings file.
    """
    #   Check source file.
    if not os.path.exists(SRC_PATH) or not os.path.isfile(SRC_PATH):
        _failed_message("Can't find source file %s" % SRC_PATH)
    #   Setup parameters.
    parameters = {
        'GENERATED_BY': _program_name(),
        'GENERATED_AT': datetime.now().strftime('%Y/%m/%d %H:%M:%S'),
        'SECRET_KEY': ''.join([random.choice(ACCEPT_CHARS) for x in range(50)]),
    }
    #   read file.
    buf = open(SRC_PATH, 'r').read()
    for k, v in parameters.items():
        buf = buf.replace('@%s@' % k, v)
    #   write buffer.
    fp = open(DIST_PATH, 'w')
    fp.write(buf)
    fp.close()
    _success_message('process complete.')

if __name__ == '__main__':
    configure()