diff --git a/common/config.py b/common/config.py index a6c3cb4..855fb1e 100644 --- a/common/config.py +++ b/common/config.py @@ -70,7 +70,7 @@ default = { "global": 0, "ip": 0, "desc": "请求速率限制,global为全局,ip为单个ip,填入的值为至少间隔多久才能进行一次请求,单位:秒,不限制请填为0" - } + }, "key": { "enable": False, "_enable-desc": "是否开启请求key,开启后只有请求头中包含key,且值一样时可以访问API", diff --git a/main.py b/main.py index 9ca018c..9914405 100644 --- a/main.py +++ b/main.py @@ -78,13 +78,13 @@ def check(): return utils.format_dict_json({"code": 1, "msg": "您的IP已被封禁", "data": None}), 403 # check global rate limit if ( - (config.getRequestTime('global') - time.time()) + (time.time() - config.getRequestTime('global')) < (config.read_config("security.rate_limit.global")) ): return utils.format_dict_json({"code": 5, "msg": "全局限速", "data": None}), 429 if ( - (config.getRequestTime(request.remote_addr) - time.time()) + (time.time() - config.getRequestTime(request.remote_addr)) < (config.read_config("security.rate_limit.ip")) ):