#!/usr/bin/env python
import os
import sys
import subprocess
import pylibmc
import traceback
import logging
import time

logging.basicConfig(format="%(asctime)s restart-server: %(message)s",
                    level=logging.INFO)
# Color codes
OKBLUE = '\033[94m'
OKGREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'

os.chdir("/home/humbug/humbug-deployments/current")

# Send a statsd event on restarting the server
subprocess.check_call(["python", "./manage.py", "send_stats", "incr", "events.server_restart", str(int(time.time()))])

# Restart the FastCGI and related processes via supervisorctl.  We
# minimize downtime by restarting Django and Tornado separately from
# the other worker processes.
logging.info("Killing daemons")
subprocess.check_call(["supervisorctl", "stop",    "humbug-workers:*"])
subprocess.check_call(["supervisorctl", "stop",    "humbug-django"])
subprocess.check_call(["supervisorctl", "restart", "humbug-tornado"])
subprocess.check_call(["supervisorctl", "start",   "humbug-django"])
subprocess.check_call(["supervisorctl", "start",   "humbug-workers:*"])

logging.info("Flushing memcached")
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()

logging.info("Refilling memcached caches")
subprocess.check_call(["python", "./manage.py", "fill_memcached_caches"])

logging.info("Done!")
print OKGREEN + "Application restarted successfully!" + ENDC