init

main.py

@@ -0,0 +1,79 @@
+#!/usr/bin/env python3
+
+# ----------------------------------------
+# - mode: python - 
+# - author: helloplhm-qwq - 
+# - name: main.py - 
+# - project: lx-music-api-server - 
+# - 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 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 apis import SongURL
+
+require = utils.require
+
+@app.route('/')
+def index():
+    return utils.format_dict_json({"code": 0, "msg": "success", "data": None}), 200
+
+@app.route('/<method>/<source>/<songId>/<quality>')
+async def handle(method, source, songId, 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
+    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
+    
+    if method == 'url':
+        return utils.format_dict_json(await SongURL(source, songId, quality))
+    else:
+        return utils.format_dict_json({'code': 6, 'msg': '未知的请求类型: ' + method, 'data': None}), 400
+
+@app.errorhandler(500)
+def _500(_):
+    return utils.format_dict_json({'code': 4, 'msg': '内部服务器错误', 'data': None}), 500
+
+@app.errorhandler(404)
+def _404(_):
+    return utils.format_dict_json({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}), 404
+
+@app.before_request
+def check():
+    if (request.headers.get("X-Real-IP")):
+        request.remote_addr = request.headers.get("X-Real-IP")
+    if (config.check_ip_banned(request.remote_addr)):
+        return utils.format_dict_json({"code": 1, "msg": "您的IP已被封禁", "data": None}), 403
+    config.updateRequestTime(request.remote_addr)
+    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
+    
+
+app.run(host=config.read_config('common.host'), port=config.read_config('common.port'))