pyqqq 0.12.182__tar.gz → 0.12.184__tar.gz

{pyqqq-0.12.182 → pyqqq-0.12.184}/PKG-INFO

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

{pyqqq-0.12.182 → pyqqq-0.12.184}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyqqq"
-version = "0.12.182"
+version = "0.12.184"
 description = "Package for quantitative strategy development on the PyQQQ platform"
 authors = ["PyQQQ team <pyqqq.cs@gmail.com>"]
 readme = "README.md"
@@ -30,6 +30,8 @@ ipykernel = "^6.29.4"
 pydata-sphinx-theme = "^0.15.2"
 sphinx-copybutton = "^0.5.2"
 sphinx-togglebutton = "^0.3.2"
+yappi = "^1.6.10"
+snakeviz = "^2.2.2"
 
 [tool.black]
 line-length = 240

{pyqqq-0.12.182 → pyqqq-0.12.184}/pyqqq/brokerage/kis/simple.py

@@ -1,15 +1,18 @@
+import asyncio
+import datetime as dtm
 from decimal import Decimal
+from typing import AsyncGenerator, Dict, List, Optional
+
+import pandas as pd
+
 from pyqqq.brokerage.kis.domestic_stock import KISDomesticStock
 from pyqqq.brokerage.kis.oauth import KISAuth
 from pyqqq.data.realtime import get_all_last_trades
 from pyqqq.datatypes import *
+from pyqqq.utils.limiter import CallLimiter
 from pyqqq.utils.logger import get_logger
 from pyqqq.utils.market_schedule import get_market_schedule, is_full_day_closed
 from pyqqq.utils.mock_api import with_mock
-from typing import AsyncGenerator, Dict, List, Optional
-import asyncio
-import datetime as dtm
-import pandas as pd
 
 
 class KISStockPosition(StockPosition):
@@ -336,6 +339,8 @@ class KISSimpleDomesticStock:
             request_time = request_datetime.replace(hour=schedule.close_time.hour, minute=schedule.close_time.minute)
 
         while True:
+            CallLimiter().wait_limit_rate(20, scope="kis/inquire_time_itemchartprice")
+
             r = self.stock_api.inquire_time_itemchartprice(
                 asset_code,
                 request_time.time(),

{pyqqq-0.12.182 → pyqqq-0.12.184}/pyqqq/data/domestic.py

@@ -244,7 +244,9 @@ def get_tickers(
 
     KRX 거래소에서는 거래정지 종목 등이 포함되어 있으나, NXT 거래소에서는 거래정지 종목이 제외되어 있습니다.
 
-    2018년 1월 1일 데이터 부터 조회 가능합니다. 수정주가는 소수점 첫째 자리에서 반올림합니다.
+    KRX 거래소는 2018년 1월 1일 데이터 부터 조회 가능합니다. 수정주가는 소수점 첫째 자리에서 반올림합니다.
+
+    NXT 거래소는 2025년 3월 4일 데이터 부터 조회 가능합니다. 수정주가는 소수점 첫째 자리에서 반올림합니다.
 
     Args:
         date (Optional[dtm.date]): 조회할 날짜. 기본값은 현재 날짜입니다.
@@ -265,7 +267,7 @@ def get_tickers(
         - listing_date (str or None): 상장일
 
     Raises:
-        AssertionError: 잘못된 시장 이름이 입력된 경우 오류를 발생시킵니다.
+        AssertionError: 잘못된 시장이나 거래소 이름이 입력된 경우 오류를 발생시킵니다.
         HTTPError: API 요청이 실패했을 때 발생.
 
     Examples:
@@ -285,7 +287,10 @@ def get_tickers(
             "KOSDAQ",
         ], "market은 'KOSPI' 또는 'KOSDAQ'이어야 합니다."
 
-    # TODO: NXT 거래소 2025년 3월 4일 이전 요청 시 에러 발생
+    assert exchange in [
+        "KRX", "NXT", DataExchange.KRX, DataExchange.NXT
+    ], "exchange는 'KRX' 또는 'NXT'이어야 합니다."
+
     exchange = DataExchange.validate(exchange)
 
     if date is None:
@@ -294,6 +299,10 @@ def get_tickers(
         if schedule.full_day_closed:
             date = get_last_trading_day(date)
 
+    # 2025년 3월 4일 이전 NXT 거래소 요청 시 빈 데이터 반환
+    if exchange == DataExchange.NXT and date < dtm.date(2025, 3, 4):
+        return pd.DataFrame()
+
     return _get_tickers(date, market, adjusted, exchange)
 
 

{pyqqq-0.12.182 → pyqqq-0.12.184}/pyqqq/utils/daily_tickers.py

@@ -28,6 +28,7 @@ class DailyTickers:
             self._date = get_last_trading_day(self._today)
         else:
             self._date = self._today
+        self._is_before_today_krx_open = True
         self._tickers = None
         self._change_date()
 
@@ -36,12 +37,12 @@ class DailyTickers:
 
     @lru_cache(maxsize=30)  # memory_profiler 로 확인 결과 하루치 fetch 결과가 약 2.5MiB
     @staticmethod
-    def fetch_tickers(date: dtm.date) -> pd.DataFrame:
+    def fetch_tickers(date: dtm.date, exchange: str) -> pd.DataFrame:
         DailyTickers.logger.debug(f'\tfetch_tickers date={date} cache_info={DailyTickers.fetch_tickers.cache_info()}')
-        return domestic.get_tickers(date)
+        return domestic.get_tickers(date, exchange=exchange)
 
     def _change_date(self, date: Optional[dtm.date] = None, force: bool = False):
-        """_tickers 가 비었거나 날짜가 바뀌었으면 새로 채워넣는다."""
+        """_tickers 가 비었거나 날짜가 바뀌었으면 새로 채워넣는다. 8시 50분 이전에는 NXT 거래소 데이터를 사용한다."""
         if not date:
             if self._chk_days_passed():
                 self._today = dtm.date.today()
@@ -52,14 +53,27 @@ class DailyTickers:
             else:
                 date = self._date
 
+        if self._today == date:
+            if dtm.datetime.now().time() < dtm.time(8, 50) and not self._is_before_today_krx_open:
+                self._is_before_today_krx_open = True
+                self._tickers = None
+            elif dtm.datetime.now().time() >= dtm.time(8, 50) and self._is_before_today_krx_open:
+                self._is_before_today_krx_open = False
+                self._tickers = None
+            else:
+                pass
+        else:
+            self._is_before_today_krx_open = False
+
         if force:
             DailyTickers.fetch_tickers.cache_clear()
             self._tickers = None
 
         if self._tickers is None or self._date != date:
             self._date = date
+            exchange = "NXT" if self._is_before_today_krx_open else "KRX"
             self.logger.debug(f'\tdate changed. date={self._date} wait for get_tickers()')
-            self._tickers = DailyTickers.fetch_tickers(self._date)
+            self._tickers = DailyTickers.fetch_tickers(self._date, exchange=exchange)
 
     def get_tickers(self, date: Optional[dtm.date] = None, force: bool = False) -> pd.DataFrame:
         """

{pyqqq-0.12.182 → pyqqq-0.12.184}/pyqqq/utils/limiter.py

@@ -3,7 +3,7 @@ import time
 
 _MAX_CALLS = 5
 _PERIOD = 1
-_SCOPE = 'default'
+_SCOPE = "default"
 
 
 class CallLimiter:
@@ -13,12 +13,13 @@ class CallLimiter:
     이 클래스는 API 호출 빈도를 제한하여 너무 많은 요청으로 인해 서비스 제한을 받는 것을 방지합니다.
     이 클래스의 인스턴스는 전역에서 단 하나만 존재하며, 다양한 'scope'에 대한 호출 제한 윈도우를 관리합니다.
     """
+
     _instance = None
 
     def __new__(cls):
         if cls._instance is None:
             cls._instance = super(CallLimiter, cls).__new__(cls)
-            cls._instance.windows = {'default': []}
+            cls._instance.windows = {"default": []}
 
         return cls._instance
 
@@ -63,6 +64,7 @@ def limit_calls(max_calls: int = _MAX_CALLS, period: float = _PERIOD, scope: str
         period (float): 기간(초) 동안의 호출 횟수를 제한합니다.
         scope (str): 호출 제한을 적용할 스코프 이름입니다.
     """
+
     def decorator(func):
         def wrapper(*args, **kwargs):
             CallLimiter().wait_limit_rate(max_calls=max_calls, period=period, scope=scope)

{pyqqq-0.12.182 → pyqqq-0.12.184}/pyqqq/utils/market_schedule.py

@@ -144,7 +144,14 @@ def is_trading_time(
     if exchange == Exchange.NXT and now.date() < datetime.date(2025, 3, 4):
         raise ValueError("NXT 거래소는 2025년 3월 4일 부터 운영되었습니다. 이전 날짜는 지원하지 않습니다.")
 
-    return not (is_full_day_closed(now, exchange) or is_before_opening(now, exchange) or is_after_closing(now, exchange))
+    schedule = get_market_schedule(now.date(), exchange)
+    if schedule.full_day_closed:
+        return False
+
+    if now.time() < schedule.open_time or now.time() > schedule.close_time:
+        return False
+
+    return True
 
 
 def get_market_schedule(