run.py

import logging
from sys import exit

import tornado.httpserver
import tornado.ioloop
import tornado.web
import tornado.wsgi
from tornado.options import options

# 引入Async应用
from async_app.view import async_app
from core.config import TAICHI_OS_LOGO, TAICHI_OS_WELCOME_MESSAGE
# 引入Flask应用
from core.main import create_app as make_core_app
from settings import get_server_settings, check_encoding_setting
# 引入WebSSH应用
from webssh.main import make_app as make_webssh_app


def add_handlers_to_app(app, pattern, handlers):
    """
    将处理程序添加到应用程序中。

    参数:
    app: Tornado web应用程序实例,我们将在其中添加处理程序。
    pattern: 一个字符串,表示URL模式,我们将在其中查找匹配的请求。
    handlers: 一个处理程序类或者函数,当URL匹配到模式时,将会被调用。

    返回:
    无返回值。此函数将处理程序直接添加到提供的应用程序实例中。
    """
    app.add_handlers(r'.*', [(pattern, handlers)])


def main():
    # 解析命令行参数
    try:
        options.parse_command_line()
    except tornado.options.Error:
        logging.error('命令行参数解析失败')
        exit(1)

    # 检查编码设置
    check_encoding_setting(options.encoding)

    # 获取当前的I/O循环
    loop = tornado.ioloop.IOLoop.current()

    # 创建主应用
    main_app = tornado.web.Application()

    # 添加WebSSH应用
    webssh_app = make_webssh_app(loop, options)
    add_handlers_to_app(main_app, r'/webssh/.*', webssh_app)

    # 添加WebSocket应用
    async_handlers = async_app()
    add_handlers_to_app(main_app, r'.*', async_handlers)

    # 添加Flask应用
    flask_app = make_core_app()
    wsgi_app = tornado.wsgi.WSGIContainer(flask_app)
    add_handlers_to_app(main_app, r'.*', wsgi_app)

    try:
        # 获取服务器设置
        server_settings = get_server_settings(options)
        # 启动服务器
        main_app.listen(options.port, options.address, **server_settings)
        logging.info(TAICHI_OS_LOGO)
        logging.info(TAICHI_OS_WELCOME_MESSAGE)
        logging.info(f"服务器启动成功! 运行在 {options.address}:{options.port}")
        loop.start()
    except KeyboardInterrupt:
        logging.info('服务器已停止')
    except OSError as e:
        logging.error('服务器启动失败, 请检查端口是否被占用, 或者地址是否正确, 然后使用root账户启动程序')
        logging.exception(e)
    except Exception as e:
        logging.error('服务器启动失败,请检查错误日志')
        logging.exception(e)


if __name__ == "__main__":
    main()