MuseQ/server.py at master · xinyunh/MuseQ · GitHub

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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import logging
import tornado.escape
import tornado.ioloop
import tornado.options
import tornado.web
import tornado.websocket
import os.path
import core
import const
from MuseQ import MuseQ

from tornado.options import define, options

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

class Application(tornado.web.Application):
    def __init__(self):
        handlers = [
            (r"/", MainHandler),
            (r"/ws", SocketHandler),
        ]
        settings = dict(
            template_path=os.path.join(os.path.dirname(__file__), "templates"),
            static_path=os.path.join(os.path.dirname(__file__), "static"),
            debug=True,
            )
        tornado.web.Application.__init__(self, handlers, **settings)

class MainHandler(tornado.web.RequestHandler):
    def get(self):
        self.render("index.html")

class SocketHandler(tornado.websocket.WebSocketHandler):
    machine = MuseQ(const.PATH, const.DB_NAME)
    machine.start()

    def playlist_changed(self):
        playlist = SocketHandler.machine.get_playlist()
        msg = {}
        msg["command"] = "update"
        msg["arg"] = playlist
        try:
            self.write_message(msg)
        except:
            logging.error("Error sending message", exc_info=True)

    def playstatus_changed(self):
        status = SocketHandler.machine.get_playstatus()
        msg = {}
        msg["command"] = "toggle"
        msg["arg"] = status
        try:
            self.write_message(msg)
        except:
            logging.error("Error sending message", exc_info=True)

    def allow_draft76(self):
        return True

    def open(self):
        SocketHandler.machine.register_updates(self)
        self.playlist_changed()
        self.playstatus_changed()

    def on_close(self):
        SocketHandler.machine.deregister_updates(self)

    def on_message(self, message):
        logging.info("got message %r", message)
        parsed = tornado.escape.json_decode(message)
        if parsed["command"] == "addurl":
            SocketHandler.machine.play(parsed["url"])
        elif parsed["command"] == "next":
            SocketHandler.machine.next()
        elif parsed["command"] == "stop":
            SocketHandler.machine.stop()
        elif parsed["command"] == "volumeup":
            SocketHandler.machine.volumeup()
        elif parsed["command"] == "volumedown":
            SocketHandler.machine.volumedown()
        elif parsed["command"] == "toggle":
            SocketHandler.machine.toggle()


def main():
    tornado.options.parse_command_line()
    app = Application()
    app.listen(options.port)
    tornado.ioloop.IOLoop.instance().start()

if __name__ == "__main__":
    main()