manage.py

#!/usr/bin/env python


def _check_git_status():
    # check clean working copy using 'git status'
    proc = subprocess.Popen(["git", "status", "--porcelain"], stdout=subprocess.PIPE)
    git_status_result = proc.communicate()[0]
    if len(git_status_result) > 0:
        sys.stderr.write("Uncommitted changes, please clean up first:\n")
        sys.stdout.write(git_status_result)
        sys.exit(1)


def _pull():
    subprocess.check_call(["git", "pull", "origin", "master"])


def _push(tag):
    subprocess.check_call(["git", "push", "origin", "master"])
    subprocess.check_call(["git", "push", "origin", tag])


def _commit_prod_deploy(version):
    subprocess.check_call(["git", "commit", "-a", "-m", "Prepare prod deployment, version " + str(version)])


def _commit_dev(next_version):
    subprocess.check_call(["git", "commit", "-a", "-m", "Prepare development, next version " + str(next_version)])


def _tag_prod_deploy(version):
    tag = "v"+str(version)
    subprocess.check_call(["git", "tag", "-a", tag, "-m", "Tag version " + str(version)])
    sys.stdout.write("Created tag " + tag + " for prod deployment\n")
    return tag


def _deploy():
    #subprocess.check_call(["./manage.py", "deploy", "--nosyncdb"])
    subprocess.check_call(["./manage.py", "deploy"])


def _update_app_yaml_for_prod_deploy():
    '''
    Updates 'app.yaml', sets the application to 'techism2'.
    '''
    from yaml import load, dump
    from yaml import Loader, Dumper
    infile = file('app.yaml')
    data = load(infile, Loader=Loader)
    infile.close()
    data['application'] = "techism2"
    data['version'] = str(data['version']).split('-devel')[0]
    outfile = file('app.yaml', 'w')
    dump(data, outfile)
    outfile.close()
    version = data['version']
    sys.stdout.write("Updated app.yaml for prod deployment, version " + str(version) + "\n")
    return version


def _update_app_yaml_for_dev():
    '''
    Updates 'app.yaml', sets the application to 'techism2-devel' and increments the version and append '-devel'.
    '''
    from yaml import load, dump
    from yaml import Loader, Dumper
    infile = file('app.yaml')
    data = load(infile, Loader=Loader)
    infile.close()
    data['application'] = "techism2-devel"
    data['version'] = str(int(data['version']) + 1) + "-devel"
    outfile = file('app.yaml', 'w')
    dump(data, outfile)
    outfile.close()
    next_version = data['version']
    sys.stdout.write("Updated app.yaml for development, next version " + str(next_version) + "\n")
    return next_version


def _update_version_html(version):
    '''
    Updates the 'templates/version.html'.
    '''
    infile = file('templates/version.html')
    newlines = []
    for line in infile:
        if 'version.number' in line:
            newlines.append("  <span id=\"version.number\">Version %s,</span>\n" % str(version))
        elif 'version.tag' in line:
            newlines.append("  <span id=\"version.tag\"><a href=\"http://github.com/cko/techism2/tree/v%s\">Git Tag</a>,</span>\n" % str(version))
        elif 'version.date' in line:
            newlines.append("  <span id=\"version.date\">%s</span>\n" % date.today().isoformat())
        else:
            newlines.append(line)
    infile.close()
    outfile = file('templates/version.html', 'w')
    outfile.writelines(newlines)
    outfile.close()
    sys.stdout.write("Updated version.html, version " + str(version) + "\n")


# Initialize App Engine SDK first
from djangoappengine.boot import setup_env
setup_env()

import sys, subprocess
from datetime import date
from django.core.management import execute_manager
import settings # Assumed to be in the same directory.


if __name__ == "__main__":
    proddeploy = False
    argv = sys.argv
    if len(sys.argv) == 2 and sys.argv[1] == "proddeploy":
        proddeploy = True
        argv = [sys.argv[0], "deploy"]
    
    if proddeploy:
        _pull()
        _check_git_status()
        version = _update_app_yaml_for_prod_deploy()
        _update_version_html(version)
        _commit_prod_deploy(version)
        tag = _tag_prod_deploy(version)
    
    execute_manager(settings, argv)
    
    if proddeploy:
        next_dev_version = _update_app_yaml_for_dev()
        _update_version_html(next_dev_version)
        _commit_dev(next_dev_version)
        _push(tag)
        sys.stdout.write("Prod deployment successful, check http://%s.latest.techism2.appspot.com and update default version in App Engine console.\n" % version)