oop

main.py

@@ -1,246 +0,0 @@
-# encoding: utf-8
-
-import json
-import os
-from importlib import import_module
-
-from core.config_manager import Config, ConfigManager
-
-if os.path.exists('config.py') or os.path.exists('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':
-    # 异步
-    from gevent import monkey
-    monkey.patch_all()
-
-
-import sys
-from logging.config import dictConfig
-from multiprocessing import Process, current_process, set_start_method
-from traceback import format_exc
-
-from flask import Flask, make_response, request, send_from_directory
-
-import api
-import server
-import web.index
-import web.login
-# import webapi
-from core.bundle import BundleDownload
-from core.constant import Constant
-from core.download import UserDownload
-from core.error import ArcError, NoAccess, RateLimit
-from core.init import FileChecker
-from core.sql import Connect
-from server.func import error_return
-
-app = Flask(__name__)
-
-if Config.USE_PROXY_FIX:
-    # 代理修复
-    from werkzeug.middleware.proxy_fix import ProxyFix
-    app.wsgi_app = ProxyFix(app.wsgi_app, x_proto=1, x_host=1)
-if Config.USE_CORS:
-    # 服务端跨域
-    from flask_cors import CORS
-    CORS(app, supports_credentials=True)
-
-
-os.chdir(sys.path[0])  # 更改工作路径，以便于愉快使用相对路径
-
-
-app.config.from_mapping(SECRET_KEY=Config.SECRET_KEY)
-app.config['SESSION_TYPE'] = 'filesystem'
-app.register_blueprint(web.login.bp)
-app.register_blueprint(web.index.bp)
-app.register_blueprint(api.bp)
-list(map(app.register_blueprint, server.get_bps()))
-# app.register_blueprint(webapi.bp)
-
-
-@app.route('/')
-def hello():
-    return "Hello World!"
-
-
-@app.route('/favicon.ico', methods=['GET'])  # 图标
-def favicon():
-    # Pixiv ID: 82374369
-    # 我觉得这张图虽然并不是那么精细，但很有感觉，色彩的强烈对比下给人带来一种惊艳
-    # 然后在压缩之下什么也看不清了:(
-
-    return app.send_static_file('favicon.ico')
-
-
-@app.route('/download/<path:file_path>', methods=['GET'])  # 下载
-def download(file_path):
-    with Connect(in_memory=True) as c:
-        try:
-            x = UserDownload(c)
-            x.token = request.args.get('t')
-            x.song_id, x.file_name = file_path.split('/', 1)
-            x.select_for_check()
-            if x.is_limited:
-                raise RateLimit(
-                    f'User `{x.user.user_id}` has reached the download limit.', 903)
-            if not x.is_valid:
-                raise NoAccess('Expired token.')
-            x.download_hit()
-            if Config.DOWNLOAD_USE_NGINX_X_ACCEL_REDIRECT:
-                # nginx X-Accel-Redirect
-                response = make_response()
-                response.headers['Content-Type'] = 'application/octet-stream'
-                response.headers['X-Accel-Redirect'] = Config.NGINX_X_ACCEL_REDIRECT_PREFIX + file_path
-                return response
-            return send_from_directory(Constant.SONG_FILE_FOLDER_PATH, file_path, as_attachment=True, conditional=True)
-        except ArcError as e:
-            if Config.ALLOW_WARNING_LOG:
-                app.logger.warning(format_exc())
-            return error_return(e)
-    return error_return()
-
-
-@app.route('/bundle_download/<string:token>', methods=['GET'])  # 热更新下载
-def bundle_download(token: str):
-    with Connect(in_memory=True) as c_m:
-        try:
-            file_path = BundleDownload(c_m).get_path_by_token(
-                token, request.remote_addr)
-            if Config.DOWNLOAD_USE_NGINX_X_ACCEL_REDIRECT:
-                # nginx X-Accel-Redirect
-                response = make_response()
-                response.headers['Content-Type'] = 'application/octet-stream'
-                response.headers['X-Accel-Redirect'] = Config.BUNDLE_NGINX_X_ACCEL_REDIRECT_PREFIX + file_path
-                return response
-            return send_from_directory(Constant.CONTENT_BUNDLE_FOLDER_PATH, file_path, as_attachment=True, conditional=True)
-        except ArcError as e:
-            if Config.ALLOW_WARNING_LOG:
-                app.logger.warning(format_exc())
-            return error_return(e)
-    return error_return()
-
-
-if Config.DEPLOY_MODE == 'waitress':
-    # 给waitress加个日志
-    @app.after_request
-    def after_request(response):
-        app.logger.info(
-            f'{request.remote_addr} - - {request.method} {request.path} {response.status_code}')
-        return response
-
-# @app.before_request
-# def before_request():
-#     print(request.path)
-#     print(request.headers)
-#     print(request.data)
-
-
-def tcp_server_run():
-    if Config.DEPLOY_MODE == 'gevent':
-        # 异步 gevent WSGI server
-        host_port = (Config.HOST, Config.PORT)
-        app.logger.info('Running gevent WSGI server... (%s:%s)' % host_port)
-        from gevent.pywsgi import WSGIServer
-        WSGIServer(host_port, app, log=app.logger).serve_forever()
-    elif Config.DEPLOY_MODE == 'waitress':
-        # waitress WSGI server
-        import logging
-        from waitress import serve  # type: ignore
-        logger = logging.getLogger('waitress')
-        logger.setLevel(logging.INFO)
-        serve(app, host=Config.HOST, port=Config.PORT)
-    else:
-        if Config.SSL_CERT and Config.SSL_KEY:
-            app.run(Config.HOST, Config.PORT, ssl_context=(
-                Config.SSL_CERT, Config.SSL_KEY))
-        else:
-            app.run(Config.HOST, Config.PORT)
-
-
-def generate_log_file_dict(level: str, filename: str) -> dict:
-    return {
-        "class": "logging.handlers.RotatingFileHandler",
-        "maxBytes": 1024 * 1024,
-        "backupCount": 1,
-        "encoding": "utf-8",
-        "level": level,
-        "formatter": "default",
-        "filename": filename
-    }
-
-
-def pre_main():
-    log_dict = {
-        'version': 1,
-        'root': {
-            'level': 'INFO',
-            'handlers': ['wsgi', 'error_file']
-        },
-        'handlers': {
-            'wsgi': {
-                'class': 'logging.StreamHandler',
-                'stream': 'ext://flask.logging.wsgi_errors_stream',
-                'formatter': 'default'
-            },
-            "error_file": generate_log_file_dict('ERROR', f'{Config.LOG_FOLDER_PATH}/error.log')
-        },
-        'formatters': {
-            'default': {
-                'format': '[%(asctime)s] %(levelname)s in %(module)s: %(message)s'
-            }
-        }
-    }
-    if Config.ALLOW_INFO_LOG:
-        log_dict['root']['handlers'].append('info_file')
-        log_dict['handlers']['info_file'] = generate_log_file_dict(
-            'INFO', f'{Config.LOG_FOLDER_PATH}/info.log')
-    if Config.ALLOW_WARNING_LOG:
-        log_dict['root']['handlers'].append('warning_file')
-        log_dict['handlers']['warning_file'] = generate_log_file_dict(
-            'WARNING', f'{Config.LOG_FOLDER_PATH}/warning.log')
-
-    dictConfig(log_dict)
-
-    Connect.logger = app.logger
-    if not FileChecker(app.logger).check_before_run():
-        app.logger.error('Some errors occurred. The server will not run.')
-        input('Press ENTER key to exit.')
-        sys.exit()
-
-
-def main():
-    if Config.LINKPLAY_HOST and Config.SET_LINKPLAY_SERVER_AS_SUB_PROCESS:
-        from linkplay_server import link_play
-        process = [Process(target=link_play, args=(
-            Config.LINKPLAY_HOST, int(Config.LINKPLAY_UDP_PORT), int(Config.LINKPLAY_TCP_PORT)))]
-        [p.start() for p in process]
-        app.logger.info(
-            f"Link Play UDP server is running on {Config.LINKPLAY_HOST}:{Config.LINKPLAY_UDP_PORT} ...")
-        app.logger.info(
-            f"Link Play TCP server is running on {Config.LINKPLAY_HOST}:{Config.LINKPLAY_TCP_PORT} ...")
-        tcp_server_run()
-        [p.join() for p in process]
-    else:
-        tcp_server_run()
-
-
-# must run for init
-# this ensures avoiding duplicate init logs for some reason
-if current_process().name == 'MainProcess':
-    pre_main()
-
-if __name__ == '__main__':
-    set_start_method("spawn")
-    main()
-
-
-# Made By Lost  2020.9.11