mirror of
https://github.com/MeoProject/lx-music-api-server.git
synced 2025-07-07 06:52:11 +08:00
feat: 更换flask框架为aiohttp
This commit is contained in:
helloplhm-qwq
145
main.py
145
main.py
@ -8,88 +8,103 @@
|
||||
# - license: MIT -
|
||||
# ----------------------------------------
|
||||
# This file is part of the "lx-music-api-server" project.
|
||||
# Do not edit except you know what you are doing.
|
||||
|
||||
# flask
|
||||
from flask import Flask, request
|
||||
|
||||
# create flask app
|
||||
app = Flask("LXMusicTestAPI")
|
||||
|
||||
# redirect the default flask logging to custom
|
||||
import logging
|
||||
from aiohttp import web
|
||||
from common import config
|
||||
from common import log
|
||||
|
||||
flask_logger = log.log('flask')
|
||||
logging.getLogger('werkzeug').addHandler(log.flaskLogHelper(flask_logger))
|
||||
logger = log.log("main")
|
||||
|
||||
from common import utils
|
||||
from common import lxsecurity
|
||||
from common import utils
|
||||
from common import log
|
||||
from common import Httpx
|
||||
from apis import SongURL
|
||||
from apis import handle_api_request
|
||||
import traceback
|
||||
import time
|
||||
|
||||
logger = log.log("main")
|
||||
aiologger = log.log('aiohttp_web')
|
||||
|
||||
Httpx.checkcn()
|
||||
|
||||
@app.route('/')
|
||||
def index():
|
||||
return utils.format_dict_json({"code": 0, "msg": "success", "data": None}), 200
|
||||
# check request info before start
|
||||
async def handle_before_request(app, handler):
|
||||
async def handle_request(request):
|
||||
# nginx proxy header
|
||||
if (request.headers.get("X-Real-IP")):
|
||||
request.remote = request.headers.get("X-Real-IP")
|
||||
# check ip
|
||||
if (config.check_ip_banned(request.remote)):
|
||||
return utils.handle_response({"code": 1, "msg": "您的IP已被封禁", "data": None}, 403)
|
||||
# check global rate limit
|
||||
if (
|
||||
(time.time() - config.getRequestTime('global'))
|
||||
<
|
||||
(config.read_config("security.rate_limit.global"))
|
||||
):
|
||||
return utils.handle_response({"code": 5, "msg": "全局限速", "data": None}, 429)
|
||||
if (
|
||||
(time.time() - config.getRequestTime(request.remote))
|
||||
<
|
||||
(config.read_config("security.rate_limit.ip"))
|
||||
):
|
||||
return utils.handle_response({"code": 5, "msg": "IP限速", "data": None}, 429)
|
||||
# update request time
|
||||
config.updateRequestTime('global')
|
||||
config.updateRequestTime(request.remote)
|
||||
# check host
|
||||
if (config.read_config("security.allowed_host.enable")):
|
||||
if request.remote_host.split(":")[0] not in config.read_config("security.allowed_host.list"):
|
||||
if config.read_config("security.allowed_host.blacklist.enable"):
|
||||
config.ban_ip(request.remote, int(config.read_config("security.allowed_host.blacklist.length")))
|
||||
return utils.handle_response({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}, 404)
|
||||
try:
|
||||
resp = await handler(request)
|
||||
aiologger.info(f'{request.remote} - {request.method} "{request.path}", {resp.status}')
|
||||
return resp
|
||||
except web.HTTPException as ex:
|
||||
if ex.status == 500: # 捕获500错误
|
||||
return utils.handle_response({"code": 4, "msg": "内部服务器错误", "data": None}, 500)
|
||||
else:
|
||||
logger.error(traceback.format_exc())
|
||||
return utils.handle_response({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}, 404)
|
||||
return handle_request
|
||||
|
||||
@app.route('/<method>/<source>/<songId>/<quality>')
|
||||
async def handle(method, source, songId, quality):
|
||||
async def main(request):
|
||||
return utils.handle_response({"code": 0, "msg": "success", "data": None})
|
||||
|
||||
|
||||
async def handle(request):
|
||||
method = request.match_info.get('method')
|
||||
source = request.match_info.get('source')
|
||||
songId = request.match_info.get('songId')
|
||||
quality = request.match_info.get('quality')
|
||||
if (config.read_config("security.key.enable") and request.host.split(':')[0] not in config.read_config('security.whitelist_host')):
|
||||
if (request.headers.get("X-Request-Key")) != config.read_config("security.key.value"):
|
||||
if (config.read_config("security.key.ban")):
|
||||
config.ban_ip(request.remote_addr)
|
||||
return utils.format_dict_json({"code": 1, "msg": "key验证失败", "data": None}), 403
|
||||
config.ban_ip(request.remote)
|
||||
return utils.handle_response({"code": 1, "msg": "key验证失败", "data": None}, 403)
|
||||
if (config.read_config('security.check_lxm.enable') and request.host.split(':')[0] not in config.read_config('security.whitelist_host')):
|
||||
lxm = request.headers.get('lxm')
|
||||
if (not lxsecurity.checklxmheader(lxm, request.url)):
|
||||
if (config.read_config('security.lxm_ban.enable')):
|
||||
config.ban_ip(request.remote_addr)
|
||||
return utils.format_dict_json({"code": 1, "msg": "lxm请求头验证失败", "data": None}), 403
|
||||
config.ban_ip(request.remote)
|
||||
return utils.handle_response({"code": 1, "msg": "lxm请求头验证失败", "data": None}, 403)
|
||||
|
||||
if method == 'url':
|
||||
try:
|
||||
return utils.format_dict_json(await SongURL(source, songId, quality))
|
||||
except Exception as e:
|
||||
logger.error(traceback.format_exc())
|
||||
return utils.format_dict_json({'code': 4, 'msg': '内部服务器错误', 'data': None}), 500
|
||||
else:
|
||||
return utils.format_dict_json({'code': 6, 'msg': '未知的请求类型: ' + method, 'data': None}), 400
|
||||
try:
|
||||
return utils.handle_response(await handle_api_request(method, source, songId, quality))
|
||||
except Exception as e:
|
||||
logger.error(traceback.format_exc())
|
||||
return utils.handle_response({'code': 4, 'msg': '内部服务器错误', 'data': None}, 500)
|
||||
|
||||
@app.errorhandler(500)
|
||||
def _500(_):
|
||||
return utils.format_dict_json({'code': 4, 'msg': '内部服务器错误', 'data': None}), 500
|
||||
async def handle_404(request):
|
||||
return utils.handle_response({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}, 404)
|
||||
|
||||
@app.errorhandler(404)
|
||||
def _404(_):
|
||||
return utils.format_dict_json({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}), 404
|
||||
app = web.Application(middlewares=[handle_before_request])
|
||||
# mainpage
|
||||
app.router.add_get('/', main)
|
||||
|
||||
@app.before_request
|
||||
def check():
|
||||
# nginx proxy header
|
||||
if (request.headers.get("X-Real-IP")):
|
||||
request.remote_addr = request.headers.get("X-Real-IP")
|
||||
# check ip
|
||||
if (config.check_ip_banned(request.remote_addr)):
|
||||
return utils.format_dict_json({"code": 1, "msg": "您的IP已被封禁", "data": None}), 403
|
||||
# check global rate limit
|
||||
if ((time.time() - config.getRequestTime('global')) <= (config.read_config("security.rate_limit.global"))):
|
||||
return utils.format_dict_json({"code": 5, "msg": "全局限速", "data": None}), 429
|
||||
if ((time.time() - config.getRequestTime(request.remote_addr)) <= (config.read_config("security.rate_limit.ip"))):
|
||||
return utils.format_dict_json({"code": 5, "msg": "IP限速", "data": None}), 429
|
||||
# update request time
|
||||
config.updateRequestTime('global')
|
||||
config.updateRequestTime(request.remote_addr)
|
||||
# check host
|
||||
if (config.read_config("security.allowed_host.enable")):
|
||||
if request.remote_host.split(":")[0] not in config.read_config("security.allowed_host.list"):
|
||||
if config.read_config("security.allowed_host.blacklist.enable"):
|
||||
config.ban_ip(request.remote_addr, int(config.read_config("security.allowed_host.blacklist.length")))
|
||||
return utils.format_dict_json({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}), 404
|
||||
|
||||
# run
|
||||
app.run(host=config.read_config('common.host'), port=config.read_config('common.port'))
|
||||
# api
|
||||
app.router.add_get('/{method}/{source}/{songId}/{quality}', handle)
|
||||
|
||||
# 404
|
||||
app.router.add_route('*', '/{tail:.*}', handle_404)
|
||||
|
||||
web.run_app(app, host=config.read_config('common.host'), port=config.read_config('common.port'))
|
||||
|
||||
Reference in New Issue
Block a user