lesscode-flask 0.2.38__tar.gz → 0.2.39__tar.gz

{lesscode-flask-0.2.38 → lesscode-flask-0.2.39}/PKG-INFO

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

{lesscode-flask-0.2.38 → lesscode-flask-0.2.39}/lesscode_flask/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.2.38'
+__version__ = "0.2.39"
 
 import functools
 import logging
@@ -17,8 +17,16 @@ class SQ_Blueprint(Blueprint):
             kwargs["import_name"] = __name__
         super().__init__(name=name, url_prefix=url_prefix, **kwargs)
 
-    def decorator_handler(self, title: str, url: str = None, cache_enalbe: bool = False, cache_ex: int = 3600 * 10,
-                          content_type: str = "json", methods=['POST'], export_enable: bool = False):
+    def decorator_handler(
+            self,
+            title: str,
+            url: str = None,
+            cache_enalbe: bool = False,
+            cache_ex: int = 3600 * 10,
+            content_type: str = "json",
+            methods=["POST"],
+            export_enable: bool = False,
+    ):
         options = {"methods": methods}
 
         def decorator(func):
@@ -41,9 +49,7 @@ class SQ_Blueprint(Blueprint):
                     # 如果没有开启缓存，或者缓存未命中，则执行原始函数
                     data = func(*args, **kwargs)
                 if isinstance(data, dict) and export_enable:
-                    data.update({
-                        "download_key": download_key
-                    })
+                    data.update({"download_key": download_key})
                 return data
 
             wrapper._title = title
@@ -54,12 +60,31 @@ class SQ_Blueprint(Blueprint):
 
         return decorator
 
-    def post_route(self, title: str, url: str = None, cache_enalbe: bool = False, cache_ex: int = 3600 * 10,
-                   content_type: str = "json", methods=['POST'], export_enable: bool = False):
-        decorator = self.decorator_handler(title, url, cache_enalbe, cache_ex, content_type, methods, export_enable)
+    def post_route(
+            self,
+            title: str,
+            url: str = None,
+            cache_enalbe: bool = False,
+            cache_ex: int = 3600 * 10,
+            content_type: str = "json",
+            methods=["POST"],
+            export_enable: bool = False,
+    ):
+        decorator = self.decorator_handler(
+            title, url, cache_enalbe, cache_ex, content_type, methods, export_enable
+        )
         return decorator
 
-    def get_route(self, title: str, url: str = None, cache_enalbe: bool = False, cache_ex: int = 3600 * 10,
-                  content_type: str = "json", methods=['GET']):
-        decorator = self.decorator_handler(title, url, cache_enalbe, cache_ex, content_type, methods)
+    def get_route(
+            self,
+            title: str,
+            url: str = None,
+            cache_enalbe: bool = False,
+            cache_ex: int = 3600 * 10,
+            content_type: str = "json",
+            methods=["GET"],
+    ):
+        decorator = self.decorator_handler(
+            title, url, cache_enalbe, cache_ex, content_type, methods
+        )
         return decorator

lesscode-flask-0.2.39/lesscode_flask/utils/limit/req/rate_limiter_handler.py

@@ -0,0 +1,88 @@
+import time
+import logging
+
+import requests
+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，则发送告警通知
+        # 收集用户相关信息
+        current_user = flask_login.current_user
+        fs_oam_service_url = current_app.config.get("FS_OAM_SERVICE_URL")
+        token = request.headers.get("Authorization", "").replace("Bearer ", "")
+        lock_account_url = (
+            f"{fs_oam_service_url}/icp/authUser/lock_account?id={current_user.id}"
+        )
+        logout_url = f"{fs_oam_service_url}/icp/oauth/logout_token?token={token}"
+
+        # 发送 GET 请求到 lock_account_url
+        try:
+            lock_response = requests.get(lock_account_url, timeout=30)
+            logger.info(
+                "Lock account request sent, status: %s", lock_response.status_code
+            )
+        except requests.RequestException as e:
+            logger.error("Failed to send lock account request: %s", str(e))
+
+        # 发送 GET 请求到 logout_url
+        try:
+            logout_response = requests.get(logout_url, timeout=30)
+            logger.info("Logout request sent, status: %s", logout_response.status_code)
+        except requests.RequestException as e:
+            logger.error("Failed to send logout request: %s", str(e))
+
+        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(
+                {"tag": "text", "text": f"用户名称：{current_user.display_name}\n"}
+            )
+            phone_no = (
+                current_user.phone_no if current_user.phone_no is not None else "-"
+            )
+            content.append({"tag": "text", "text": f"手机号：{phone_no}\n"})
+            content.append({"tag": "text", "text": f"用户IP：{request.remote_addr}\n"})
+            content.append({"tag": "text", "text": f"资源地址：{request.path}\n"})
+            token = request.headers.get("Authorization", "").replace("Bearer ", "")
+            if token:
+                FS_OAM_SERVICE_URL = current_app.config.get("FS_OAM_SERVICE_URL")
+                content.append({"tag": "text", "text": "运维处理："})
+                # lock_account_url = f"{FS_OAM_SERVICE_URL}/icp/authUser/lock_account?id={current_user.id}"
+                # url = f"{FS_OAM_SERVICE_URL}/icp/oauth/logout_token?token={token}"
+                # content.append({"tag": "a","text": "强制下线","href": f"{url}"})
+                # content.append({"tag": "a", "text": "    禁止登录    ", "href": f"{lock_account_url}"})
+                ban_ip_url = f"{FS_OAM_SERVICE_URL}/icp/accessLog/ban_ip?ip={request.remote_addr}"
+                content.append({"tag": "a", "text": "封禁IP ", "href": f"{ban_ip_url}"})
+
+            # 发送飞书 webhook 告警
+            fs_webhook(limit_fs_webhook_url, "触发频率限流告警", content)
+        return ResponseResult.fail(
+            status_code="403", http_code="403", message="请求过于频繁，请稍后再试！"
+        )

lesscode-flask-0.2.39/lesscode_flask/utils/limit/req_count/count_limiter_handler.py

@@ -0,0 +1,84 @@
+import logging
+import requests
+
+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状态码和错误信息的响应结果
+        """
+        # 收集用户相关信息
+        current_user = flask_login.current_user
+        fs_oam_service_url = current_app.config.get("FS_OAM_SERVICE_URL")
+        token = request.headers.get("Authorization", "").replace("Bearer ", "")
+        lock_account_url = (
+            f"{fs_oam_service_url}/icp/authUser/lock_account?id={current_user.id}"
+        )
+        logout_url = f"{fs_oam_service_url}/icp/oauth/logout_token?token={token}"
+
+        # 发送 GET 请求到 lock_account_url
+        try:
+            lock_response = requests.get(lock_account_url, timeout=30)
+            logger.info(
+                "Lock account request sent, status: %s", lock_response.status_code
+            )
+        except requests.RequestException as e:
+            logger.error("Failed to send lock account request: %s", str(e))
+
+        # 发送 GET 请求到 logout_url
+        try:
+            logout_response = requests.get(logout_url, timeout=30)
+            logger.info("Logout request sent, status: %s", logout_response.status_code)
+        except requests.RequestException as e:
+            logger.error("Failed to send logout request: %s", str(e))
+        limit_fs_webhook_url = current_app.config.get("LIMIT_FS_WEBHOOK_URL")
+        # 如果配置了飞书 webhook URL，则发送告警通知
+        if limit_fs_webhook_url:
+            content = []
+            # 收集用户相关信息
+            content.append(
+                {"tag": "text", "text": f"用户名称：{current_user.display_name}\n"}
+            )
+            phone_no = (
+                current_user.phone_no if current_user.phone_no is not None else "-"
+            )
+            content.append({"tag": "text", "text": f"手机号：{phone_no}\n"})
+            content.append({"tag": "text", "text": f"用户IP：{request.remote_addr}\n"})
+            content.append({"tag": "text", "text": f"资源地址：{request.path}\n"})
+
+            if token:
+
+                content.append({"tag": "text", "text": "运维处理："})
+                # content.append({"tag": "a", "text": "强制下线", "href": f"{url}"})
+                # content.append({"tag": "a", "text": "    禁止登录    ",
+                # "href": f"{lock_account_url}"})
+                ban_ip_url = f"{fs_oam_service_url}/icp/accessLog/ban_ip?ip={request.remote_addr}"
+                content.append({"tag": "a", "text": "封禁IP ", "href": f"{ban_ip_url}"})
+            # 发送飞书 webhook 告警
+            fs_webhook(limit_fs_webhook_url, "触发总量限流告警", content)
+
+        return ResponseResult.fail(
+            status_code="403", http_code="403", message="请求过于频繁，请稍后再试！"
+        )

{lesscode-flask-0.2.38 → lesscode-flask-0.2.39}/lesscode_flask.egg-info/PKG-INFO

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

lesscode-flask-0.2.38/lesscode_flask/utils/limit/req/rate_limiter_handler.py

@@ -1,58 +0,0 @@
-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({"tag": "text","text": f"用户名称：{current_user.display_name}\n"})
-            phone_no = current_user.phone_no if current_user.phone_no is not None else "-"
-            content.append({"tag": "text","text": f"手机号：{phone_no}\n"})
-            content.append({"tag": "text","text": f"用户IP：{request.remote_addr}\n"})
-            content.append({"tag": "text","text": f"资源地址：{request.path}\n"})
-            token = request.headers.get("Authorization", "").replace("Bearer ", "")
-            if token:
-                FS_OAM_SERVICE_URL = current_app.config.get("FS_OAM_SERVICE_URL")
-                content.append({"tag": "text","text": f"运维处理：\n"})
-                lock_account_url = f"{FS_OAM_SERVICE_URL}/icp/authUser/lock_account?id={current_user.id}"
-                url = f"{FS_OAM_SERVICE_URL}/icp/oauth/logout_token?token={token}"
-                content.append({"tag": "a","text": "强制下线","href": f"{url}"})
-                content.append({"tag": "a", "text": "    禁止登录    ", "href": f"{lock_account_url}"})
-                ban_ip_url = f"{FS_OAM_SERVICE_URL}/icp/accessLog/ban_ip?ip={request.remote_addr}"
-                content.append({"tag": "a", "text": "封禁IP ", "href": f"{ban_ip_url}"})
-
-            # 发送飞书 webhook 告警
-            fs_webhook(limit_fs_webhook_url, "触发频率限流告警", content)
-        return ResponseResult.fail(status_code="403", http_code="403", message="请求过于频繁，请稍后再试！")
-
-

lesscode-flask-0.2.38/lesscode_flask/utils/limit/req_count/count_limiter_handler.py

@@ -1,57 +0,0 @@
-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
-            # 收集用户相关信息
-            current_user = flask_login.current_user
-            # 收集用户相关信息
-            content.append({"tag": "text","text": f"用户名称：{current_user.display_name}\n"})
-            phone_no = current_user.phone_no if current_user.phone_no is not None else "-"
-            content.append({"tag": "text","text": f"手机号：{phone_no}\n"})
-            content.append({"tag": "text","text": f"用户IP：{request.remote_addr}\n"})
-            content.append({"tag": "text","text": f"资源地址：{request.path}\n"})
-            token = request.headers.get("Authorization", "").replace("Bearer ", "")
-            if token:
-                FS_OAM_SERVICE_URL = current_app.config.get("FS_OAM_SERVICE_URL")
-                content.append({"tag": "text", "text": f"运维处理：\n"})
-                lock_account_url = f"{FS_OAM_SERVICE_URL}/icp/authUser/lock_account?id={current_user.id}"
-                url = f"{FS_OAM_SERVICE_URL}/icp/oauth/logout_token?token={token}"
-                content.append({"tag": "a", "text": "强制下线", "href": f"{url}"})
-                content.append({"tag": "a", "text": "    禁止登录    ", "href": f"{lock_account_url}"})
-                ban_ip_url = f"{FS_OAM_SERVICE_URL}/icp/accessLog/ban_ip?ip={request.remote_addr}"
-                content.append({"tag": "a", "text": "封禁IP ", "href": f"{ban_ip_url}"})
-            # 发送飞书 webhook 告警
-            fs_webhook(limit_fs_webhook_url, "触发总量限流告警", content)
-        return ResponseResult.fail(status_code="403", http_code="403", message="请求过于频繁，请稍后再试！")