Bennett/Arcaea-server
1
0
Fork 0
mirror of https://github.com/Lost-MSth/Arcaea-server.git synced 2026-02-09 01:07:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cache song

Browse Source
This commit is contained in:
A-random-githuber
2025-12-26 00:33:28 +07:00
committed by GitHub
parent 707f881bdb
commit 73f28f38d3
1 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
latest version/main.py
View File

@@ -1,5 +1,6 @@
# encoding: utf-8 # encoding: utf-8
import json
import os import os
from importlib import import_module from importlib import import_module
@@ -7,7 +8,14 @@ from core.config_manager import Config, ConfigManager
if os.path.exists('config.py') or os.path.exists('config'): if os.path.exists('config.py') or os.path.exists('config'):
# 导入用户自定义配置 # 导入用户自定义配置
ConfigManager.load(import_module('config').Config) ConfigManager.load(import_module("config").Config)
else:
# Allow importing the config from a custom path given through an environment variable
configPath = os.environ.get("ARCAEA_JSON_CONFIG_PATH")
if configPath and os.path.exists(configPath):
with open(configPath, 'r') as file:
ConfigManager.load_dict(json.load(file))
if Config.DEPLOY_MODE == 'gevent': if Config.DEPLOY_MODE == 'gevent':
# 异步 # 异步
@@ -17,7 +25,7 @@ if Config.DEPLOY_MODE == 'gevent':
import sys import sys
from logging.config import dictConfig from logging.config import dictConfig
from multiprocessing import Process, set_start_method from multiprocessing import Process, current_process, set_start_method
from traceback import format_exc from traceback import format_exc
from flask import Flask, make_response, request, send_from_directory from flask import Flask, make_response, request, send_from_directory
@@ -183,7 +191,7 @@ def pre_main():
'stream': 'ext://flask.logging.wsgi_errors_stream', 'stream': 'ext://flask.logging.wsgi_errors_stream',
'formatter': 'default' 'formatter': 'default'
}, },
"error_file": generate_log_file_dict('ERROR', './log/error.log') "error_file": generate_log_file_dict('ERROR', f'{Config.LOG_FOLDER_PATH}/error.log')
}, },
'formatters': { 'formatters': {
'default': { 'default': {
@@ -194,11 +202,11 @@ def pre_main():
if Config.ALLOW_INFO_LOG: if Config.ALLOW_INFO_LOG:
log_dict['root']['handlers'].append('info_file') log_dict['root']['handlers'].append('info_file')
log_dict['handlers']['info_file'] = generate_log_file_dict( log_dict['handlers']['info_file'] = generate_log_file_dict(
'INFO', './log/info.log') 'INFO', f'{Config.LOG_FOLDER_PATH}/info.log')
if Config.ALLOW_WARNING_LOG: if Config.ALLOW_WARNING_LOG:
log_dict['root']['handlers'].append('warning_file') log_dict['root']['handlers'].append('warning_file')
log_dict['handlers']['warning_file'] = generate_log_file_dict( log_dict['handlers']['warning_file'] = generate_log_file_dict(
'WARNING', './log/warning.log') 'WARNING', f'{Config.LOG_FOLDER_PATH}/warning.log')
dictConfig(log_dict) dictConfig(log_dict)
@@ -226,7 +234,9 @@ def main():
# must run for init # must run for init
pre_main() # this ensures avoiding duplicate init logs for some reason
if current_process().name == 'MainProcess':
pre_main()
if __name__ == '__main__': if __name__ == '__main__':
set_start_method("spawn") set_start_method("spawn")