🎈 Perf(optimized code): 优化代码

Author: RanY-Luck

backend/plugin/api_testing/api/v1/request.py

@@ -3,7 +3,6 @@
 """
 接口请求API
 """
-from typing import Any, Dict
 from fastapi import APIRouter
 from backend.plugin.api_testing.schema.request import ApiRequestSchema, ApiResponseSchema
 from backend.plugin.api_testing.utils.http_client import send_request
@@ -27,7 +26,7 @@ async def send_api_request(request_data: ApiRequestSchema) -> ResponseModel | Re
             try:
                 with open(file_path, "rb") as file:
                     files[field_name] = (file_path.split("/")[-1], file.read())
-            except Exception as error:
+            except Exception:
                 return response_base.fail()
 
     # 处理认证信息
@@ -64,6 +63,5 @@ async def send_api_request(request_data: ApiRequestSchema) -> ResponseModel | Re
         )
 
         return response_base.success(data=response_data.model_dump())
-    except Exception as error:
-        # self._handle_exception(error)
+    except Exception:
         return response_base.fail()

backend/plugin/api_testing/schema/request.py

@@ -3,10 +3,8 @@
 """
 请求相关模型
 """
-from typing import Any, Dict, List, Optional, Union
-
+from typing import Any, Dict, List, Optional
 from pydantic import BaseModel, Field
-
 from backend.plugin.api_testing.utils.http_client import RequestOptions
 
 
@@ -100,4 +98,4 @@ class TestStepResponse(BaseModel):
     order: int
     status: int
     create_time: str
-    update_time: str
+    update_time: str

backend/plugin/api_testing/utils/http_client.py

@@ -4,12 +4,9 @@
 接口请求客户端
 基于httpx实现HTTP请求功能
 """
-import json
 import time
-from typing import Any, Dict, List, Optional, Union
-
 import httpx
-from fastapi import HTTPException
+from typing import Any, Dict, Optional
 from pydantic import BaseModel
 
 
@@ -39,27 +36,27 @@ class RequestResult(BaseModel):
 
 class HttpClient:
     """HTTP客户端"""
-    
+
     def __init__(self, options: RequestOptions = None):
         """
         初始化HTTP客户端
         
         :param options: 请求选项
         """
         self.options = options or RequestOptions()
-        
+
     async def request(
-        self,
-        method: str,
-        url: str,
-        params: Dict[str, Any] = None,
-        headers: Dict[str, str] = None,
-        cookies: Dict[str, str] = None,
-        data: Dict[str, Any] = None,
-        json_data: Dict[str, Any] = None,
-        files: Dict[str, Any] = None,
-        auth: tuple = None,
-        options: RequestOptions = None
+            self,
+            method: str,
+            url: str,
+            params: Dict[str, Any] = None,
+            headers: Dict[str, str] = None,
+            cookies: Dict[str, str] = None,
+            data: Dict[str, Any] = None,
+            json_data: Dict[str, Any] = None,
+            files: Dict[str, Any] = None,
+            auth: tuple = None,
+            options: RequestOptions = None
     ) -> RequestResult:
         """
         发送HTTP请求
@@ -77,15 +74,15 @@ async def request(
         :return: 请求结果
         """
         opts = options or self.options
-        
+
         # 准备客户端参数
         client_kwargs = {
             "timeout": opts.timeout,
             "verify": opts.verify_ssl,
             "follow_redirects": opts.follow_redirects,
             "max_redirects": opts.max_redirects
         }
-        
+
         # 准备请求参数
         request_kwargs = {
             "method": method,
@@ -98,15 +95,15 @@ async def request(
             "files": files,
             "auth": auth
         }
-        
+
         # 过滤None值
         request_kwargs = {k: v for k, v in request_kwargs.items() if v is not None}
-        
+
         start_time = time.time()
         error = None
         response = None
         retry_count = 0
-        
+
         # 支持请求重试
         while retry_count <= opts.retry_count:
             try:
@@ -121,9 +118,9 @@ async def request(
                     time.sleep(opts.retry_interval)
                 else:
                     break
-        
+
         elapsed_time = (time.time() - start_time) * 1000  # 转换为毫秒
-        
+
         if response is None:
             # 请求失败，返回错误信息
             return RequestResult(
@@ -137,15 +134,15 @@ async def request(
                 text="",
                 error=error
             )
-        
+
         # 尝试解析JSON
         json_result = None
         try:
             if response.headers.get('content-type', '').startswith('application/json'):
                 json_result = response.json()
         except Exception:
             pass
-            
+
         # 构建结果
         result = RequestResult(
             url=str(response.url),
@@ -158,7 +155,7 @@ async def request(
             text=response.text,
             json_data=json_result
         )
-        
+
         return result
 
 
@@ -167,16 +164,16 @@ async def request(
 
 
 async def send_request(
-    method: str,
-    url: str,
-    params: Dict[str, Any] = None,
-    headers: Dict[str, str] = None,
-    cookies: Dict[str, str] = None,
-    data: Dict[str, Any] = None,
-    json_data: Dict[str, Any] = None,
-    files: Dict[str, Any] = None,
-    auth: tuple = None,
-    options: RequestOptions = None
+        method: str,
+        url: str,
+        params: Dict[str, Any] = None,
+        headers: Dict[str, str] = None,
+        cookies: Dict[str, str] = None,
+        data: Dict[str, Any] = None,
+        json_data: Dict[str, Any] = None,
+        files: Dict[str, Any] = None,
+        auth: tuple = None,
+        options: RequestOptions = None
 ) -> RequestResult:
     """
     使用默认客户端发送请求的辅助函数
@@ -204,4 +201,4 @@ async def send_request(
         files=files,
         auth=auth,
         options=options
-    )
+    )