# TODO: web frontend
import os
import tornado.web
import tornado.template
from .ranges import yamiIchiCodes

viewdir = os.path.dirname(os.path.realpath(__file__)) + '/../views'
loader = tornado.template.Loader(viewdir)

class FrontendHandler(tornado.web.RequestHandler):

    def initialize(self, conn):
        """
        conn: sqlite3 connection
        """
        self.conn = conn

    def get(self, page):
        print("getting", page)

        if page in yamiIchiCodes:
            range = yamiIchiCodes[page]
            ranges = {
                "24h": ("2018-11-04 10:00:00","2018-11-05 12:00:00"),
                "3d": ("2018-11-04 10:00:00","2018-11-07 12:00:00"),
                "1w": ("2018-11-04 10:00:00","2018-11-11 12:00:00")
            }
            startDate = ranges[range][0]
            endDate = ranges[range][1]
            html = loader.load("redeem.html").generate(code=page, startDate=startDate, endDate=endDate)
            self.write(html)
        elif page == "yamiichi":
            # c = self.conn.cursor()
            # c.execute("SELECT * FROM beats ORDER BY createdAt DESC LIMIT 20")
            # beats = [row for row in c]
            # print(beats)
            # html = loader.load("sockets.html").generate(beats=beats)
            html = loader.load("plot.html").generate()
            self.write(html)
        else:
            raise tornado.web.HTTPError(404)