Added the ability to open a deal at a trigger price

app/services/Bybit/functions/Futures.py

@@ -1,6 +1,7 @@
 import asyncio
 import logging.config
 import time
+import json
 import app.services.Bybit.functions.balance as balance_g
 import app.services.Bybit.functions.price_symbol as price_symbol
 import app.telegram.database.requests as rq
@@ -106,6 +107,7 @@ def format_order_details_position(data):
     symbol = msg.get("symbol", "N/A")
     order_type = msg.get("orderType", "N/A")
     side = msg.get("side", "")
+    trigger_price = msg.get("triggerPrice", "N/A")
 
     movement = ""
     if side.lower() == "buy":
@@ -149,6 +151,28 @@ def format_order_details_position(data):
             f"Движение: {movement}\n"
         )
         return text
+    elif order_status.lower() == "untriggered":
+        text = (
+            f"Условный ордер создан:\n"
+            f"Торговая пара: {symbol}\n"
+            f"Цена: {price:.6f}\n"
+            f"Триггер цена: {trigger_price}\n"
+            f"Количество: {qty}\n"
+            f"Тип ордера: {order_type}\n"
+            f"Движение: {movement}\n"
+        )
+        return text
+    elif order_status.lower() == "deactivated":
+        text = (
+            f"Условный ордер отменен:\n"
+            f"Торговая пара: {symbol}\n"
+            f"Цена: {price:.6f}\n"
+            f"Триггер цена: {trigger_price}\n"
+            f"Количество: {qty}\n"
+            f"Тип ордера: {order_type}\n"
+            f"Движение: {movement}\n"
+        )
+        return text
     return None
 
 
@@ -200,6 +224,7 @@ async def handle_execution_message(message, msg):
     """
     tg_id = message.from_user.id
     data = msg.get("data", [{}])[0]
+    logger.info("исполнео:", json.dumps(msg, indent=4))
     data_main_risk_stgs = await rq.get_user_risk_management_settings(tg_id)
     commission_fee = data_main_risk_stgs.get("commission_fee", "ДА")
     pnl = parse_pnl_from_msg(msg)
@@ -275,7 +300,7 @@ async def handle_order_message(message, msg: dict) -> None:
     Обработчик сообщений об исполнении ордера.
     Логирует событие и проверяет условия для мартингейла и TP.
     """
-    # logger.info(f"Исполнен ордер:\n{json.dumps(msg, indent=4, ensure_ascii=False)}")
+    logger.info(f"Исполнен ордер:\n{json.dumps(msg, indent=4, ensure_ascii=False)}")
 
     trade_info = format_order_details_position(msg)
 
@@ -322,7 +347,7 @@ async def open_position(
         bybit_margin_mode = (
             "ISOLATED_MARGIN" if margin_mode == "Isolated" else "REGULAR_MARGIN"
         )
-
+        trigger_price = await rq.get_trigger_price(tg_id)
         limit_price = None
         if order_type == "Limit":
             limit_price = await rq.get_limit_price(tg_id)
@@ -426,18 +451,33 @@ async def open_position(
 
         if bybit_margin_mode == "ISOLATED_MARGIN":
             # Открываем позицию
-            response = client.place_order(
-                category="linear",
-                symbol=symbol,
-                side=side,
-                orderType=order_type,
-                qty=str(quantity),
-                price=(
-                    str(limit_price) if order_type == "Limit" and limit_price else None
-                ),
-                timeInForce="GTC",
-                orderLinkId=f"deal_{symbol}_{int(time.time())}",
-            )
+            if trigger_price and float(trigger_price) > 0:
+                response = client.place_order(
+                    category="linear",
+                    symbol=symbol,
+                    side=side,
+                    orderType="Stop" if order_type == "Conditional" else order_type,
+                    qty=str(quantity),
+                    price=(str(limit_price) if order_type == "Limit" and limit_price else None),
+                    triggerPrice=str(trigger_price),
+                    triggerBy="LastPrice",
+                    triggerDirection=2 if side == "Buy" else 1,
+                    timeInForce="GTC",
+                    orderLinkId=f"deal_{symbol}_{int(time.time())}",
+                )
+            else:
+                # Обычный ордер, без триггера
+                response = client.place_order(
+                    category="linear",
+                    symbol=symbol,
+                    side=side,
+                    orderType=order_type,
+                    qty=str(quantity),
+                    price=(str(limit_price) if order_type == "Limit" and limit_price else None),
+                    timeInForce="GTC",
+                    orderLinkId=f"deal_{symbol}_{int(time.time())}",
+                )
+
             if response.get("retCode", -1) == 0:
                 return True
             if response.get("retCode", -1) != 0:
@@ -544,6 +584,9 @@ async def open_position(
                 slOrderType=sl_order_type,
                 slLimitPrice=str(sl_limit_price) if sl_limit_price else None,
                 tpslMode=tpsl_mode,
+                triggerPrice=str(trigger_price) if trigger_price and float(trigger_price) > 0 else None,
+                triggerBy="LastPrice" if trigger_price and float(trigger_price) > 0 else None,
+                triggerDirection=2 if side == "Buy" else 1 if trigger_price and float(trigger_price) > 0 else None,
                 timeInForce="GTC",
                 orderLinkId=f"deal_{symbol}_{int(time.time())}",
             )
@@ -724,20 +767,20 @@ async def get_active_orders(tg_id, message):
     """
     client = await get_bybit_client(tg_id)
     response = client.get_open_orders(
-        category="linear", settleCoin="USDT", orderType="Limit"
+        category="linear", settleCoin="USDT", orderType="Limit" and "Market"
     )
     orders = response.get("result", {}).get("list", [])
-    limit_orders = [order for order in orders if order.get("orderType") == "Limit"]
+    limit_orders = [order for order in orders]
 
     if limit_orders:
         symbols = [order["symbol"] for order in limit_orders]
         await message.answer(
-            "📈 Ваши активные лимитные ордера:",
+            "📈 Ваши активные ордера:",
             reply_markup=inline_markup.create_trades_inline_keyboard_limits(symbols),
         )
     else:
         await message.answer(
-            "❗️ У вас нет активных лимитных ордеров.",
+            "❗️ У вас нет активных ордеров.",
             reply_markup=inline_markup.back_to_main,
         )
         return
@@ -752,12 +795,12 @@ async def get_active_orders_by_symbol(tg_id, symbol, message):
     limit_orders = [
         order
         for order in active_orders.get("result", {}).get("list", [])
-        if order.get("orderType") == "Limit"
     ]
+    logger.info(limit_orders)
 
     if not limit_orders:
         await message.answer(
-            "Нет активных лимитных ордеров по данной торговой паре.",
+            "Нет активных ордеров по данной торговой паре.",
             reply_markup=inline_markup.back_to_main,
         )
         return
@@ -769,9 +812,8 @@ async def get_active_orders_by_symbol(tg_id, symbol, message):
             f"Тип ордера: {order.get('orderType')}\n"
             f"Сторона: {order.get('side')}\n"
             f"Цена: {order.get('price')}\n"
+            f"Триггер цена: {order.get('triggerPrice')}\n"
             f"Количество: {order.get('qty')}\n"
-            f"Тейк-профит: {order.get('takeProfit')}\n"
-            f"Стоп-лосс: {order.get('stopLoss')}\n"
         )
         texts.append(text)