diff --git a/src/config.py b/src/config.py index e90ddf8..9d59143 100644 --- a/src/config.py +++ b/src/config.py @@ -22,6 +22,9 @@ class Config: LOG_MAX_BYTES = 10 * 1024 * 1024 # 10MB LOG_BACKUP_COUNT = 5 + # strategy data + DATA_DIR = "../data" + # API Rate limiting RATE_LIMIT_REQUESTS = 100 RATE_LIMIT_PERIOD = 60 # seconds diff --git a/src/position_manager.py b/src/position_manager.py index 0e1c2b5..8a50601 100644 --- a/src/position_manager.py +++ b/src/position_manager.py @@ -1,6 +1,7 @@ import os import json from logger_config import get_logger +from config import Config from trade_constants import ( ORDER_DIRECTION_BUY, ORDER_TYPE_LIMIT, @@ -29,7 +30,7 @@ class PositionManager: self.positions: Dict[str, LocalPosition] = {} # {股票代码 -> LocalPosition} # 待处理订单信息 self.pending_orders = {} # {order_id -> LocalOrder} - self.data_path = self.strategy_name + "_positions.json" + self.data_path = os.path.join(Config.DATA_DIR, self.strategy_name + "_positions.json") self.load_data() def update_position(self, code, direction, amount):