dispatch.py

import tornado.web
import json
import os
import starlight
import time
try:
    from plop.collector import Collector, PlopFormatter
except ImportError:
    pass

ROUTES = []

def conditional_route(yes, reason, *regexes):
    if yes:
        return route(*regexes)
    else:
        print(reason)
        return lambda func: func


def route(*regexes):
    def wrapper(handler):
        for regex in regexes:
            ROUTES.append((regex, handler))
        return handler
    return wrapper


def expose_static_json(path, an_object):
    precomputed = json.dumps(an_object)

    class ret(tornado.web.RequestHandler):
        def get(self):
            self.set_header("Content-Type", "application/json")
            self.set_header("Cache-Control", "no-cache")
            self.set_header("Expires", "0")
            self.write(precomputed)

    route(path)(ret)


def dev_mode_only(wrapped):
    if os.environ.get("DEV", ""):
        return wrapped

    class not_dev_error(tornado.web.RequestHandler):
        def get(self, *args):
            self.set_status(400)
            self.set_header("Content-Type", "text/plain; charset=utf-8")
            self.write(
                "The requested endpoint is only available in development mode.")

    return not_dev_error


class HandlerSyncedWithMaster(tornado.web.RequestHandler):
    def prepare(self):
        starlight.data.reset_statistics()
        self.did_trigger_update = starlight.update.check_version()

        super().prepare()

        if self.get_argument("profile", None) and os.environ.get("ALLOW_PROFILING"):
            self.collector = Collector()
            self.collector.start()

    def finish(self, *args, **kw):
        super().finish(*args, **kw)

        if self.get_argument("profile", None) and os.environ.get("ALLOW_PROFILING"):
            self.collector.stop()
            formatter = PlopFormatter(max_stacks=9001)
            if self.collector.samples_taken:
                formatter.store(self.collector, "{0}_{1}.profile".format(self.__class__.__name__, time.time()))