kaq-quant-common 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

kaq_quant_common/api/common/__init__.py

@@ -1 +1 @@
-
+

kaq_quant_common/api/common/api_interface.py

@@ -1,38 +1,38 @@
-from abc import ABC
-from functools import wraps
-from typing import Callable, Type
-
-from pydantic import BaseModel
-
-
-def api_method(request_model: Type[BaseModel], response_model: Type[BaseModel]):
-    """
-    api 方法注解
-    :param request_model: 请求模型
-    :param response_model: 响应模型
-    :return:
-    """
-
-    def decorator(func: Callable):
-        # 将注解信息绑定到原始函数
-        func._is_api_method = True
-        func._request_model = request_model
-        func._response_model = response_model
-
-        @wraps(func)
-        def wrapper(*args, **kwargs):
-            return func(*args, **kwargs)
-
-        # 同步注解信息到包装函数，便于通过inspect发现
-        wrapper._is_api_method = True
-        wrapper._request_model = request_model
-        wrapper._response_model = response_model
-
-        return wrapper
-
-    return decorator
-
-
-# 定义 api 接口，暂时没啥用
-class ApiInterface(ABC):
-    pass
+from abc import ABC
+from functools import wraps
+from typing import Callable, Type
+
+from pydantic import BaseModel
+
+
+def api_method(request_model: Type[BaseModel], response_model: Type[BaseModel]):
+    """
+    api 方法注解
+    :param request_model: 请求模型
+    :param response_model: 响应模型
+    :return:
+    """
+
+    def decorator(func: Callable):
+        # 将注解信息绑定到原始函数
+        func._is_api_method = True
+        func._request_model = request_model
+        func._response_model = response_model
+
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            return func(*args, **kwargs)
+
+        # 同步注解信息到包装函数，便于通过inspect发现
+        wrapper._is_api_method = True
+        wrapper._request_model = request_model
+        wrapper._response_model = response_model
+
+        return wrapper
+
+    return decorator
+
+
+# 定义 api 接口，暂时没啥用
+class ApiInterface(ABC):
+    pass

kaq_quant_common/api/rest/api_client_base.py

@@ -1,42 +1,42 @@
-from typing import Optional, Type, TypeVar
-
-import requests
-from kaq_quant_common.api.common.auth import get_auth_token
-from kaq_quant_common.utils import logger_utils
-from pydantic import BaseModel
-
-R = TypeVar("R", bound=BaseModel)
-
-
-class ApiClientBase:
-    """
-    api 客户端
-    """
-
-    def __init__(self, base_url: str, token: Optional[str] = None):
-        self._base_url = base_url.rstrip("/")
-        self._token = token if token is not None else get_auth_token()
-        self._logger = logger_utils.get_logger(self)
-
-    # 发送请求
-    def _make_request(self, method_name: str, request_data: BaseModel, response_model: Type[R]) -> R:
-        url = f"{self._base_url}/api/{method_name}"
-        headers = {}
-        if self._token:
-            headers["Authorization"] = f"Bearer {self._token}"
-        try:
-            # 发送post请求
-            response = requests.post(url, json=request_data.model_dump(), headers=headers or None)
-            # 检查响应状态码，如果不成功，则尝试解析错误信息并抛出异常
-            if not response.ok:
-                try:
-                    error_data = response.json()
-                    error_message = error_data.get("error", response.text)
-                except ValueError:
-                    error_message = response.text
-                raise requests.exceptions.HTTPError(f"HTTP error occurred: {response.status_code} - {error_message}", response=response)
-            # 返回请求结果
-            return response_model(**response.json())
-        except requests.exceptions.RequestException as e:
-            self._logger.error(f"An error occurred: {e}")
-            raise
+from typing import Optional, Type, TypeVar
+
+import requests
+from kaq_quant_common.api.common.auth import get_auth_token
+from kaq_quant_common.utils import logger_utils
+from pydantic import BaseModel
+
+R = TypeVar("R", bound=BaseModel)
+
+
+class ApiClientBase:
+    """
+    api 客户端
+    """
+
+    def __init__(self, base_url: str, token: Optional[str] = None):
+        self._base_url = base_url.rstrip("/")
+        self._token = token if token is not None else get_auth_token()
+        self._logger = logger_utils.get_logger(self)
+
+    # 发送请求
+    def _make_request(self, method_name: str, request_data: BaseModel, response_model: Type[R]) -> R:
+        url = f"{self._base_url}/api/{method_name}"
+        headers = {}
+        if self._token:
+            headers["Authorization"] = f"Bearer {self._token}"
+        try:
+            # 发送post请求
+            response = requests.post(url, json=request_data.model_dump(), headers=headers or None)
+            # 检查响应状态码，如果不成功，则尝试解析错误信息并抛出异常
+            if not response.ok:
+                try:
+                    error_data = response.json()
+                    error_message = error_data.get("error", response.text)
+                except ValueError:
+                    error_message = response.text
+                raise requests.exceptions.HTTPError(f"HTTP error occurred: {response.status_code} - {error_message}", response=response)
+            # 返回请求结果
+            return response_model(**response.json())
+        except requests.exceptions.RequestException as e:
+            self._logger.error(f"An error occurred: {e}")
+            raise