aboutsummaryrefslogtreecommitdiff
path: root/server.py
blob: 4159736aa99353a73bb47782d9546f0b528da7e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import os
import subprocess
import tornado.web
import tornado.ioloop
import tornado.options
import tornado.process
import tornado.template
import tornado.httpserver
from mfiapi import MFiAPI
from tornado.options import define, options

define("port", default=8888, help="run on the given port", type=int)


class PowerStatusHandler(tornado.web.RequestHandler):

    def __init__(self, *args, **kwargs):
        self.api = MFiAPI('172.16.0.38:6443')
        super(PowerStatusHandler, self).__init__(*args, **kwargs)

    def get(self):
        self.api.login('admin', 'password')
        self.finish(self.api.port_status())

    def post(self):
        self.api.login('admin', 'password')

        for key, value in self.request.arguments.items():
            self.api.toggle_port(key)


class IndexHandler(tornado.web.RequestHandler):

    def __init__(self, *args, **kwargs):
        self.api = MFiAPI('172.16.0.38:6443')
        super(IndexHandler, self).__init__(*args, **kwargs)

    def get(self):
        self.api.login('admin', 'password')

        ports = [(port, port.replace(" ", "_").replace("'", "-")) for port in self.api.port_status().keys()]

        self.render("index.html", ports=ports)


class Application(tornado.web.Application):

    def __init__(self):
        handlers = [
            (r"/power-status/?", PowerStatusHandler),
            (r"/?", IndexHandler),
        ]
        settings = dict(
            template_path=os.path.join(os.path.dirname(__file__), "templates"),
            static_path=os.path.join(os.path.dirname(__file__), "static"),
            debug=True,
            autoescape=None,
        )
        tornado.web.Application.__init__(self, handlers, **settings)


if __name__ == "__main__":
    tornado.options.parse_command_line()
    http_server = tornado.httpserver.HTTPServer(Application())
    http_server.listen(options.port)
    tornado.ioloop.IOLoop.instance().start()