v2-ui.py

# -*- coding:utf-8 -*-
import asyncio
import logging
import os
import platform
import sys

import tornado
import tornado.log
import tornado.options
from tornado import web, wsgi
from tornado.httpserver import HTTPServer
from tornado.ioloop import IOLoop

from init import app, BASE_DIR
from util import config


def init_windows():
    if platform.system() == 'Windows':
        asyncio.set_event_loop(asyncio.SelectorEventLoop())


def logging_init():
    logging.basicConfig(filename='/etc/v2-ui/v2-ui.log',
                        datefmt='%Y-%m-%d %H:%M:%S',
                        format='%(asctime)s-%(name)s-%(levelname)s-%(message)s',
                        level=logging.WARN)
    tornado.log.access_log.setLevel('ERROR')
    tornado.log.app_log.setLevel('ERROR')
    tornado.log.gen_log.setLevel('ERROR')


def get_ssl_option():
    cert_file = config.get_cert_file()
    key_file = config.get_key_file()
    if cert_file != '' and key_file != '':
        return {
            'certfile': cert_file,
            'keyfile': key_file,
        }
    return None


def main():
    base_path = config.get_base_path()
    settings = {
        'static_path': os.path.join(BASE_DIR, 'static'),
        'static_url_prefix': base_path + '/static/',
    }
    wsgi_app = wsgi.WSGIContainer(app)
    handlers = []
    if base_path:
        handlers += [(base_path, web.RedirectHandler, dict(url=base_path + '/'))]
    handlers += [(base_path + r'/.*', web.FallbackHandler, dict(fallback=wsgi_app))]
    tornado_app = web.Application(handlers, **settings)
    http_server = HTTPServer(tornado_app, ssl_options=get_ssl_option())
    http_server.listen(config.get_port(), config.get_address())
    print("Start success on port %d" % config.get_port())
    IOLoop.current().start()


if __name__ == '__main__':
    if len(sys.argv) > 1:
        if sys.argv[1] == 'resetconfig':
            config.reset_config()
            print('All panel settings have been reset to default values, now please restart the panel')
        elif sys.argv[1] == 'resetuser':
            from base.models import User
            from init import db
            User.query.update({'username': 'admin', 'password': 'admin'})
            db.session.commit()
            print('The username and password have been reset to admin, please restart the panel now')
        elif sys.argv[1] == 'setport':
            if len(sys.argv) > 2:
                port = sys.argv[2]
            else:
                port = 65432
            config.update_setting_by_key('port', port)
            print('Set port to ' + port + ' successfully')
        else:
            print('Invalid command')
            print('resetconfig: Reset all panel settings to default values')
            print('resetuser: Reset username and password to \'admin\'')
            print('setport [number]: Set web port to [number], default is 65432')
        os._exit(0)
    else:
        init_windows()
        logging_init()
        try:
            main()
        except BaseException as e:
            logging.error(str(e))
            raise e