openfund-core 0.0.4__py3-none-any.whl → 1.0.5__py3-none-any.whl

core/Exchange.py

@@ -0,0 +1,533 @@
+import logging
+import time
+import ccxt
+import pandas as pd
+
+
+from decimal import Decimal
+from core.utils.OPTools import OPTools
+from ccxt.base.exchange import ConstructorArgs
+
+
+class Exchange:
+    BUY_SIDE = 'buy'
+    SELL_SIDE = 'sell'
+    LONG_KEY = 'long'
+    SHORT_KEY = 'short'
+    SIDE_KEY = 'side'
+    SYMBOL_KEY = 'symbol'
+    ENTRY_PRICE_KEY = 'entryPrice'
+    MARK_PRICE_KEY = 'markPrice'
+    CONTRACTS_KEY = 'contracts'
+    def __init__(self, config:ConstructorArgs, exchangeKey:str = "okx",) :
+        # 配置交易所
+        self.exchange = getattr(ccxt, exchangeKey)(config)
+        self.logger = logging.getLogger(__name__)
+
+
+
+    def getMarket(self, symbol:str):
+        # 配置交易对
+        self.exchange.load_markets()
+        
+        return self.exchange.market(symbol)
+
+    def get_tick_size(self,symbol) -> Decimal:     
+        
+        market = self.getMarket(symbol)
+        if market and 'precision' in market and 'price' in market['precision']:            
+            return OPTools.toDecimal(market['precision']['price'])
+        else:
+            raise ValueError(f"{symbol}: 无法从市场数据中获取价格精度")
+
+    def amount_to_precision(self,symbol, contract_size):
+        return self.exchange.amount_to_precision(symbol, contract_size)
+    
+    def get_position_mode(self):
+    
+        try:
+            # 假设获取账户持仓模式的 API
+            response = self.exchange.private_get_account_config()
+            data = response.get('data', [])
+            if data and isinstance(data, list):
+                # 取列表的第一个元素（假设它是一个字典），然后获取 'posMode'
+                position_mode = data[0].get('posMode', 'single')  # 默认值为单向
+                
+                return position_mode
+            else:
+               
+                return 'single'  # 返回默认值
+        except Exception as e:        
+            error_message = f"Error fetching position mode: {e}"
+            self.logger.error(error_message)        
+            raise Exception(error_message)  
+    
+    def set_leverage(self,symbol, leverage, mgnMode='isolated',posSide=None):
+        try:
+            # 设置杠杆
+            params = {
+                # 'instId': instId,
+                'leverage': leverage,
+                'marginMode': mgnMode
+            }
+            if posSide:
+                params['side'] = posSide
+                
+            self.exchange.set_leverage(leverage, symbol=symbol, params=params)
+            self.logger.info(f"{symbol} Successfully set leverage to {leverage}x")
+        except Exception as e:
+            error_message = f"{symbol} Error setting leverage: {e}"
+            self.logger.error(error_message)        
+            raise Exception(error_message)  
+    # 获取价格精度
+    def get_precision_length(self,symbol) -> int:
+        tick_size = self.get_tick_size(symbol)
+        return len(f"{tick_size:.15f}".rstrip('0').split('.')[1]) if '.' in f"{tick_size:.15f}" else 0
+
+    def format_price(self, symbol, price:Decimal) -> str:
+        precision = self.get_precision_length(symbol)
+        return f"{price:.{precision}f}"
+    
+    def convert_contract(self, symbol, amount, price:Decimal, direction='cost_to_contract'):
+        """
+        进行合约与币的转换
+        :param symbol: 交易对符号，如 'BTC/USDT:USDT'
+        :param amount: 输入的数量，可以是合约数量或币的数量
+        :param direction: 转换方向，'amount_to_contract' 表示从数量转换为合约，'cost_to_contract' 表示从金额转换为合约
+        :return: 转换后的数量
+        """
+
+        # 获取合约规模
+        market_contractSize = OPTools.toDecimal(self.getMarket(symbol)['contractSize'])
+        amount = OPTools.toDecimal(amount)
+        if direction == 'amount_to_contract':
+            contract_size = amount / market_contractSize
+        elif direction == 'cost_to_contract':
+            contract_size = amount / price / market_contractSize
+        else:
+            raise Exception(f"{symbol}:{direction} 是无效的转换方向，请输入 'amount_to_contract' 或 'cost_to_contract'。")
+        
+        return self.amount_to_precision(symbol, contract_size)
+   
+     
+    def cancel_all_orders(self, symbol):
+        max_retries = 3
+        retry_count = 0
+        
+        while retry_count < max_retries:
+            try:
+                # 获取所有未完成订单
+                params = {
+                    # 'instId': instId
+                }
+                open_orders = self.exchange.fetch_open_orders(symbol=symbol, params=params)
+                
+                # 批量取消所有订单
+                if open_orders:
+                    order_ids = [order['id'] for order in open_orders]
+                    self.exchange.cancel_orders(order_ids, symbol, params=params)
+                    
+                    self.logger.debug(f"{symbol}: {order_ids} 挂单取消成功.")
+                else:
+                    self.logger.debug(f"{symbol}: 无挂单.")
+                return True
+                
+            except Exception as e:
+                retry_count += 1
+                if retry_count == max_retries:
+                    error_message = f"{symbol} 取消挂单失败(重试{retry_count}次): {str(e)}"
+                    self.logger.error(error_message)
+                    raise Exception(error_message)                
+                else:
+                    self.logger.warning(f"{symbol} 取消挂单失败，正在进行第{retry_count}次重试: {str(e)}")
+                    time.sleep(0.1)  # 重试前等待0.1秒
+
+    def cancel_all_algo_orders(self, symbol, attachType=None) -> bool:
+        """_summary_
+
+        Args:
+            symbol (_type_): _description_
+            attachType (_type_, optional): "TP"|"SL". Defaults to None.
+        """
+        
+        params = {
+            "ordType": "conditional",
+        }
+        try:
+            orders = self.fetch_open_orders(symbol=symbol,params=params)
+        except Exception as e:
+            error_message = f"!!{symbol} : Error fetching open orders: {e}"
+            self.logger.error(error_message)
+            raise Exception(error_message)
+       
+        
+        if len(orders) == 0:           
+            self.logger.debug(f"{symbol} 未设置策略订单列表。")
+            return True
+     
+        algo_ids = []
+        if attachType and attachType == 'SL':
+            algo_ids = [order['id'] for order in orders if order['stopLossPrice'] and order['stopLossPrice'] > 0.0 ]
+        elif attachType and attachType == 'TP':
+            algo_ids = [order['id'] for order in orders if order['takeProfitPrice'] and order['takeProfitPrice'] > 0.0]
+        else :
+            algo_ids = [order['id'] for order in orders ]
+            
+        if len(algo_ids) == 0 :         
+            self.logger.debug(f"{symbol} 未设置策略订单列表。")
+            return True
+        
+        max_retries = 3
+        retry_count = 0
+        
+        while retry_count < max_retries:
+            try:
+                params = {
+                    "algoId": algo_ids,
+                    "trigger": 'trigger'
+                }
+                rs = self.exchange.cancel_orders(ids=algo_ids, symbol=symbol, params=params)
+                
+                return len(rs) > 0
+       
+            except Exception as e:
+                retry_count += 1
+                if retry_count == max_retries:
+                    error_message = f"!!{symbol} : Error cancelling order {algo_ids}: {e}"
+                    self.logger.error(error_message)
+                    raise Exception(error_message) 
+                   
+                self.logger.warning(f"{symbol} : Error cancelling order {algo_ids}: {str(e)}")
+                time.sleep(0.1)  # 重试前等待0.1秒 
+    def place_algo_orders(self, symbol, position, price: Decimal, order_type, sl_or_tp='SL', params={}) -> bool:
+        """
+        下单
+        Args:
+            symbol: 交易对
+            position: 仓位
+            price: 下单价格
+            order_type: 订单类型
+        """
+        # 计算下单数量
+        amount = abs(position[self.CONTRACTS_KEY])
+    
+        if amount <= 0:
+            self.logger.warning(f"{symbol}: amount is 0 for {symbol}")
+            return
+
+        # 止损单逻辑 
+        adjusted_price = self.format_price(symbol, price)
+        
+        # 默认市价止损，委托价格为-1时，执行市价止损。
+        sl_params = {
+            **params,
+            'slTriggerPx':adjusted_price , 
+            'slOrdPx':'-1', # 委托价格为-1时，执行市价止损
+            # 'slOrdPx' : adjusted_price,
+            'slTriggerPxType':'last',
+            'tdMode':position['marginMode'],
+            'sz': str(amount),        
+            'cxlOnClosePos': True,
+            'reduceOnly':True,            
+        }
+        
+        tp_params = {  
+            **params,
+            'tpTriggerPx':adjusted_price,
+            'tpOrdPx' : adjusted_price,
+            'tpOrdKind': 'condition',
+            'tpTriggerPxType':'last',            
+            'tdMode':position['marginMode'],
+            'sz': str(amount),          
+            'cxlOnClosePos': True,
+            'reduceOnly':True
+        }
+        
+        order_params = sl_params if sl_or_tp == 'SL' else tp_params
+        # order_params.update(params)
+        
+        if order_type == 'limit' and sl_or_tp =='SL':
+            order_params['slOrdPx'] = adjusted_price
+        
+        orderSide = self.BUY_SIDE if position[self.SIDE_KEY] == self.SHORT_KEY else self.SELL_SIDE # 和持仓反向相反下单
+        
+        order = {
+            'symbol': symbol,
+            'side': orderSide,
+            'type': order_type,
+            'amount': amount,
+            'price': adjusted_price,
+            'params': order_params
+        }
+        
+        max_retries = 3
+        retry_count = 0
+        self.logger.debug(f"{symbol} : Pre Algo Order placed:  {order} ")
+        while retry_count < max_retries:
+            try:
+                
+                self.exchange.create_order(
+                    **order
+                    # symbol=symbol,
+                    # type=order_type,
+                    # price=adjusted_price,
+                    # side=orderSide,
+                    # amount=amount,
+                    # params=order_params
+                )             
+                
+                break               
+               
+            except ccxt.NetworkError as e:
+                # 处理网络相关错误
+                retry_count += 1
+                self.logger.warning(f"{symbol} : 设置止盈止损时发生网络错误,正在进行第{retry_count}次重试: {str(e)}")
+                time.sleep(0.1)  # 重试前等待1秒
+                continue
+            except ccxt.ExchangeError as e:
+                # 处理交易所API相关错误
+                retry_count += 1
+                self.logger.warning(f"{symbol} : 设置止盈止损单时发生交易所错误,正在进行第{retry_count}次重试: {str(e)}")
+                time.sleep(0.1)
+                continue
+            except Exception as e:
+                # 处理其他未预期的错误
+                retry_count += 1
+                self.logger.warning(f"{symbol} : 设置止盈止损单时发生未知错误,正在进行第{retry_count}次重试: {str(e)}")
+                time.sleep(0.1)
+                continue
+            
+        if retry_count >= max_retries:
+            # 重试次数用完仍未成功设置止损单
+            error_message = f"!! {symbol}: 设置止盈止损单时重试次数用完仍未成功设置成功。 "
+            self.logger.error(error_message)
+            raise Exception(error_message)
+        self.logger.debug(f"{symbol} : --------- ++ Order placed done. --------")
+        return True
+        
+
+    def place_order(self, symbol, price: Decimal, amount_usdt, side, leverage=20, order_type='limit', params={}) -> bool: 
+        """
+        下单
+        Args:
+            symbol: 交易对
+            price: 下单价格
+            amount_usdt: 下单金额
+            side: 下单方向
+            order_type: 订单类型
+        """       
+        # 格式化价格
+        adjusted_price = self.format_price(symbol, price)
+
+        if amount_usdt <= 0:
+            self.logger.warning(f"{symbol}: amount_usdt must be greater than 0")
+            return
+    
+        pos_side = self.LONG_KEY if side == self.BUY_SIDE else self.SHORT_KEY
+
+        # 设置杠杆 
+        self.set_leverage(symbol=symbol, leverage=leverage, mgnMode='isolated',posSide=pos_side)  
+        # 20250220 SWAP类型计算合约数量 
+        contract_size = self.convert_contract(symbol=symbol, price = OPTools.toDecimal(adjusted_price) ,amount=amount_usdt)
+
+        order_params = {
+            **params,
+            "tdMode": 'isolated',
+            "side": side,
+            "ordType": order_type,
+            "sz": contract_size,
+            "px": adjusted_price
+        } 
+        
+        # # 模拟盘(demo_trading)需要 posSide
+        # if self.is_demo_trading == 1 :
+        #     params["posSide"] = pos_side
+            
+        order = {
+            'symbol': symbol,
+            'side': side,
+            'type': order_type,
+            'amount': contract_size,
+            'price': adjusted_price,
+            'params': order_params
+        }
+        
+        max_retries = 3
+        retry_count = 0
+
+        while retry_count < max_retries:
+            try:
+                # 使用ccxt创建订单
+                self.logger.debug(f"{symbol} : Pre Order placed:  {order} ")
+                order_result = self.exchange.create_order(
+                    **order
+                    # symbol=symbol,
+                    # type='limit',
+                    # side=side,
+                    # amount=amount_usdt,
+                    # price=float(adjusted_price),
+                    # params=params
+                )
+            except ccxt.NetworkError as e:
+                # 处理网络相关错误
+                retry_count += 1
+                self.logger.warning(f"{symbol} : 设置下单时发生网络错误,正在进行第{retry_count}次重试: {str(e)}")
+                time.sleep(0.1)  # 重试前等待1秒
+                continue
+            except ccxt.ExchangeError as e:
+                # 处理交易所API相关错误
+                retry_count += 1
+                self.logger.warning(f"{symbol} : 设置下单时发生交易所错误,正在进行第{retry_count}次重试: {str(e)}")
+                time.sleep(0.1)
+                continue
+            except Exception as e:
+                # 处理其他未预期的错误
+                retry_count += 1
+                self.logger.warning(f"{symbol} : 设置下单时发生未知错误,正在进行第{retry_count}次重试: {str(e)}")
+                time.sleep(0.1)
+                continue
+        if retry_count >= max_retries:
+            # 重试次数用完仍未成功设置止损单
+            error_message = f"!! {symbol}: 设置止盈止损单时重试次数用完仍未成功设置成功。 "
+            self.logger.error(error_message)
+            raise Exception(error_message)        
+        self.logger.debug(f"{symbol} : --------- ++ Order placed done. --------")  
+        return True
+      
+    def fetch_position(self, symbol):
+        """_summary_
+
+        Args:
+            symbol (_type_): _description_
+
+        Returns:
+            _type_: _description_
+        """
+        
+        max_retries = 3
+        retry_count = 0
+        
+        while retry_count < max_retries:
+            try:
+                position = self.exchange.fetch_position(symbol=symbol)
+                if position :
+                    # self.logger.debug(f"{symbol} 有持仓合约数: {position['contracts']}")
+                    return position
+                return None
+            except Exception as e:
+                retry_count += 1
+                if retry_count == max_retries:
+                    error_message = f"!!{symbol} 获取持仓失败(重试{retry_count}次): {str(e)}"
+                    self.logger.error(error_message)
+                    raise Exception(error_message)
+                   
+                self.logger.warning(f"{symbol} 检查持仓失败，正在进行第{retry_count}次重试: {str(e)}")
+                time.sleep(0.1)  # 重试前等待0.1秒
+    
+    def fetch_positions(self):
+        """_summary_
+        Returns:
+            _type_: _description_
+        """
+        max_retries = 3
+        retry_count = 0
+
+        while retry_count < max_retries:
+            try:
+                positions = self.exchange.fetch_positions()
+                return positions
+            except Exception as e:
+                retry_count += 1
+                if retry_count == max_retries:
+                    error_message = f"!! 获取持仓列表失败(重试{retry_count}次): {str(e)}"
+                    self.logger.error(error_message)
+                    raise Exception(error_message)
+
+                self.logger.warning(f"获取持仓列表失败，正在进行第{retry_count}次重试: {str(e)}")
+                time.sleep(0.1)  # 重试前等待0.1秒
+    
+    def fetch_open_orders(self,symbol,params={}):
+        max_retries = 3
+        retry_count = 0
+        
+        while retry_count < max_retries:
+            try:
+                orders = self.exchange.fetch_open_orders(symbol=symbol,params=params)
+                return orders
+            
+            except Exception as e:
+                retry_count += 1
+                if retry_count == max_retries:
+                    error_message = f"{symbol} : fetching open orders(retry {retry_count} times): {str(e)}"
+                    self.logger.error(error_message)
+                    raise Exception(error_message)
+                   
+                self.logger.warning(f"{symbol} : Error fetching open orders: {str(e)}")
+                time.sleep(0.1)  # 重试前等待0.1秒     
+    def get_market_price(self, symbol) -> Decimal:
+        """
+        获取最新价格
+        Args:
+            symbol: 交易对
+        """
+        max_retries = 3
+        retry_count = 0
+
+        while retry_count < max_retries:
+            try:
+                ticker = self.exchange.fetch_ticker(symbol)
+                if ticker and 'last' in ticker:
+                    return OPTools.toDecimal(ticker['last'])
+                else:
+                    raise Exception(f"{symbol} : Unexpected response structure or missing 'last' price")
+            except Exception as e:
+                retry_count += 1
+                if retry_count == max_retries:
+                    error_message = f"{symbol} 获取最新价格失败(重试{retry_count}次): {str(e)}"
+                    self.logger.error(error_message)
+                    raise Exception(error_message)
+            
+    def get_historical_klines(self, symbol, bar='15m', limit=300, after:str=None, params={}):
+        """
+        获取历史K线数据
+        Args:
+            symbol: 交易对
+            bar: K线周期
+            limit: 数据条数
+            after: 之后时间，格式为 "2025-05-21 23:00:00+08:00"
+        """
+           
+        params = {
+            **params,
+            # 'instId': instId,
+        }
+        since = None
+        if after:
+            since = self.exchange.parse8601(after)
+            limit = None
+            if since:
+                params['paginate'] = True
+            
+        klines = self.exchange.fetch_ohlcv(symbol, timeframe=bar,since=since, limit=limit, params=params)
+        # if 'data' in response and len(response['data']) > 0:
+        if klines :
+            # return response['data']
+            return klines
+        else:
+            raise Exception(f"{symbol} : Unexpected response structure or missing candlestick data")
+        
+    def get_historical_klines_df(self, symbol, bar='15m', limit=300, after:str=None, params={}) -> pd.DataFrame:
+        klines = self.get_historical_klines(symbol, bar=bar, limit=limit, after=after, params=params)
+        return self.format_klines(klines)
+
+    def format_klines(self, klines) -> pd.DataFrame:       
+        """_summary_
+            格式化K线数据
+        Args:
+            klines (_type_): _description_
+        """
+        klines_df = pd.DataFrame(klines, columns=['timestamp', 'open', 'high', 'low', 'close', 'volume']) 
+        # 转换时间戳为日期时间
+        klines_df['timestamp'] = pd.to_datetime(klines_df['timestamp'], unit='ms').dt.tz_localize('UTC').dt.tz_convert('Asia/Shanghai')
+  
+        return klines_df

core/main.py

@@ -0,0 +1,23 @@
+import logging
+from pyfiglet import Figlet
+
+def main():   
+
+    # import importlib.metadata
+    # package_name = __package__ or "openfund-core"
+    # version = importlib.metadata.version("openfund-core")
+
+    # 创建日志记录器并设置输出到屏幕
+    logger = logging.getLogger(__name__)
+    console_handler = logging.StreamHandler()
+    logger.addHandler(console_handler)
+    # # 设置日志级别为INFO
+    logger.setLevel(logging.INFO)
+    
+    f = Figlet(font="standard")  # 字体可选（如 "block", "bubble"）
+    logger.info(f"\n{f.renderText("OpenFund Core")}")
+    
+
+
+if __name__ == "__main__":
+    main()

core/smc/SMCBase.py

@@ -0,0 +1,130 @@
+from decimal import Decimal
+import logging
+import pandas as pd
+import numpy as np
+from core.utils.OPTools import OPTools
+
+class SMCBase(object):
+    HIGH_COL = "high"
+    LOW_COL = "low"
+    CLOSE_COL = "close"
+    OPEN_COL = "open"
+    VOLUME_COL = "volume"
+    AMOUNT_COL = "amount"
+    TIMESTAMP_COL = "timestamp"
+    ATR_COL = "atr"
+
+    BUY_SIDE = "buy"
+    SELL_SIDE = "sell"
+    
+
+    def __init__(self):  
+        self.logger = logging.getLogger(__name__)
+
+    @staticmethod
+    def check_columns(df: pd.DataFrame, required_columns: list) -> bool:
+        """
+        检查DataFrame是否包含指定的列
+        参数:
+            df (pd.DataFrame): 要检查的DataFrame
+            columns (list): 要检查的列名列表
+        返回:
+            bool: 如果DataFrame包含所有指定的列，则返回True；否则返回False
+        """
+        has_pass = all(col in df.columns for col in required_columns)
+        if not has_pass:
+            raise ValueError(f"DataFrame必须包含列: {required_columns}")
+        return has_pass
+
+    @staticmethod
+    def toDecimal(value, precision:int=None) -> Decimal:
+        return OPTools.toDecimal(value, precision)
+
+    @staticmethod
+    def get_precision_length(value) -> int:
+        return len(f"{value:.15f}".rstrip('0').split('.')[1]) if '.' in f"{value:.15f}" else 0
+
+    @staticmethod  
+    def calculate_atr(df, period=14, multiplier=2):
+        """
+        计算增强版ATR指标，等效于Pine Script中的 ta.highest(ta.atr(200),200)*2
+        
+        参数:
+            df: 包含OHLCV数据的DataFrame
+            period: ATR计算周期，默认200
+            multiplier: 放大倍数，默认2
+        
+        返回:
+            增强版ATR序列
+        """
+        # df = data.copy()
+        # 计算真实波幅(TR)
+        high = df[SMCBase.HIGH_COL]
+        low = df[SMCBase.LOW_COL]
+        close = df[SMCBase.CLOSE_COL]
+        
+        close_prev = close.shift(1)
+        tr = pd.DataFrame({
+            'tr1': high - low,
+            'tr2': abs(high - close_prev),
+            'tr3': abs(low - close_prev)
+        }).max(axis=1)
+        
+        # 计算ATR (使用简单移动平均)
+        atr = tr.rolling(window=period, min_periods=1).mean()
+        
+        # 计算ATR的N周期最大值
+        max_atr = atr.rolling(window=period, min_periods=1).max()
+
+        
+        # 应用放大倍数
+        enhanced_atr = max_atr * multiplier
+        
+        return enhanced_atr
+
+
+
+
+    def calculate_atr_with_smoothing(df, length=14, smoothing='RMA'):  
+        """
+        计算ATR (Average True Range) 指标
+        
+        参数:
+        df (pd.DataFrame): 包含OHLCV数据的DataFrame，需包含列：['high', 'low', 'close']
+        length (int): 计算周期，默认为14
+        smoothing (str): 平滑方法，支持 'RMA', 'SMA', 'EMA', 'WMA'，默认为 'RMA'
+        
+        返回:
+        pd.Series: ATR值序列
+        """
+        # 确保数据包含所需的列
+        required_columns = [SMCBase.HIGH_COL, SMCBase.LOW_COL, SMCBase.CLOSE_COL]
+        SMCBase.check_columns(df, required_columns)
+        
+        # 计算真实波幅 (TR)
+        high_low = df[SMCBase.HIGH_COL] - df[SMCBase.LOW_COL]
+        high_close = (df[SMCBase.HIGH_COL] - df[SMCBase.CLOSE_COL].shift()).abs()
+        low_close = (df[SMCBase.LOW_COL] - df[SMCBase.CLOSE_COL].shift()).abs()
+        
+        # 计算TR列，取三个值中的最大值
+        tr = pd.concat([high_low, high_close, low_close], axis=1).max(axis=1)
+        
+        # Apply smoothing
+        if smoothing == 'RMA':
+            # RMA is approximately an EMA with alpha = 1/length
+            atr = tr.ewm(alpha=1/length, adjust=False).mean()
+        elif smoothing == 'SMA':
+            atr = tr.rolling(window=length).mean()
+        elif smoothing == 'EMA':
+            atr = tr.ewm(span=length, adjust=False).mean()
+        elif smoothing == 'WMA':
+            # WMA implementation
+            weights = pd.Series(range(1, length+1))
+            def wma(series):
+                return (series * weights).sum() / weights.sum()
+            atr = tr.rolling(window=length).apply(wma)
+        else:
+            raise ValueError("Invalid smoothing method. Use 'RMA', 'SMA', 'EMA', or 'WMA'")
+        
+        return atr
+