pyqqq 0.12.177__tar.gz → 0.12.179__tar.gz

{pyqqq-0.12.177 → pyqqq-0.12.179}/PKG-INFO

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

{pyqqq-0.12.177 → pyqqq-0.12.179}/pyproject.toml

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

{pyqqq-0.12.177 → pyqqq-0.12.179}/pyqqq/backtest/broker.py

@@ -25,6 +25,7 @@ from pyqqq.data.us_stocks import get_ticker_info as get_us_ticker_info
 from pyqqq.datatypes import *
 from pyqqq.utils.casting import casting
 from pyqqq.utils.market_schedule import get_last_trading_day, get_market_schedule, get_trading_day_with_offset
+from pyqqq.utils.position_classfier import PositionClassifier
 
 MarketType = Literal["kr_stock", "us_stock"]
 
@@ -249,6 +250,8 @@ class TradingBroker(BaseBroker):
         trading_api: Union[KISSimpleDomesticStock, KISSimpleOverseasStock],
         clock: WallClock,
         market_nxt_on: bool = False,
+        strategy_name: Optional[str] = None,
+        classifier_type: Optional[str] = None,
     ):
         """TradingBroker 클래스의 초기화 메서드입니다.
 
@@ -266,6 +269,14 @@ class TradingBroker(BaseBroker):
         self.data_api = data_api
         self.trading_api = trading_api
         self.market_nxt_on = market_nxt_on
+        self.position_classifier = None
+
+        if classifier_type and classifier_type in ["auto", "direct"]:
+            if strategy_name:
+                kv_store_collection = f"{strategy_name}_classifier"
+            else:
+                kv_store_collection = f"account_{self.trading_api.account_no}_classifier"
+            self.position_classifier = PositionClassifier(simple_data_api=self.trading_api, kv_store_collection=kv_store_collection, default_type=classifier_type)
 
     def get_account(self) -> dict:
         return self.trading_api.get_account()
@@ -314,6 +325,7 @@ class TradingBroker(BaseBroker):
                     result = self.data_api.get_historical_daily_data(code, from_date, to_date, adjusted_price=True, data_exchange=DataExchange.KRX)
             else:
                 result = self.data_api.get_historical_daily_data(code, from_date, to_date, adjusted_price=True, data_exchange=DataExchange.KRX)
+        result = result.sort_index(ascending=True)
         return result
 
     def get_orderbook(self, code: str, data_exchange: Optional[DataExchange] = None):
@@ -349,7 +361,20 @@ class TradingBroker(BaseBroker):
 
     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:
         self.logger.debug(f"create_order: {asset_code} {side} {quantity} {order_type} {price} {exchange}")
-        return self.trading_api.create_order(asset_code, side, quantity, order_type, price, exchange=exchange)
+
+        if side is OrderSide.SELL and self.position_classifier:
+            # 포지션 분류기를 사용하여 전략에서 보유중인 수량으로 매도 가능 수량 재 계산
+            calc_quantity = self.position_classifier.get_sellable_quantity(asset_code, quantity)
+
+            if calc_quantity != quantity:
+                self.logger.warning(f"Requested sell quantity {quantity} for {asset_code} adjusted to {calc_quantity} based on position classifier.")
+                quantity = calc_quantity
+
+        order_no = self.trading_api.create_order(asset_code, side, quantity, order_type, price, exchange=exchange)
+        if order_no and self.position_classifier:
+            # 주문이 성공적으로 생성되면 포지션 분류기에 주문 정보를 업데이트
+            self.position_classifier.tagging_order_auto(order_no)
+        return order_no
 
     def update_order(self, org_order_no: str, order_type: OrderType, price: int | Decimal, quantity: int = 0, exchange: OrderExchange = OrderExchange.KRX):
         if isinstance(self.trading_api, KISSimpleOverseasStock):
@@ -386,6 +411,13 @@ class TradingBroker(BaseBroker):
             self.logger.debug(f"cancel_order: {order_no} {quantity}")
             return self.trading_api.cancel_order(order_no, quantity)
 
+    async def start_classifier(self):
+        """ 포지션 분류기를 시작합니다.
+        이 메서드는 포지션 분류기를 사용한다면 반드시 호출해야 합니다.
+        """
+        if self.position_classifier:
+            await self.position_classifier.start()
+
 
 # 백테스팅 할때 사용되는 브로커
 class MockBroker(BaseBroker):

{pyqqq-0.12.177 → pyqqq-0.12.179}/pyqqq/backtest/environment.py

@@ -1,7 +1,7 @@
 import datetime as dtm
 import os
 from abc import ABC
-from typing import Literal
+from typing import Literal, Optional
 from zoneinfo import ZoneInfo
 
 from pyqqq.backtest.broker import BaseBroker, MockBroker, TradingBroker
@@ -142,12 +142,18 @@ class KISDomesticEnvironment(TradingEnvironment):
         ```
     """
 
-    def __init__(self, paper_trading: bool = False, market_nxt_on: bool = False):
+    def __init__(self, paper_trading: bool = False, market_nxt_on: bool = False, strategy_name: Optional[str] = None, classifier_type: Optional[str] = None):
         """KISDomesticEnvironment 클래스의 초기화 메서드입니다.
 
         Args:
             paper_trading (bool, optional): 모의투자 사용 여부. Defaults to False.
                                           True일 경우 모의투자 계좌 사용
+            market_nxt_on (bool, optional): NXT 시장 처리 여부. Defaults to False.
+            strategy_name (str, optional): 전략 이름. Defaults to None.
+                position classifier 데이터가 저장되는 kvstore collection 이름에 사용됨.
+            classifier_type (str, optional): "direct", "auto" 중 하나로 설정시 position classifier 사용하게 됨. Defaults to None.
+                해당 값은 분류기가 기본 포지션을 어떤 방식으로 결정할지에 대한 설정.
+
 
         Raises:
             KeyError: 필요한 환경변수가 설정되지 않은 경우
@@ -160,6 +166,8 @@ class KISDomesticEnvironment(TradingEnvironment):
             trading_api=conn.broker_simple if not paper_trading else conn.paper_broker_simple,
             clock=self.clock,
             market_nxt_on=market_nxt_on,
+            strategy_name=strategy_name,
+            classifier_type=classifier_type,
         )
 
 

{pyqqq-0.12.177 → pyqqq-0.12.179}/pyqqq/brokerage/kis/simple.py

@@ -291,6 +291,7 @@ class KISSimpleDomesticStock:
         분봉 데이터 검색
 
         Note:
+            - 시간외 종가, 시간외 단일가 데이터는 포함되어 있지 않습니다.
             - 두 거래소에서 공통으로 거래정지된 종목의 시가/고가/저가/종가는 모두 동일하며, 그 외 값은 모두 0 입니다.
             - NXT 매매체결 종목이 아니거나, NXT 거래소에서 거래가 불가능한 종목을 DataExchange.NXT 또는 DataExchange.UN으로 조회하면, 모든 값은 0 입니다.
 
@@ -307,6 +308,10 @@ class KISSimpleDomesticStock:
         result = []
         schedule = get_market_schedule(dtm.date.today(), exchange="KRX" if data_exchange == DataExchange.KRX else "NXT")
 
+        # 만약 현재 시간이 거래소 마감 시간을 지났다면, 마감 시간으로 설정
+        if request_time.time() > schedule.close_time:
+            request_time = request_datetime.replace(hour=schedule.close_time.hour, minute=schedule.close_time.minute)
+
         while True:
             r = self.stock_api.inquire_time_itemchartprice(
                 asset_code,

{pyqqq-0.12.177 → pyqqq-0.12.179}/pyqqq/data/minutes.py

@@ -18,7 +18,7 @@ minuteCache = DiskCacheManager("minute_cache")
 @minuteCache.memoize()
 def get_all_minute_data(
     time: datetime.datetime,
-    source: str = "ebest",
+    source: str = "kis",
     adjusted: bool = True,
     exchange: Union[str, DataExchange] = "KRX",
 ) -> pd.DataFrame:
@@ -31,7 +31,7 @@ def get_all_minute_data(
 
     Args:
         time (datetime.datetime): 조회할 시간
-        source (str): 데이터를 검색할 API. 'ebest' 또는 'kis'를 지정할 수 있습니다. 기본값은 'ebest'입니다.
+        source (str): 데이터를 검색할 API. 'ebest' 또는 'kis'를 지정할 수 있습니다. 기본값은 'kis'입니다.
         adjusted (bool): 수정주가 여부. 기본값은 True.
         exchange (Union[str, DataExchange]): 거래소. 기본값은 KRX.
 
@@ -130,7 +130,7 @@ def get_all_day_data(
     date: datetime.date,
     codes: list[str] | str,
     period: datetime.timedelta = datetime.timedelta(minutes=1),
-    source: str = "ebest",
+    source: str = "kis",
     adjusted: bool = True,
     ascending: bool = True,
     exchange: Union[str, DataExchange] = "KRX",
@@ -146,7 +146,7 @@ def get_all_day_data(
         date (datetime.date): 데이터를 검색할 날짜.
         codes (list[str]): 조회할 주식 코드들의 리스트. 최대 20개까지 지정할 수 있습니다.
         period (datetime.timedelta, optional): 반환된 데이터의 시간 간격. 기본값은 1분입니다. 30초 이상의 값을 30초간격으로 지정할 수 있습니다.
-        source (str, optional): 데이터를 검색할 API. 'ebest' 또는 'kis'를 지정할 수 있습니다. 기본값은 'ebest'입니다.
+        source (str, optional): 데이터를 검색할 API. 'ebest' 또는 'kis'를 지정할 수 있습니다. 기본값은 'kis'입니다.
         adjusted (bool): 수정주가 여부. 기본값은 True.
         ascending (bool): 오름차순 여부. 기본값은 True.
         exchange (Union[str, DataExchange]): 거래소. 기본값은 KRX. (cf. NXT의 경우 해당되지 않는 종목은 Empty DataFrame이 반환됩니다.)

{pyqqq-0.12.177 → pyqqq-0.12.179}/pyqqq/utils/position_classifier.py

@@ -107,7 +107,7 @@ class PositionClassifier:
         self.logger.info(f"set_initial_position.\npositions: {cur_pos}\nauto_positions: {self.auto_positions}\ndirect_positions: {self.direct_positions}")
 
     def set_initial_order(self):
-        cur_order = self.api.get_pending_orders()
+        cur_order = self.api.get_pending_orders(exchanges=[OrderExchange.KRX, OrderExchange.NXT, OrderExchange.SOR])
         kv_auto_orders = self.kv_store.get(self.DEF_AUTO_ORDER_KEY) or {}
         kv_direct_orders = self.kv_store.get(self.DEF_DIRECT_ORDER_KEY) or {}
         kv_auto_tag_orders = self.kv_store.get(self.DEF_TAG_AUTO_ORDER_KEY) or []