utils/interface_limiter.py · 辉/Python_RESTfulAPI_Codegen

代码拉取完成,页面将自动刷新

扫描微信二维码支付

取消

支付完成

richgiteeai

Watch

不关注关注所有动态仅关注版本发行动态关注但不提醒动态

0 Star 0 Fork 17

辉/Python_RESTfulAPI_Codegen

forked from 华北电力大学-iDeal/Python_RESTfulAPI_Codegen

代码 Issues 0 Pull Requests 0 Wiki 统计流水线

服务

加入 Gitee

与超过 1400万开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)

免费加入

已有帐号? 立即登录

文件

release

分支 (4)

标签 (10)

管理

release

master

dev

dev_flasgger_backup

v2.3.0

v2.2.0

v2.1.2

v2.1.1

v2.1.0

v2.0.0

v2.0.1

v1.0.0

v0.1.1

v0.1.0

克隆/下载

HTTPS SSH SVN SVN+SSH 下载ZIP

提示

下载代码请复制以下命令到终端执行

为确保你提交的代码身份被 Gitee 正确识别,请执行以下命令完成配置

git config --global user.name userName 
git config --global user.email userEmail

初次使用 SSH 协议进行代码克隆、推送等操作时,需按下述提示完成 SSH 配置

1 生成 RSA 密钥

2 获取 RSA 公钥内容,并配置到 SSH公钥中

在 Gitee 上使用 SVN,请访问使用指南

使用 HTTPS 协议时,命令行会出现如下账号密码验证步骤。基于安全考虑,Gitee 建议配置并使用私人令牌替代登录密码进行克隆、推送等操作

Username for 'https://gitee.com': userName

Password for 'https://userName@gitee.com': # 私人令牌

分支 4

标签 10

Python_RESTfulAPI_Codegen

utils

interface_limiter.py

interface_limiter.py 1.77 KB

#!/user/bin/env python3
# -*- coding:utf-8 -*-

# file:interface_limiter.py
# author:Superclass
# datetime:2022年5月23日 12:32
# software: PyCharm
'''
 获取全局接口访问频率限制器
'''
import json
import os

from flask import jsonify
from flask_limiter import Limiter
from flask_limiter.util import get_remote_address
from utils.response_code import RET

class InterfaceLimiter(object):
 # 读取限制器有关配置
 default_limits = None
 error_message = None

@classmethod
 def get_settings(cls):
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 CONFIG_PATH = os.path.join(BASE_DIR, "config/limiter.json")
 with open(CONFIG_PATH, 'r', encoding='utf-8') as f:
 settings = json.load(f)
 cls.default_limits = settings['default_limits']
 cls.error_message = settings['error_message']
 cls.second_limit = settings['second_limit']
 cls.minute_limit = settings['minute_limit']
 cls.hour_limit = settings['hour_limit']
 cls.day_limit = settings['day_limit']

# 获取限制器,进行初始化
 @classmethod
 def get_limiter(cls, app):
 cls.get_settings()
 limiter = Limiter(
 app,
 key_func=get_remote_address,
 default_limits=cls.default_limits
 )
 app.register_error_handler(429, cls.limiter_error_handler)

return limiter

# 限制超出时的错误页面,注册路由方法
 @classmethod
 def limiter_error_handler(cls, e):
 data = {
 'code': RET.REQERR,
 'message': cls.error_message,
 'data': {'error': '访问频率超出限制:一分钟{}次'.format(cls.minute_limit)}
 }
 res = jsonify(data)
 res.code = 429
 return res

一键复制编辑原始数据按行查看历史

neuroamazing 提交于 2022年05月23日 13:47 +08:00 . 设置超出限制的返回数据类型为json格式,调整状态码

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61

#!/user/bin/env python3
# -*- coding:utf-8 -*-

# file:interface_limiter.py
# author:Superclass
# datetime:2022年5月23日 12:32
# software: PyCharm
'''
 获取全局接口访问频率限制器
'''
import json
import os

from flask import jsonify
from flask_limiter import Limiter
from flask_limiter.util import get_remote_address
from utils.response_code import RET


class InterfaceLimiter(object):
 # 读取限制器有关配置
 default_limits = None
 error_message = None

 @classmethod
 def get_settings(cls):
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 CONFIG_PATH = os.path.join(BASE_DIR, "config/limiter.json")
 with open(CONFIG_PATH, 'r', encoding='utf-8') as f:
 settings = json.load(f)
 cls.default_limits = settings['default_limits']
 cls.error_message = settings['error_message']
 cls.second_limit = settings['second_limit']
 cls.minute_limit = settings['minute_limit']
 cls.hour_limit = settings['hour_limit']
 cls.day_limit = settings['day_limit']

 # 获取限制器,进行初始化
 @classmethod
 def get_limiter(cls, app):
 cls.get_settings()
 limiter = Limiter(
 app,
 key_func=get_remote_address,
 default_limits=cls.default_limits
 )
 app.register_error_handler(429, cls.limiter_error_handler)

 return limiter

 # 限制超出时的错误页面,注册路由方法
 @classmethod
 def limiter_error_handler(cls, e):
 data = {
 'code': RET.REQERR,
 'message': cls.error_message,
 'data': {'error': '访问频率超出限制:一分钟{}次'.format(cls.minute_limit)}
 }
 res = jsonify(data)
 res.code = 429
 return res