From 01b4a4e4958cfcdedb372532da7f3939b2596e81 Mon Sep 17 00:00:00 2001 From: wzx <1241414545@qq.com> Date: Mon, 27 Nov 2023 17:00:09 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E8=AF=B7=E6=B1=82?= =?UTF-8?q?=E9=97=B4=E9=9A=94=E6=97=B6=E9=97=B4=E8=AE=A1=E7=AE=97=E9=94=99?= =?UTF-8?q?=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- common/config.py | 2 +- main.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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")) ):