lesscode-flask 0.2.7__tar.gz → 0.2.9__tar.gz

{lesscode_flask-0.2.7 → lesscode_flask-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lesscode-flask
-Version: 0.2.7
+Version: 0.2.9
 Summary: lesscode-flask 是基于flask的web开发脚手架项目，该项目初衷为简化开发过程，让研发人员更加关注业务。
 Home-page: https://lesscode-flask
 Author: Chao.yy

{lesscode_flask-0.2.7 → lesscode_flask-0.2.9}/lesscode_flask/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.2.7'
+__version__ = '0.2.9'
 
 import functools
 import logging

{lesscode_flask-0.2.7 → lesscode_flask-0.2.9}/lesscode_flask/setting/__init__.py

@@ -100,13 +100,13 @@ class BaseConfig:
     RATE_LIMIT_BURST = 0
 
     # 是否启用限流计数验证
-    COUNT_LIMIT_ENABLE: bool = True
+    COUNT_LIMIT_ENABLE: bool = False
     # 触发限流的处理函数
     COUNT_LIMIT_HANDLER = CountLimitHandler
     # 窗口期内最大访问量
-    COUNT_LIMIT_COUNT = 5
+    COUNT_LIMIT_COUNT = 500
     # 窗口期时长 单位秒
-    COUNT_LIMIT_TIME_WINDOW = 60
+    COUNT_LIMIT_TIME_WINDOW = 60*60*12
     # 白名单
     LIMIT_WHITE_LIST: list = ["/ike2b/user/WXOA/login", "/ike2b/detail/company/node/get_note",
                                      "/ike2b/user/attention/attention_company_status",
@@ -114,6 +114,9 @@ class BaseConfig:
                                      "/toolEngine/targetedInvestmentPromotion/get_reach_chain",
                                      "/ike2b/user/WXOA/bind", "/ike2b/select/park/park_list",
                                      "/ike2b/pool/company/area_distribution/park_company_count_by_special_tag"]
+    # 飞书 webhook URL 常量定义  用于向指定的飞书机器人发送消息的 webhook 地址
+    LIMIT_FS_WEBHOOK_URL = "https://open.feishu.cn/open-apis/bot/v2/hook/545140ef-8234-4167-9f0c-8ee72abae430"
+    #
     #
     # # 外网地址
     # OUTSIDE_SCREEN_IP: str = ""

lesscode_flask-0.2.9/lesscode_flask/utils/fs_util.py

@@ -0,0 +1,55 @@
+import logging
+
+import json
+
+import requests
+
+logger = logging.getLogger(__name__)
+
+def fs_webhook(webhook_url:str,title:str, content:[str,list]):
+    """
+    发送飞书 webhook 消息
+
+    参数:
+        webhook_url (str): 飞书 webhook 的 URL 地址
+        title (str): 消息的标题
+        content (str or list): 消息内容，可以是字符串或字符串列表
+
+    返回值:
+        无返回值，直接发送 HTTP 请求
+    """
+    headers = {'Content-Type': 'application/json;charset=utf-8'}
+
+    # 根据内容类型构建飞书消息格式
+    if isinstance(content, list):
+        fs_content = []
+        for i in content:
+            fs_content.append({
+                "tag": "text",
+                "text": f"{i}\n"
+            })
+    else:
+        fs_content = [{
+            "tag": "text",
+            "text": f"{content}\n"
+        }]
+
+    # 构造飞书消息的 JSON 结构
+    json_text = {
+        "msg_type": "post",
+        "content": {
+            "post": {
+                "zh_cn": {
+                    "title": title,
+                    "content": [
+                        fs_content
+                    ]
+                }
+            }
+        }
+    }
+
+    # 发送 POST 请求并记录响应结果
+    result = requests.post(webhook_url, json.dumps(json_text), headers=headers).content
+    logger.info(result)
+

lesscode_flask-0.2.9/lesscode_flask/utils/limit/req/rate_limiter_handler.py

@@ -0,0 +1,46 @@
+import time
+import logging
+
+from flask import current_app, request
+
+from lesscode_flask.model.response_result import ResponseResult
+from lesscode_flask.model.user import flask_login
+from lesscode_flask.utils.fs_util import fs_webhook
+from lesscode_flask.utils.redis.redis_helper import RedisHelper
+
+logger = logging.getLogger(__name__)
+
+
+class RateLimitHandler:
+    """
+    限流后的处理函数实现
+    """
+    def __init__(self, req, delay: float, excess: float):
+        """
+         初始化
+        :param req:
+        :param delay: 延迟时间
+        :param excess: 超出数量
+        """
+        self.req = req
+        self.delay = delay
+        self.excess = excess
+
+
+    def response_handler(self):
+        # 如果配置了飞书 webhook URL，则发送告警通知
+        limit_fs_webhook_url = current_app.config.get("LIMIT_FS_WEBHOOK_URL")
+        # 如果配置了飞书 webhook URL，则发送告警通知
+        if limit_fs_webhook_url:
+            content = []
+            current_user = flask_login.current_user
+            # 收集用户相关信息
+            content.append(f"用户名称：{current_user.display_name}")
+            phone_no = current_user.phone_no if current_user.phone_no is not None else "-"
+            content.append(f"手机号：{phone_no}")
+            content.append(f"用户IP：{request.remote_addr}")
+            content.append(f"资源地址：{request.url_rule.rule}")
+
+            # 发送飞书 webhook 告警
+            fs_webhook(limit_fs_webhook_url, "触发频率限流告警", content)
+        return ResponseResult.fail(status_code="403", http_code="403", message="请求过于频繁，请稍后再试！")

lesscode_flask-0.2.9/lesscode_flask/utils/limit/req_count/count_limiter_handler.py

@@ -0,0 +1,46 @@
+import logging
+
+from flask import current_app, request
+
+from lesscode_flask.model.response_result import ResponseResult
+from lesscode_flask.model.user import flask_login
+from lesscode_flask.utils.fs_util import fs_webhook
+
+logger = logging.getLogger(__name__)
+
+
+class CountLimitHandler:
+    """
+    限流后的处理函数实现
+    """
+
+    def __init__(self, req, remaining):
+        self.req = req
+        self.remaining = remaining
+
+
+    def response_handler(self):
+        """
+        处理请求频率超限的响应
+
+        该函数用于当用户请求过于频繁触发限流时，返回相应的错误响应。
+        同时会发送告警信息到飞书 webhook，记录相关用户和请求信息。
+
+        Returns:
+            ResponseResult: 包含403状态码和错误信息的响应结果
+        """
+        limit_fs_webhook_url = current_app.config.get("LIMIT_FS_WEBHOOK_URL")
+        # 如果配置了飞书 webhook URL，则发送告警通知
+        if limit_fs_webhook_url:
+            content = []
+            current_user = flask_login.current_user
+            # 收集用户相关信息
+            content.append(f"用户名称：{current_user.display_name}")
+            phone_no = current_user.phone_no if current_user.phone_no is not None else "-"
+            content.append(f"手机号：{phone_no}")
+            content.append(f"用户IP：{request.remote_addr}")
+            content.append(f"资源地址：{request.url_rule.rule}")
+
+            # 发送飞书 webhook 告警
+            fs_webhook(limit_fs_webhook_url, "触发总量限流告警", content)
+        return ResponseResult.fail(status_code="403", http_code="403", message="请求过于频繁，请稍后再试！")

{lesscode_flask-0.2.7 → lesscode_flask-0.2.9}/lesscode_flask.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lesscode-flask
-Version: 0.2.7
+Version: 0.2.9
 Summary: lesscode-flask 是基于flask的web开发脚手架项目，该项目初衷为简化开发过程，让研发人员更加关注业务。
 Home-page: https://lesscode-flask
 Author: Chao.yy

{lesscode_flask-0.2.7 → lesscode_flask-0.2.9}/lesscode_flask.egg-info/SOURCES.txt

@@ -27,8 +27,8 @@ lesscode_flask/service/resource_param_template_service.py
 lesscode_flask/setting/__init__.py
 lesscode_flask/setup/__init__.py
 lesscode_flask/static/swagger.py
-lesscode_flask/utils/__init__.py
 lesscode_flask/utils/dify_utils.py
+lesscode_flask/utils/fs_util.py
 lesscode_flask/utils/helpers.py
 lesscode_flask/utils/decorator/__init__.py
 lesscode_flask/utils/decorator/cache.py

lesscode_flask-0.2.7/lesscode_flask/utils/limit/req/rate_limiter_handler.py

@@ -1,28 +0,0 @@
-import time
-import logging
-
-from lesscode_flask.model.response_result import ResponseResult
-from lesscode_flask.utils.redis.redis_helper import RedisHelper
-
-logger = logging.getLogger(__name__)
-
-
-class RateLimitHandler:
-    """
-    限流后的处理函数实现
-    """
-    def __init__(self, req, delay: float, excess: float):
-        """
-         初始化
-        :param req:
-        :param delay: 延迟时间
-        :param excess: 超出数量
-        """
-        self.req = req
-        self.delay = delay
-        self.excess = excess
-
-
-    def response_handler(self):
-        message = "请求过于频繁，请稍后再试"
-        return ResponseResult.fail(status_code="403", http_code="403", message=message)

lesscode_flask-0.2.7/lesscode_flask/utils/limit/req_count/count_limiter_handler.py

@@ -1,20 +0,0 @@
-import logging
-
-from lesscode_flask.model.response_result import ResponseResult
-
-logger = logging.getLogger(__name__)
-
-
-class CountLimitHandler:
-    """
-    限流后的处理函数实现
-    """
-
-    def __init__(self, req, remaining):
-        self.req = req
-        self.remaining = remaining
-
-
-    def response_handler(self):
-        message = "请求过于频繁，请稍后再试！"
-        return ResponseResult.fail(status_code="403", http_code="403", message=message)