zulip/tools/update-deployment

#!/usr/bin/env python
import os
import sys
import subprocess
import pylibmc
import traceback
import logging
from humbug_tools import check_output
import datetime
import shutil

logging.basicConfig(format="%(asctime)s update-deployment: %(message)s",
                    level=logging.INFO)

# Color codes
OKBLUE = '\033[94m'
OKGREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'

LOCK_DIR = "/home/humbug/humbug-deployments/lock"

os.chdir("/home/humbug/humbug")
if len(sys.argv) > 1:
    oldrev = sys.argv[1]
    newrev = sys.argv[2]
    refname = sys.argv[3]

subprocess.check_call(["mkdir", '-p',
                       '/home/humbug/humbug-deployments',
                       '/home/humbug/logs'])

try:
    os.mkdir(LOCK_DIR)
except OSError:
    print FAIL + "Deployment already in progress.  Please run\n" \
               + "  update-deployment %s %s %s\n" % (oldrev, newrev, refname) \
               + "manually when the current deployment finishes." + ENDC
    sys.exit(1)

subprocess.check_call(["git", "fetch"], stdout=open('/dev/null', 'w'))
subprocess.check_call(["git", "reset", "--hard", refname], stdout=open('/dev/null', 'w'))

# Delete all .pyc files to avoid old module files hanging around
subprocess.check_call(["find", ".", "-name", "*.pyc", "-delete"], stdout=open('/dev/null', 'w'))

# Update static files
logging.info("Updating static files")
subprocess.check_call(["./tools/update-prod-static"])

timestamp = datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')
deploy_path = "/home/humbug/humbug-deployments/%s" % (timestamp,)
subprocess.check_call(["cp", '-a', '/home/humbug/humbug', deploy_path])
subprocess.check_call(["ln", '-nsf', deploy_path, "/home/humbug/humbug-deployments/current"])

logging.info("Restarting server...")
subprocess.check_call(["/home/humbug/humbug-deployments/current/tools/restart-server"])


if newrev == '0000000000000000000000000000000000000000':
    # 0000000000000000000000000000000000000000 means we're deleting the ref
    commits = ''
else:
    commits = check_output(["git", "log", "%s..%s" % (oldrev, newrev)])

if '[schema]' in commits:
    print
    print FAIL + "Schema change detected!  Please make the appropriate changes manually." + ENDC
    print

shutil.rmtree(LOCK_DIR)