create_app.py

#!/usr/bin/env python

import os, random, subprocess, sys

HAS_VENV = bool(subprocess.Popen(['which','virtualenv'], stdout=subprocess.PIPE).communicate()[0])
if not HAS_VENV:
    print "virtualenv is required to run this script. Please install it with\n  easy_install virtualenv\n\nor\n\n  pip virtualenv"
    sys.exit(1)

HAS_VENVW = bool(subprocess.Popen(['which','virtualenvwrapper.sh'], stdout=subprocess.PIPE).communicate()[0])
if not HAS_VENVW:
    print "virtualenvwrapper is required to run this script. Please install it with\n  easy_install virtualenvwrapper\n\nor\n\n  pip virtualenvwrapper"
    sys.exit(1)

CHARS = 'abcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*(-_=+)'
BLACKLIST = (
    'jquery',
    '.tar.gz',
    'admin/css',
    'admin/img',
    'admin/js',
    '/.git/',
    '.svn',
    '.hg',
)

def replace(repl, text):
    text = text.replace('/gitignore', '/.gitignore')
    for key, value in repl.iteritems():
        text = text.replace('$$$$%s$$$$' % (key,), value)
    return text

def main(repl, dest, templ_dir):
    try:
        os.makedirs(dest)
    except OSError:
        pass
    
    for root, dirs, files in os.walk(templ_dir):
        for filename in files:
            source_fn = os.path.join(root, filename)
            dest_fn = replace(repl, os.path.join(dest, root.replace(templ_dir, ''), replace(repl, filename)))
            try:
                os.makedirs(os.path.dirname(dest_fn))
            except OSError:
                pass
            print 'Copying %s to %s' % (source_fn, dest_fn)
            should_replace = True
            for bl_item in BLACKLIST:
                if bl_item in dest_fn:
                    should_replace = False
            data = open(source_fn, 'r').read()
            if should_replace:
                data = replace(repl, data)
            open(dest_fn, 'w').write(data)
            os.chmod(dest_fn, os.stat(source_fn)[0])
    
    print "Making the virtual environment (%s)..." % repl['VIRTENV']
    create_env_cmds = [
        'source virtualenvwrapper.sh', 
        'cd %s' % dest,
        'mkvirtualenv --no-site-packages --distribute %s' % repl['VIRTENV'],
        'easy_install pip'
    ]
    create_pa_cmd = [
        'source virtualenvwrapper.sh',
        'cat > $WORKON_HOME/%s/bin/postactivate '\
        '<<END\n#!/bin/bash/\ncd %s\nEND\n'\
        'chmod +x $WORKON_HOME/%s/bin/postactivate' % (repl['VIRTENV'], dest,repl['VIRTENV'])
    ]
    subprocess.call([';'.join(create_env_cmds)], env=os.environ, executable='/bin/bash', shell=True)
    subprocess.call([';'.join(create_pa_cmd)], env=os.environ, executable='/bin/bash', shell=True)
    
    print "Now type: workon %s" % repl['VIRTENV']

if __name__ == '__main__':
    from optparse import OptionParser
    parser = OptionParser()
    parser.add_option("-a", "--author", dest="author", help="The name of the author.")
    parser.add_option("-e", "--author-email", dest="author_email", help="The email of the author.")
    parser.add_option("-u", "--url", dest="url", help="The URL of the project page.")
    parser.add_option("-n", "--name", dest="app_name", help="The name of the application, like 'django-coolapp'.")
    parser.add_option("-p", "--package", dest="pkg_name", help="The name of the installed package, like 'coolapp'.")
    parser.add_option("-v", "--VIRTENV", dest="VIRTENV", help="The name of the virtualenv.")
    parser.add_option("-d", "--dest", dest="destination", help="Where to put the new application. Relative paths are recognized.")
    parser.add_option("-t", "--template", dest="template", help="The application template to use as a basis for the new application.")
    (options, args) = parser.parse_args()
    
    repl = {
        'APP_NAME': None,
        'PKG_NAME': None,
        'AUTHOR': None,
        'AUTHOR_EMAIL': None,
        'URL': None,
    }
    dest_dir = None
    templ_dir = None
    
    cur_user = os.getlogin()
    
    if options.app_name:
        repl['APP_NAME'] = options.app_name
    elif len(args) > 0:
        repl['APP_NAME'] = args[0]
    
    while not repl['APP_NAME']:
        repl['APP_NAME'] = raw_input('Application name: ')
    
    if options.pkg_name:
        repl['PKG_NAME'] = options.pkg_name
    while not repl['PKG_NAME']:
        default_name = repl['APP_NAME'].replace('django-', '').replace('-','_')
        repl['PKG_NAME'] = raw_input('Package Name [%s]:' % default_name) or default_name
    
    if options.author:
        repl['AUTHOR'] = options.author
    while not repl['AUTHOR']:
        repl['AUTHOR'] = raw_input('Author [%s]:' % cur_user) or cur_user

    if options.author:
        repl['AUTHOR_EMAIL'] = options.author_email
    while not repl['AUTHOR_EMAIL']:
        repl['AUTHOR_EMAIL'] = raw_input('Author\'s Email: ')
    
    if options.url:
        repl['URL'] = options.url
    while not repl['URL']:
        repl['URL'] = raw_input('Project Page URL: ')
    
    repl['SECRET_KEY'] = ''.join([random.choice(CHARS) for i in xrange(50)])
    
    if options.destination:
        dest_dir = options.destination
    
    while not dest_dir:
        dest_dir = raw_input('Destination directory [%s]: ' % (os.getcwd(),)) or os.getcwd()
    dest_dir =  os.path.realpath(os.path.expanduser(dest_dir))
    dest = os.path.join(dest_dir, repl['APP_NAME'])

    if options.template:
        templ_dir = options.template

    default = os.path.abspath(os.path.join(os.path.dirname(__file__), 'skel'))
    while not templ_dir:
        templ_dir = raw_input('Application template directory [%s]: ' % default) or default
    templ_dir = os.path.realpath(os.path.expanduser(templ_dir))
    if templ_dir[-1] != '/':
        templ_dir = templ_dir + "/"
    
    if options.VIRTENV:
        repl['VIRTENV'] = options.VIRTENV
    else:
        repl['VIRTENV'] = None
    while not repl['VIRTENV']:
        repl['VIRTENV'] = raw_input('Virtual environment name [%s]: ' % repl['APP_NAME']) or repl['APP_NAME']

    main(repl, dest, templ_dir)