pyqqq 0.12.200__tar.gz → 0.12.202__tar.gz

{pyqqq-0.12.200 → pyqqq-0.12.202}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyqqq
-Version: 0.12.200
+Version: 0.12.202
 Summary: Package for quantitative strategy development on the PyQQQ platform
 License: MIT
 Author: PyQQQ team

{pyqqq-0.12.200 → pyqqq-0.12.202}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyqqq"
-version = "0.12.200"
+version = "0.12.202"
 description = "Package for quantitative strategy development on the PyQQQ platform"
 authors = ["PyQQQ team <pyqqq.cs@gmail.com>"]
 readme = "README.md"

{pyqqq-0.12.200 → pyqqq-0.12.202}/pyqqq/brokerage/ebest/tr_client.py

@@ -1,16 +1,18 @@
-from enum import Enum
-from pyqqq.brokerage.ebest.oauth import EBestAuth
-from pyqqq.utils.logger import get_logger
-from pyqqq.utils.market_schedule import *
-from pyqqq.utils.retry import retry
-from typing import AsyncGenerator, Callable
 import asyncio
 import json
-import ssl
 import random
+import ssl
+from enum import Enum
+from typing import AsyncGenerator, Callable
+
 import requests
 import websockets
 
+from pyqqq.brokerage.ebest.oauth import EBestAuth
+from pyqqq.utils.logger import get_logger
+from pyqqq.utils.market_schedule import *
+from pyqqq.utils.retry import retry
+
 
 class EBestTRClient:
     logger = get_logger(__name__ + ".EBestTRClient")
@@ -26,18 +28,10 @@ class EBestTRClient:
     def __init__(self, auth: EBestAuth, corp_data: dict = None):
         self.auth = auth
         self.corp_data = corp_data
+        self.session = requests.Session()
 
-    @retry(requests.HTTPError)
-    def request(
-        self,
-        path: str,
-        tr_cd: str,
-        tr_cont: str = "N",
-        tr_cont_key: str = "",
-        params: dict = None,
-        body: dict = None,
-        method: str = "POST",
-    ):
+    @retry(requests.RequestException, delay=1)
+    def request(self, path: str, tr_cd: str, tr_cont: str = "N", tr_cont_key: str = "", params: dict = None, body: dict = None, method: str = "POST"):
         """
         TR 요청을 보내고 응답을 받는 메서드
 
@@ -67,13 +61,14 @@ class EBestTRClient:
 
         url = f"{self.auth.host_url}{path}"
 
-        r = requests.request(method=method, url=url, headers=headers, params=params, json=body)
+        r = self.session.request(method=method, url=url, headers=headers, params=params, json=body)
 
         if r.status_code != 200:
             try:
                 res_body = r.json()
                 if "rsp_cd" in res_body and res_body["rsp_cd"] == "IGW00121":
                     self.auth.get_token(refresh=True)
+
             except Exception as e:
                 self.logger.exception(e)