run

#!/usr/bin/python3

import json
import os
import shlex
import subprocess
import sys
import threading

from http.server import ThreadingHTTPServer
from http.server import SimpleHTTPRequestHandler

sys.path.append("scripts")
import data_util

MAPBOX_API_VARNAME = "MAPBOX_API_TOKEN"
with open("config.json") as f:
    CONFIG = json.loads(f.read())
    f.close()

if len(sys.argv) < 2:
    print("Please give me the disease ID as an argument, for instance 'covid-19'")
    sys.exit(1)

DISEASE_ID = sys.argv[1]

class VizHandler(SimpleHTTPRequestHandler):

    def send_head(self):
        path = self.translate_path(self.path)
        if "favicon.ico" in path:
            self.send_response(404)
            self.end_headers()
            return None
        f = None
        if os.path.isdir(path):
            if not self.path.endswith('/'):
                self.send_response(301)
                self.send_header("Location", self.path + "/")
                self.end_headers()
                return None
            for index in "index.html", "index.htm":
                index = os.path.join(path, index)
                if os.path.exists(index):
                    path = index
                    break
            else:
                return self.list_directory(path)

        binary = path.endswith(".gif") or path.endswith(".png")
        f = open(path, "rb" if binary else "r")
        contents = f.read()
        if "diseasemap.js" in self.path:
            if MAPBOX_API_VARNAME in os.environ:
                contents = contents.replace("{{MAPBOX_API_TOKEN}}",
                                            os.environ[MAPBOX_API_VARNAME])
        if "sidebar.js" in self.path or "view.js" in self.path:
            contents = contents.replace(
                "{{TITLE}}", CONFIG[DISEASE_ID]["name"])
        if "sidebar.js" in self.path:
            other_diseases = []
            for disease_id in CONFIG[DISEASE_ID]["linkto"]:
                other_diseases.append("|".join([
                    disease_id, CONFIG[disease_id]["name"], CONFIG[disease_id]["url"]]))
            contents = contents.replace(
                "{{OTHER_DISEASES}}", ",".join(other_diseases))
        if "main.js" in self.path:
            destination = CONFIG[DISEASE_ID]["data_src_url"]
            if not destination.endswith("/"):
                destination += "/"
            contents = contents.replace("{{DATA_SRC_URL}}", destination)
        encoded_contents = contents if binary else contents.encode()
        ctype = self.guess_type(path)
        self.send_response(200)
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header("Content-type", ctype)
        fs = os.fstat(f.fileno())
        self.send_header("Content-Length", str(len(encoded_contents)))
        self.send_header("Last-Modified", self.date_time_string(fs.st_mtime))
        self.end_headers()
        self.wfile.write(encoded_contents)
        # We've taken care of everything, nothing left to do.
        return None

# Returns True if everything we need for sass is here, False otherwise.
def check_sass():
    try:
        subprocess.check_call(shlex.split("sass --version"))
    except (subprocess.CalledProcessError, OSError):
        print("WARNING: 'sass' is not installed, CSS will not update on the fly")
        return False
    return True

def run_sass_precompiler():
    input_files = [f for f in os.listdir("css") if f.endswith(".scss")]
    if not len(input_files):
        return None
    return subprocess.call(shlex.split("sass --watch css:css"))


def run_http_server():
    server = ThreadingHTTPServer(("localhost", 8000), VizHandler)
    server.serve_forever()

def run():
    data_util.make_country_pages()

    if MAPBOX_API_VARNAME not in os.environ:
        print("Please set the 'MAPBOX_API_TOKEN' to your development token. "
              "Aborting.")
        print(os.environ)
        sys.exit(1)

    try:
        http = threading.Thread(target=run_http_server)
        sass = threading.Thread(target=run_sass_precompiler)
        http.start()
        if check_sass():
            sass.start()
            sass.join()
        http.join()

    except KeyboardInterrupt:
        print("Hit Crtl-C a second time to shut down.")
        sys.exit(0)

if __name__ == '__main__':
    run()