pyqqq 0.12.192__tar.gz → 0.12.194__tar.gz

{pyqqq-0.12.192 → pyqqq-0.12.194}/PKG-INFO

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

{pyqqq-0.12.192 → pyqqq-0.12.194}/pyproject.toml

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

{pyqqq-0.12.192 → pyqqq-0.12.194}/pyqqq/brokerage/kis/tr_client.py

@@ -23,6 +23,7 @@ class KISTRClient:
     def __init__(self, auth: KISAuth, 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_id: str, tr_cont: str = "", params: dict = None, body: dict = None, method: str = "GET"):
@@ -57,7 +58,7 @@ class KISTRClient:
 
         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:

{pyqqq-0.12.192 → pyqqq-0.12.194}/pyqqq/brokerage/tracker.py

@@ -31,6 +31,7 @@ class TrackerSocket:
         self.stop_event = asyncio.Event()
         self.logger = get_logger(__name__ + ".TrackerSocket")
         self.trading_tracker_counter = 1
+        self.only_my_account = True
         self.event_callbacks: Dict[callable] = {}
 
     async def start(self):
@@ -64,6 +65,9 @@ class TrackerSocket:
         Args:
             event (OrderEvent): 주문 이벤트 객체
         """
+        if event is None or (self.only_my_account and not self._check_account_match(event.account_no)):
+            return
+
         self.logger.debug(f"Relay order event: {event}")
 
         for callback in self.event_callbacks.values():
@@ -120,6 +124,20 @@ class TrackerSocket:
 
         self.task = asyncio.create_task(self._monitor_schedule())
 
+    def _check_account_match(self, event_account_no):
+        my_account_no = f"{self.simple_api.account_no}{self.simple_api.account_product_code}"
+
+        if event_account_no is not None and my_account_no == event_account_no:
+            return True
+        else:
+            return False
+
+    def activate_all_account_event(self):
+        """
+        내 hts_id에 연동된 모든 계좌의 주문을 수신하도록 설정합니다.
+        """
+        self.only_my_account = False
+
 
 class TradingTracker:
     """