Open Media Library

oml/server.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+# vi:si:et:sw=4:sts=4:ts=4
+
+import os
+import sys
+from tornado.web import StaticFileHandler, Application, FallbackHandler
+from tornado.wsgi import WSGIContainer
+from tornado.httpserver import HTTPServer
+from tornado.ioloop import IOLoop
+
+from app import app
+import settings
+import websocket
+
+import state
+import node.server
+
+def run():
+    root_dir = os.path.normpath(os.path.join(os.path.abspath(os.path.dirname(__file__)), '..'))
+    PID = sys.argv[2] if len(sys.argv) > 2 else None
+
+    state.main = IOLoop.instance()
+
+    static_path = os.path.join(root_dir, 'static')
+
+    options = {
+        'debug': not PID 
+    }
+
+    tr = WSGIContainer(app)
+
+    handlers = [
+        (r'/(favicon.ico)', StaticFileHandler, {'path': static_path}),
+        (r'/static/(.*)', StaticFileHandler, {'path': static_path}),
+        (r'/ws', websocket.Handler),
+        (r".*", FallbackHandler, dict(fallback=tr)),
+    ]
+
+    http_server = HTTPServer(Application(handlers, **options))
+
+    http_server.listen(settings.server['port'], settings.server['address'])
+    if PID:
+        with open(PID, 'w') as pid:
+            pid.write('%s' % os.getpid())
+
+    def start_node():
+        import user
+        import downloads
+        import nodes
+        state.node = node.server.start(app)
+        state.nodes = nodes.Nodes(app)
+        state.downloads = downloads.Downloads(app)
+        def add_users(app):
+            with app.app_context():
+                for p in user.models.User.query.filter_by(peered=True):
+                    state.nodes.queue('add', p.id)
+        state.main.add_callback(add_users, app)
+    state.main.add_callback(start_node)
+    state.main.start()