api.py

import asyncio
import json
import os
import sys
import time
import tornado.gen
import tornado.ioloop
import tornado.platform
import tornado.web

from tools.iremitter import IREmitter

SIGNAL_ATTEMPTS = 10
OUTLET_NAME = "outlet_a";
RESET_DELAY_SECS = 10

is_debug = "--debug" in sys.argv

class APIHandlerBase(tornado.web.RequestHandler):
  def prepare(self):
    self.emitter = IREmitter(SIGNAL_ATTEMPTS, is_debug)

class OffHandler(APIHandlerBase):
  async def post(self):
    self.write(json.dumps({"status": "success"}))
    self.finish()
    self.emitter.emit_off(OUTLET_NAME)

class OnHandler(APIHandlerBase):
  async def post(self):
    self.write(json.dumps({"status": "success"}))
    self.finish()
    self.emitter.emit_on(OUTLET_NAME)

class ResetHandler(APIHandlerBase):
  async def post(self):
    self.write(json.dumps({"status": "success"}))
    self.finish()
    self.emitter.emit_off(OUTLET_NAME)
    await asyncio.sleep(RESET_DELAY_SECS)
    self.emitter.emit_on(OUTLET_NAME)

class PingHandler(tornado.web.RequestHandler):
  async def get(self):
    self.write(json.dumps({"status": "success"}))
    self.finish()

class MainHandler(tornado.web.RequestHandler):
  def get(self):
    self.redirect("/static/index.html")

def make_app():
  return tornado.web.Application(
    [
      (r"/", MainHandler),
      (r"/off", OffHandler),
      (r"/on", OnHandler),
      (r"/reset", ResetHandler),
      (r"/ping", PingHandler),
      (
        r"/static/(.*)",
        tornado.web.StaticFileHandler,
        {"path": os.path.join(os.path.dirname(__file__), "frontend/static")}
      ),
      (
        r"/out/(.*)",
        tornado.web.StaticFileHandler,
        {"path": os.path.join(os.path.dirname(__file__), "frontend/out")}
      ),
    ],
    debug = is_debug
  )

if __name__ == "__main__":
  tornado.platform.asyncio.AsyncIOMainLoop().install()
  app = make_app()
  app.listen(8000)
  asyncio.get_event_loop().run_forever()