zulip/tools/update-deployment

71 lines
2.3 KiB
Python
Executable File

#!/usr/bin/python
import os
import sys
import subprocess
import pylibmc
import traceback
def check_output(*popenargs, **kwargs):
if 'stdout' in kwargs:
raise ValueError('stdout argument not allowed, it will be overridden.')
process = subprocess.Popen(stdout=subprocess.PIPE, *popenargs, **kwargs)
output, unused_err = process.communicate()
retcode = process.poll()
if retcode:
cmd = kwargs.get("args")
if cmd is None:
cmd = popenargs[0]
raise subprocess.CalledProcessError(retcode, cmd, output=output)
return output
subprocess.check_output = check_output
# Color codes
OKBLUE = '\033[94m'
OKGREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'
os.chdir("/home/humbug/humbug")
if len(sys.argv) > 1:
oldrev = sys.argv[1]
newrev = sys.argv[2]
refname = sys.argv[3]
try:
if not pylibmc.Client(['127.0.0.1']).flush_all():
print "%sflush_all returned False%s" % (WARNING, ENDC)
# There doesn't seem to be a method to close a pylibmc Client object.
except:
print "%sCould not flush cache:%s" % (WARNING, ENDC)
traceback.print_exc()
subprocess.check_call(["git", "fetch"], stdout=open('/dev/null', 'w'))
subprocess.check_call(["git", "reset", "--hard", refname], stdout=open('/dev/null', 'w'))
#subprocess.check_call(["python", "manage.py", "syncdb"], 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'))
# Restart the FastCGI process, which is running in a shell loop in screen.
# TODO: real daemonization
for cmd in ('runfcgi', 'runtornado'):
try:
subprocess.check_call(["pkill", "-f", "python manage.py " + cmd])
except subprocess.CalledProcessError:
print "%sCould not kill %s; is it running?%s" % (WARNING, cmd, ENDC)
print OKGREEN + "Updated deployed version of humbug application!" + ENDC
if newrev == '0000000000000000000000000000000000000000':
# 0000000000000000000000000000000000000000 means we're deleting the ref
commits = ''
else:
commits = subprocess.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