nas-tools/run.py

import os
import signal
import sys
import time
import warnings

from watchdog.events import FileSystemEventHandler
from watchdog.observers import Observer

warnings.filterwarnings('ignore')

# 运行环境判断
is_windows_exe = getattr(sys, 'frozen', False) and (os.name == "nt")
if is_windows_exe:
    # 托盘相关库
    import threading
    from windows.trayicon import TrayIcon, NullWriter

    # 初始化环境变量
    os.environ["NASTOOL_CONFIG"] = os.path.join(os.path.dirname(sys.executable),
                                                "config",
                                                "config.yaml").replace("\\", "/")
    os.environ["NASTOOL_LOG"] = os.path.join(os.path.dirname(sys.executable),
                                             "config",
                                             "logs").replace("\\", "/")
    try:
        config_dir = os.path.join(os.path.dirname(sys.executable),
                                  "config").replace("\\", "/")
        if not os.path.exists(config_dir):
            os.makedirs(config_dir)
    except Exception as err:
        print(str(err))

from config import Config
import log
from web.main import App
from app.utils import SystemUtils, ConfigLoadCache
from app.utils.commons import INSTANCES
from app.db import init_db, update_db, init_data
from app.helper import IndexerHelper, DisplayHelper, init_chrome
from app.brushtask import BrushTask
from app.rsschecker import RssChecker
from app.scheduler import run_scheduler, restart_scheduler
from app.sync import run_monitor, restart_monitor
from app.torrentremover import TorrentRemover
from app.plugins import PluginManager
from check_config import update_config, check_config
from version import APP_VERSION


def sigal_handler(num, stack):
    """
    信号处理
    """
    if SystemUtils.is_docker():
        log.warn('捕捉到退出信号：%s，开始退出...' % num)
        # 停止虚拟显示
        DisplayHelper().quit()
        # 退出主进程
        sys.exit()


def get_run_config(forcev4=False):
    """
    获取运行配置
    """
    _web_host = "::"
    _web_port = 3000
    _ssl_cert = None
    _ssl_key = None
    _debug = False

    app_conf = Config().get_config('app')
    if app_conf:
        if forcev4:
            _web_host = "0.0.0.0"
        elif app_conf.get("web_host"):
            _web_host = app_conf.get("web_host").replace('[', '').replace(']', '')
        _web_port = int(app_conf.get('web_port')) if str(app_conf.get('web_port', '')).isdigit() else 3000
        _ssl_cert = app_conf.get('ssl_cert')
        _ssl_key = app_conf.get('ssl_key')
        _ssl_key = app_conf.get('ssl_key')
        _debug = True if app_conf.get("debug") else False

    app_arg = dict(host=_web_host, port=_web_port, debug=_debug, threaded=True, use_reloader=False)
    if _ssl_cert:
        app_arg['ssl_context'] = (_ssl_cert, _ssl_key)
    return app_arg


# 退出事件
signal.signal(signal.SIGINT, sigal_handler)
signal.signal(signal.SIGTERM, sigal_handler)


def init_system():
    # 配置
    log.console('NAStool 当前版本号：%s' % APP_VERSION)
    # 数据库初始化
    init_db()
    # 数据库更新
    update_db()
    # 数据初始化
    init_data()
    # 升级配置文件
    update_config()
    # 检查配置文件
    check_config()


def start_service():
    log.console("开始启动服务...")
    # 加载索引器配置
    IndexerHelper()
    # 启动虚拟显示
    DisplayHelper()
    # 启动定时服务
    run_scheduler()
    # 启动监控服务
    run_monitor()
    # 启动刷流服务
    BrushTask()
    # 启动自定义订阅服务
    RssChecker()
    # 启动自动删种服务
    TorrentRemover()
    # 加载插件
    PluginManager()


def monitor_config():
    class _ConfigHandler(FileSystemEventHandler):
        """
        配置文件变化响应
        """

        def __init__(self):
            FileSystemEventHandler.__init__(self)

        def on_modified(self, event):
            if not event.is_directory \
                    and os.path.basename(event.src_path) == "config.yaml":
                # 10秒内只能加载一次
                if ConfigLoadCache.get(event.src_path):
                    return
                ConfigLoadCache.set(event.src_path, True)
                log.console("进程 %s 检测到配置文件已修改，正在重新加载..." % os.getpid())
                time.sleep(1)
                # 重新加载配置
                Config().init_config()
                # 重载singleton服务
                for instance in INSTANCES.values():
                    if hasattr(instance, "init_config"):
                        instance.init_config()
                # 重启定时服务
                restart_scheduler()
                # 重启监控服务
                restart_monitor()

    # 配置文件监听
    _observer = Observer(timeout=10)
    _observer.schedule(_ConfigHandler(), path=Config().get_config_path(), recursive=False)
    _observer.daemon = True
    _observer.start()


# 系统初始化
init_system()

# 启动服务
start_service()

# 监听配置文件变化
monitor_config()

# 本地运行
if __name__ == '__main__':
    # Windows启动托盘
    if is_windows_exe:
        homepage = Config().get_config('app').get('domain')
        if not homepage:
            homepage = "http://localhost:%s" % str(Config().get_config('app').get('web_port'))
        log_path = os.environ.get("NASTOOL_LOG")

        sys.stdout = NullWriter()
        sys.stderr = NullWriter()


        def traystart():
            TrayIcon(homepage, log_path)


        if len(os.popen("tasklist| findstr %s" % os.path.basename(sys.executable), 'r').read().splitlines()) <= 2:
            p1 = threading.Thread(target=traystart, daemon=True)
            p1.start()
    else:
        # 初始化浏览器驱动
        init_chrome()

    # gunicorn 启动
    App.run(**get_run_config(is_windows_exe))