60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
|
# -*- 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()
|