pyqqq 0.12.170__py3-none-any.whl → 0.12.172__py3-none-any.whl

pyqqq/backtest/broker.py

@@ -269,10 +269,12 @@ class TradingBroker(BaseBroker):
     def get_price(self, code: str, data_exchange: Optional[DataExchange] = None) -> Decimal:
         if data_exchange:
             price_data = self.data_api.get_price(code, data_exchange=data_exchange)
+            if data_exchange == DataExchange.NXT and price_data.get("open_price", 0) == 0:  # current_price는 어제의 값을 반환하므로 open_price가 0인지로 NXT 시장 여부 확인
+                price_data = self.data_api.get_price(code, data_exchange=DataExchange.KRX)
         else:
-            if self.market_nxt_on:
+            if self.market_nxt_on:  # TODO: 정규장에서도 NXT 가격을 우선 조회하는 것이니 수정 필요
                 price_data = self.data_api.get_price(code, data_exchange=DataExchange.NXT)
-                if price_data.get("current_price") == 0:
+                if price_data.get("open_price", 0) == 0:  # current_price는 어제의 값을 반환하므로 open_price가 0인지로 NXT 시장 여부 확인
                     price_data = self.data_api.get_price(code, data_exchange=DataExchange.KRX)
             else:
                 price_data = self.data_api.get_price(code, data_exchange=DataExchange.KRX)
@@ -285,10 +287,12 @@ class TradingBroker(BaseBroker):
     def get_minute_price(self, code: str, data_exchange: Optional[DataExchange] = None) -> pd.DataFrame:
         if data_exchange:
             result = self.data_api.get_today_minute_data(code, data_exchange=data_exchange)
+            if data_exchange == DataExchange.NXT and (result.empty or result.iloc[-1].close == 0):
+                result = self.data_api.get_today_minute_data(code, data_exchange=DataExchange.KRX)
         else:
             if self.market_nxt_on:
                 result = self.data_api.get_today_minute_data(code, data_exchange=DataExchange.NXT)
-                if result.iloc[-1].close == 0:
+                if result.empty or result.iloc[-1].close == 0:
                     result = self.data_api.get_today_minute_data(code, data_exchange=DataExchange.KRX)
             else:
                 result = self.data_api.get_today_minute_data(code, data_exchange=DataExchange.KRX)
@@ -297,6 +301,8 @@ class TradingBroker(BaseBroker):
     def get_daily_price(self, code: str, from_date: dtm.date, to_date: dtm.date, data_exchange: Optional[DataExchange] = None) -> pd.DataFrame:
         if data_exchange:
             result = self.data_api.get_historical_daily_data(code, from_date, to_date, adjusted_price=True, data_exchange=data_exchange)
+            if data_exchange == DataExchange.NXT and result.empty:
+                result = self.data_api.get_historical_daily_data(code, from_date, to_date, adjusted_price=True, data_exchange=DataExchange.KRX)
         else:
             if self.market_nxt_on:
                 result = self.data_api.get_historical_daily_data(code, from_date, to_date, adjusted_price=True, data_exchange=DataExchange.NXT)
@@ -309,6 +315,8 @@ class TradingBroker(BaseBroker):
     def get_orderbook(self, code: str, data_exchange: Optional[DataExchange] = None):
         if data_exchange:
             result = self.data_api.get_orderbook(code, data_exchange=data_exchange)
+            if data_exchange == DataExchange.NXT and not result:
+                result = self.data_api.get_orderbook(code, data_exchange=DataExchange.KRX)
         else:
             if self.market_nxt_on:
                 result = self.data_api.get_orderbook(code, data_exchange=DataExchange.NXT)
@@ -326,6 +334,9 @@ class TradingBroker(BaseBroker):
         return next((o for o in orders if o["order_no"] == order_no), None)
 
     def get_positions(self):
+        """
+        포지션 조회 (단, NXT 프리마켓, 애프터마켓에서 현재 가격은 NXT 가격으로 조회되지 않음)
+        """
         return self.trading_api.get_positions()
 
     def create_order(self, asset_code: str, side: OrderSide, quantity: int, order_type: OrderType = OrderType.MARKET, price: int | Decimal = 0, exchange: OrderExchange = OrderExchange.KRX) -> str:

pyqqq/brokerage/tracker.py

@@ -311,6 +311,10 @@ class TradingTracker:
     def _find_pending_order(self, order_no) -> StockOrder:
         return find(lambda x: x.order_no == order_no, self.pending_orders)
 
+    def _find_order_in_today_orders(self, order_no) -> StockOrder:
+        today_orders = self.simple_api.get_today_order_history(order_no=order_no, exchanges=[OrderExchange.KRX, OrderExchange.NXT, OrderExchange.SOR])
+        return find(lambda x: x.order_no == order_no, today_orders)
+
     def _find_position(self, asset_code) -> StockPosition:
         return find(lambda x: x.asset_code == asset_code, self.positions)
 
@@ -458,6 +462,11 @@ class TradingTracker:
             event.filled_time,
             partial,
         )
+
+        if order is None:
+            self.logger.warn(f"Order not found for order_no: {order_no}")
+            order = self._find_order_in_today_orders(order_no)
+            self.logger.warn(f"Order found in today's orders: {order}")
         self._notify_pending_order_update("partial" if partial else "completed", order)
         self._notify_position_update(position_event_type, position)
 

pyqqq/data/realtime.py

@@ -1,4 +1,5 @@
-from typing import List
+from typing import List, Union
+from pyqqq.datatypes import DataExchange
 from pyqqq.utils.api_client import raise_for_status, send_request
 from pyqqq.utils.logger import get_logger
 from pyqqq.utils.singleton import singleton
@@ -14,12 +15,13 @@ import datetime as dtm
 logger = get_logger(__name__)
 
 
-def get_all_last_trades(codes: List[str] = None):
+def get_all_last_trades(codes: List[str] = None, exchange: str = "KRX"):
     """
     모든 종목의 최근 체결 정보를 반환합니다.
 
     Args:
         codes (List[str], optional): 종목 코드 리스트. 모든 종목의 체결 정보를 반환하려면 None을 전달합니다.
+        exchange (str, optional): 거래소. 기본값은 "KRX"입니다. 지원하는 거래소는 "KRX", "UN" 입니다.
 
     Returns:
         list:
@@ -54,7 +56,7 @@ def get_all_last_trades(codes: List[str] = None):
 
     params = None
     if codes:
-        params = {"codes": ",".join(codes) if isinstance(codes, list) else codes}
+        params = {"codes": ",".join(codes) if isinstance(codes, list) else codes, "exchange": exchange}
 
     r = send_request("GET", f"{c.PYQQQ_API_URL}/domestic-stock/trades", params=params)
     raise_for_status(r)

{pyqqq-0.12.170.dist-info → pyqqq-0.12.172.dist-info}/METADATA

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

{pyqqq-0.12.170.dist-info → pyqqq-0.12.172.dist-info}/RECORD

@@ -5,7 +5,7 @@ pyqqq/ai/domestic.py,sha256=FiJNInRlhcnxG7Jxmz2hDvaLhS8_jn-JFpQMze8Ch9s,1888
 pyqqq/ai/market_schedule.py,sha256=8HiivwC-xI2EKr8lXS_g4mTj2LYpCQ2QfZsJmIq61O0,818
 pyqqq/ai/minute.py,sha256=C0sTVkBY4-Vuj8Q9VZ7d9kZYAv963FUX4k3vIvhetng,1754
 pyqqq/backtest/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pyqqq/backtest/broker.py,sha256=yliZlXGQe3YFFFdRM8teSqrNkxrQTCCTBDbZT-HPM3s,65351
+pyqqq/backtest/broker.py,sha256=ynb0hxGk89BGg0xsNGcK6vw2tYc7hchiR5H8gA4GcVQ,66556
 pyqqq/backtest/environment.py,sha256=bkdlwG9MynuNL1u7f0cQ1L271cnPqJ7gX89EYwl8oOg,9313
 pyqqq/backtest/logger.py,sha256=BmoEMjUU76z8rZtMCYCwbspD3AVaHJrdbbT1EAFgrAE,3294
 pyqqq/backtest/positionprovider.py,sha256=wrR7Bntg28Q5_vGQV6XNzxe-SYoO9_GLcV9gDVEDAN4,4164
@@ -27,7 +27,7 @@ pyqqq/brokerage/kis/simple.py,sha256=t9vCnyeiVSHra2oYe8Gb3qFzxe2wWwrnJVCPNm6N-_0
 pyqqq/brokerage/kis/simple_overseas.py,sha256=1DuQBuJosg0mJQV7Ey2N3UOY8F3uOhzPDay4ncothuc,50360
 pyqqq/brokerage/kis/tr_client.py,sha256=9fTok0d8FmfXw4YxZSdn6T8UTHIG2aN1yMSkiMJUB3c,5530
 pyqqq/brokerage/multiprocess_tracker.py,sha256=Xx0hSpRZYITBGWjxclOEtNZdHV5agX94s34q1A8EE-Y,7283
-pyqqq/brokerage/tracker.py,sha256=z_bZ3O-G0oaEE7OBJawd2GbQwvPh-lKOLdWXOV1m1mw,23981
+pyqqq/brokerage/tracker.py,sha256=p26APsjzjM_qO3zrsre3GBqUx4G4hBYcc5TLvzfoKg8,24504
 pyqqq/config.py,sha256=55Vqc_pGkdbrBdCV1aLgoH_n5IFxmMC59sbPHId3LoI,498
 pyqqq/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyqqq/data/daily.py,sha256=k4dSiXNnT_J3mo0j_6VDjkGezBieVpIaZT3W_M7EU1Y,8019
@@ -35,7 +35,7 @@ pyqqq/data/domestic.py,sha256=2FOYxDGw2W7DGwY61p3uGFb4IWqWUKiNdR3RlewjkCU,30352
 pyqqq/data/index.py,sha256=d5b-8a7IXu7yNJWt1tIe1Mj83NW0ZnQq8nsj2Sl3Gx8,6988
 pyqqq/data/minutes.py,sha256=b-bqThP39t7V5io5dzI953bi1oWrUH3uie8GsFS7HUY,14727
 pyqqq/data/overseas.py,sha256=yx7tCZHW8AvjIbtrP4dqIeC6wseRSzbg5ag3dm6H0LY,1234
-pyqqq/data/realtime.py,sha256=y48JQSr54LkeCD9oIQcBqmK0wV1r2Hf42Va6PCkp1WU,14889
+pyqqq/data/realtime.py,sha256=Kjb0fK2mtQgluS165quAdGcxtdxfiEmBLp2B0i-5f44,15105
 pyqqq/data/ticks.py,sha256=DXioiKBsGTzwXyvEH0lpm8t5g-1nHIOLKMXoSrE1Rko,4127
 pyqqq/data/us_stocks.py,sha256=jXR9dQEVigrwTLEpX1aX1_AQvOlBopW265gwx8Nq8OA,12959
 pyqqq/datatypes.py,sha256=KnanWzat6w5w0vNvKHsWt9VOBwf9gh0njdt36PJBgXc,8370
@@ -58,6 +58,6 @@ pyqqq/utils/mock_api.py,sha256=7EsaVQ9mOVZQAqtQW24isPnk9QTbJII7x3guhFyEMAE,10569
 pyqqq/utils/position_classifier.py,sha256=EaomByAWM2lVuYow5OFdJNrN64Fpukhj-lhFkjYpjeo,14908
 pyqqq/utils/retry.py,sha256=4mw9MQvgSBC8bTLvDauaCEI5N9tL8upHCk8rSfaVRG8,2066
 pyqqq/utils/singleton.py,sha256=m6NZ8fwVDpI6U-gUUihMPgVK_NkDh-Z1NSAtjisrpjY,810
-pyqqq-0.12.170.dist-info/METADATA,sha256=sSP-LU4CsxfOTURvSpFF8me_R0gxZab2kLcbPm1OpFs,1664
-pyqqq-0.12.170.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-pyqqq-0.12.170.dist-info/RECORD,,
+pyqqq-0.12.172.dist-info/METADATA,sha256=NdLxiJ-cz-aA6ycgkvgTRh0WCNVc7HgOFcAi5HG6jLA,1664
+pyqqq-0.12.172.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+pyqqq-0.12.172.dist-info/RECORD,,