import prometheus_client

from tornado import gen

from ..views import BaseHandler


class Metrics(BaseHandler):
    @gen.coroutine
    def get(self):
        self.write(prometheus_client.generate_latest())
        self.set_header("Content-Type", "text/plain")


class Healthcheck(BaseHandler):
    @gen.coroutine
    def get(self):
        self.write("OK")