Update

app/services/Bybit/functions/functions.py

@@ -1,8 +1,8 @@
 import asyncio
-
+import logging.config
 from aiogram import F, Router
-
-from app.services.Bybit.functions import Futures, func_min_qty
+from logger_helper.logger_helper import LOGGING_CONFIG
+from app.services.Bybit.functions import Futures, min_qty
 from app.services.Bybit.functions.Futures import open_position, close_user_trade, get_active_positions, trading_cycle
 from app.services.Bybit.functions.balance import get_balance
 import app.telegram.Keyboards.inline_keyboards as inline_markup
@@ -15,6 +15,11 @@ from aiogram.types import Message, CallbackQuery
 from aiogram.fsm.state import State, StatesGroup
 from aiogram.fsm.context import FSMContext
 
+from app.services.Bybit.functions.get_valid_symbol import get_valid_symbols
+
+logging.config.dictConfig(LOGGING_CONFIG)
+logger = logging.getLogger("functions")
+
 router_functions_bybit_trade = Router()
 
 
@@ -30,6 +35,10 @@ class TradeSetup(StatesGroup):
     waiting_for_timer = State()
     waiting_for_positive_percent = State()
 
+class state_limit_price(StatesGroup):
+    price = State()
+
+
 
 @router_functions_bybit_trade.callback_query(F.data.in_(['clb_start_trading', 'clb_back_to_main', 'back_to_main']))
 async def clb_start_bybit_trade_message(callback: CallbackQuery, state: FSMContext):
@@ -48,7 +57,8 @@ async def clb_start_bybit_trade_message(callback: CallbackQuery, state: FSMConte
 Как начать торговлю?
 
 1️⃣ Проверьте и тщательно настройте все параметры в вашем профиле.  
-2️⃣ Нажмите ниже кнопку 'Указать торговую пару' и введите торговую пару заглавными буквами, без лишних символов (например: BTCUSDT).  
+2️⃣ Нажмите ниже кнопку 'Указать торговую пару' и введите торговую пару, без лишних символов (например: BTCUSDT). 
+3️⃣ Нажмите кнопку 'Выбрать тип входа' и после нажмите начать торговлю. 
 '''
         await callback.message.edit_text(text=text, parse_mode='html', reply_markup=inline_markup.trading_markup)
 
@@ -69,7 +79,8 @@ async def start_bybit_trade_message(message, state):
 Как начать торговлю?
 
 1️⃣ Проверьте и тщательно настройте все параметры в вашем профиле.  
-2️⃣ Нажмите ниже кнопку 'Указать торговую пару' и введите торговую пару заглавными буквами, без лишних символов (например: BTCUSDT).  
+2️⃣ Нажмите ниже кнопку 'Указать торговую пару' и введите торговую пару, без лишних символов (например: BTCUSDT).
+3️⃣ Нажмите кнопку 'Выбрать тип входа' и после нажмите начать торговлю.  
 '''
 
         await message.answer(text=text, parse_mode='html', reply_markup=inline_markup.trading_markup)
@@ -80,17 +91,23 @@ async def update_symbol_for_trade_message(callback: CallbackQuery, state: FSMCon
     await state.set_state(state_update_symbol.symbol)
 
     await callback.message.answer(
-        text='Укажите торговую пару заглавными буквами без пробелов и лишних символов (пример: BTCUSDT): ')
+        text='Укажите торговую пару заглавными буквами без пробелов и лишних символов (пример: BTCUSDT): ',
+        reply_markup=inline_markup.cancel)
 
 
 @router_functions_bybit_trade.message(state_update_symbol.symbol)
 async def update_symbol_for_trade(message: Message, state: FSMContext):
+    user_input = message.text.strip().upper()
+
+    exists = await get_valid_symbols(message.from_user.id, user_input)
+
+    if not exists:
+        await message.answer("Введена некорректная торговая пара или такой пары нет в списке. Попробуйте снова.")
+        return
+
     await state.update_data(symbol=message.text)
-
-    data = await state.get_data()
-
     await message.answer('Пара была успешно обновлена')
-    await rq.update_symbol(message.from_user.id, data['symbol'])
+    await rq.update_symbol(message.from_user.id, user_input)
     await start_bybit_trade_message(message, state)
 
     await state.clear()
@@ -111,15 +128,47 @@ async def entry_order_type_callback(callback: CallbackQuery, state: FSMContext):
         await callback.answer("Ошибка выбора", show_alert=True)
         return
 
+    if order_type == 'Limit':
+        await state.set_state(state_limit_price.price)
+        await callback.message.answer("Введите цену лимитного ордера:", reply_markup=inline_markup.cancel)
+        await callback.answer()
+        return
+
     try:
         await state.update_data(entry_order_type=order_type)
         await rq.update_entry_order_type(callback.from_user.id, order_type)
-        await callback.answer("Тип входа в позицию был успешно обновлен")
+        await callback.message.answer(f"Выбран тип входа в позицию: {order_type}",
+                              reply_markup=inline_markup.start_trading_markup)
+        await callback.answer()
     except Exception as e:
-        await callback.message.answer("Произошла ошибка при обновлении типа входа в позицию")
+        logger.error(f"Произошла ошибка при обновлении типа входа в позицию: {e}")
+        await callback.message.answer("Произошла ошибка при обновлении типа входа в позицию",
+                                      reply_markup=inline_markup.back_to_main)
     await state.clear()
 
 
+@router_functions_bybit_trade.message(state_limit_price.price)
+async def set_limit_price(message: Message, state: FSMContext):
+    try:
+        price = float(message.text)
+        if price <= 0:
+            await message.answer("Цена должна быть положительным числом. Попробуйте снова.", reply_markup=inline_markup.cancel)
+            return
+    except ValueError:
+        await message.answer("Некорректный формат цены. Введите число.", reply_markup=inline_markup.cancel)
+        return
+
+    await state.update_data(entry_order_type='Limit', limit_price=price)
+    data = await state.get_data()
+
+    await rq.update_entry_order_type(message.from_user.id, 'Limit')
+    await rq.update_limit_price(message.from_user.id, price)
+
+    await message.answer(f"Цена лимитного ордера установлена: {price}", reply_markup=inline_markup.start_trading_markup)
+    await state.clear()
+
+
+
 @router_functions_bybit_trade.callback_query(F.data == "clb_my_deals")
 async def show_my_trades_callback(callback: CallbackQuery):
     tg_id = callback.from_user.id
@@ -227,7 +276,8 @@ async def start_trading_process(callback: CallbackQuery, state: FSMContext):
     try:
         positions_resp = client.get_positions(category='linear', symbol=symbol)
         positions = positions_resp.get('result', {}).get('list', [])
-    except Exception:
+    except Exception as e:
+        logger.error(f"Ошибка при получении позиций: {e}")
         positions = []
 
     for pos in positions:
@@ -253,7 +303,10 @@ async def start_trading_process(callback: CallbackQuery, state: FSMContext):
     await message.answer("Начинаю торговлю с использованием текущих настроек...")
 
     # Открываем позицию (вызывает Futures.open_position)
-    await open_position(tg_id, message, side=side, margin_mode=margin_mode)
+    success = await open_position(tg_id, message, side=side, margin_mode=margin_mode)
+    if not success:
+        await message.answer('⚠️ Ошибка при совершении сделки', reply_markup=inline_markup.back_to_main)
+        return
 
 
     # Проверяем таймер и информируем пользователя
@@ -271,3 +324,9 @@ async def start_trading_process(callback: CallbackQuery, state: FSMContext):
 
     await callback.answer()
 
+
+@router_functions_bybit_trade.callback_query(F.data == "clb_cancel")
+async def cancel(callback: CallbackQuery, state: FSMContext):
+    await state.clear()
+    await callback.message.answer("Отменено!", reply_markup=inline_markup.back_to_main)
+    await callback.answer()