launchapp.py

#!/usr/bin/env python
import os
import StringIO
import signal
import subprocess
import sys
import time
import threading
import urllib2
import webbrowser
import uuid

source_exec = sys.argv[1]

mac = False
if 'mac' in sys.argv:
    mac = True

path = os.path.abspath(os.path.dirname(__file__))

try:
    os.mkdir('{0}/app/output'.format(path))
except:
    pass

try:
    os.mkdir('{0}/app/static/tmp'.format(path))
except:
    pass

if os.path.isfile('app/update'):
    if mac:
        print "<h2>Updating application now</h2><br />"
    else:
        sys.stdout.write('Updating application now...')
    sys.stdout.flush()

    h = subprocess.Popen('git stash'.split())
    h.communicate()
    if h.returncode != 0:
        os.remove('app/update')
        print "Error in updating, exiting"
        if mac:
            print "<br />"
        sys.stdout.flush()
        exit(-1)

    h = subprocess.Popen('git pull'.split())
    h.communicate()
    if h.returncode != 0:
        os.remove('app/update')
        print "Error in updating, exiting"
        if mac:
            print "<br />"
        sys.stdout.flush()
        exit(-1)

    # print "Success"
    print "Done updating, relaunching {0}...".format(source_exec)
    if mac:
        print "<br />"
    os.remove('app/update')

    sys.stderr.flush()
    sys.stdout.flush()

    os.execl(source_exec, source_exec)

if mac:
    print "<h2>Running StochSS Server</h2><br />"
else:
    print "--- Running StochSS Server ---"

req = None
try:
    req = urllib2.urlopen("http://localhost:8080/")
except:
    pass

if req:
    if req.getcode() == 200:
        if mac:
            print "<font color=red>"

        print "There seems to be another webserver already running on localhost:8080"

        if mac:
            print "</font><br />"

        exit(-1)

h = subprocess.Popen(("python " + path + "/conf/stochss-env.py").split(), stdout=subprocess.PIPE,
                     stderr=subprocess.PIPE)
h.communicate()

stdout = open('stdout.log', 'w')
stderr = open('stderr.log', 'w')

# Deploy the app on localhost
# print path
def startserver():
    import sys
    if '--debug' in sys.argv:
        h = subprocess.Popen((
                         "python " + path + "/sdk/python/dev_appserver.py --host=localhost --datastore_path={0}/mydatastore --skip_sdk_update_check YES --datastore_consistency_policy=consistent --log_level=debug app".format(
                             path)).split(), stdout=stdout, stderr=stderr)
    else:
        h = subprocess.Popen((
                         "python " + path + "/sdk/python/dev_appserver.py --host=localhost --datastore_path={0}/mydatastore --skip_sdk_update_check YES --datastore_consistency_policy=consistent app".format(
                             path)).split(), stdout=stdout, stderr=stderr)


startserver()

print "Starting admin server at: http://localhost:8000"
if mac:
    print "<br />"
sys.stdout.flush()


def clean_up_and_exit(signal, stack):
    print "Killing webserver proces..."
    if mac:
        print "<br /></body></html>"
    sys.stdout.flush()

    try:
        h.terminate()
    except:
        pass

    if signal == None:
        exit(0)
    else:
        exit(-1)


signal.signal(signal.SIGHUP, clean_up_and_exit)
signal.signal(signal.SIGINT, clean_up_and_exit)
signal.signal(signal.SIGQUIT, clean_up_and_exit)
signal.signal(signal.SIGILL, clean_up_and_exit)
signal.signal(signal.SIGABRT, clean_up_and_exit)
signal.signal(signal.SIGFPE, clean_up_and_exit)

# Wait for server to launch
serverUp = False
for tryy in range(0, 20):
    try:
        req = urllib2.urlopen("http://localhost:8080/login")
    # This was a hack in place to get around issue that arose from allowing
    #  users accessing the app from localhost to not have to login.
    # except urllib2.HTTPError as e:
    #     req = None
    #     if e.code == 302 and e.reason.endswith('Found'):
    #         serverUp = True
    #         break
    except Exception:
        req = None

    if req:
        if req.getcode() == 200:
            # This is a strange sleep, but if we get a response from 8080, we must wait to make sure h has time to crash if it is going to
            # If we just barrell through here, we could be accessing another server on 8080, and h could be in the process of crashing
            time.sleep(1)

            ret = h.poll()
            if h.poll() == 0:
                serverUp = True
            else:
                if mac:
                    print "<font color=red>"

                print "There seems to be another webserver already running on localhost:8080"

                if mac:
                    print "</font><br />"

                sys.stdout.flush()
                serverUp = False
            break
        else:
            ret = h.poll()

            # Sometimes the server fails to start for weird reason, make sure it keeps trying to start
            if ret is not None:
                if ret != 0:
                    startserver()

    time.sleep(2)
    print "Checking if launched -- try " + str(tryy + 1) + " of 20"
    if mac:
        print "<br />"
    sys.stdout.flush()

###print serverUp

if serverUp:
    # Create an admin token
    admin_token = uuid.uuid4()
    generate_admin_token_command = './generate_admin_token.py {0}'.format(admin_token)
    os.system(generate_admin_token_command)
    stochss_url = 'http://localhost:8080/login?secret_key={0}'.format(admin_token)
    # Open web browser

    if mac:
        wbrowser = subprocess.Popen('open {0}'.format(stochss_url).split())
        wbrowser.communicate()
    else:
        webbrowser.open_new(stochss_url)

    if mac:
        print " Stdout available at {0}/stdout.log and <br />".format(path)
        print " Stderr available at {0}/stderr.log<br />".format(path)
        print "<br />"
    else:
        print "Logging stdout to " + path + "/stdout.log\n" + " and stderr to " + path + "/stderr.log"
    sys.stdout.flush()

    try:
        print "Navigate to {0} to access StochSS".format(stochss_url)
        if mac:
            print "<br />"

        sys.stdout.flush()
        if not mac:
            print "Press Control+C to terminate StochSS server"
            sys.stdout.flush()

        while 1:
            time.sleep(1)
    except KeyboardInterrupt:
        pass

    clean_up_and_exit(None, None)
else:
    if mac:
        print "<font color=red>"
    print "Webserver never launched, cleaning up processes and exiting. Check " + path + "/stderr.log"
    if mac:
        print "<br />"
    sys.stdout.flush()
    clean_up_and_exit(None, None)