mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-04 05:53:43 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			75 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			75 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
#!/usr/bin/python -u
 | 
						|
import os
 | 
						|
import sys
 | 
						|
import subprocess
 | 
						|
import logging
 | 
						|
import datetime
 | 
						|
import shutil
 | 
						|
import time
 | 
						|
 | 
						|
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
 | 
						|
from zulip_tools import DEPLOYMENTS_DIR, LOCK_DIR, FAIL, WARNING, ENDC, \
 | 
						|
    su_to_zulip
 | 
						|
 | 
						|
logging.basicConfig(format="%(asctime)s upgrade-zulip: %(message)s",
 | 
						|
                    level=logging.INFO)
 | 
						|
 | 
						|
if os.getuid() != 0:
 | 
						|
    logging.error("Must be run as root.")
 | 
						|
    sys.exit(1)
 | 
						|
 | 
						|
if len(sys.argv) != 2:
 | 
						|
    print FAIL + "Usage: %s <tarball>" % (sys.argv[0],) + ENDC
 | 
						|
    sys.exit(1)
 | 
						|
 | 
						|
tarball_path = sys.argv[1]
 | 
						|
 | 
						|
start_time = time.time()
 | 
						|
got_lock = False
 | 
						|
while time.time() - start_time < 300:
 | 
						|
    try:
 | 
						|
        os.mkdir(LOCK_DIR)
 | 
						|
        got_lock = True
 | 
						|
        break
 | 
						|
    except OSError:
 | 
						|
        print WARNING + "Another deployment in progress; waiting for lock..." + ENDC
 | 
						|
        time.sleep(10)
 | 
						|
 | 
						|
if not got_lock:
 | 
						|
    print FAIL + "Deployment already in progress.  Please run\n" \
 | 
						|
               + "  %s/current/scripts/upgrade-zulip %s\n" % (DEPLOYMENTS_DIR, tarball_path) \
 | 
						|
               + "manually when the previous deployment finishes, or run\n" \
 | 
						|
               + "  rmdir %s\n"  % (LOCK_DIR,) \
 | 
						|
               + "if the previous deployment crashed." \
 | 
						|
               + ENDC
 | 
						|
    sys.exit(1)
 | 
						|
 | 
						|
logging.info("Unpacking the tarball")
 | 
						|
deploy_path = subprocess.check_output([os.path.join(os.path.dirname(__file__), 'unpack-zulip'),
 | 
						|
                                       tarball_path], preexec_fn=su_to_zulip)
 | 
						|
deploy_path = deploy_path.strip()
 | 
						|
os.chdir(deploy_path)
 | 
						|
 | 
						|
logging.info("Upgrading system packages...")
 | 
						|
subprocess.check_call(["apt-get", "update"])
 | 
						|
subprocess.check_call(["apt-get", "upgrade"])
 | 
						|
 | 
						|
logging.info("Stopping Zulip...")
 | 
						|
subprocess.check_call(["supervisorctl", "stop", "zulip-workers:*", "zulip-django",
 | 
						|
                       "zulip-tornado"], preexec_fn=su_to_zulip)
 | 
						|
 | 
						|
logging.info("Applying puppet changes...")
 | 
						|
subprocess.check_call(["./scripts/zulip-puppet-apply", "--force"])
 | 
						|
subprocess.check_call(["apt-get", "upgrade"])
 | 
						|
 | 
						|
logging.info("Applying database migrations...")
 | 
						|
subprocess.check_call(["./manage.py", "migrate"], preexec_fn=su_to_zulip)
 | 
						|
 | 
						|
logging.info("Restarting server...")
 | 
						|
subprocess.check_call(["cp", "-rT", os.path.join(deploy_path, 'prod-static/serve'),
 | 
						|
                        '/home/zulip/prod-static'], preexec_fn=su_to_zulip)
 | 
						|
subprocess.check_output(["./scripts/restart-server"], preexec_fn=su_to_zulip)
 | 
						|
 | 
						|
logging.info("Deployment complete")
 | 
						|
shutil.rmtree(LOCK_DIR)
 |