Fixed martingale step

BybitBot_API.py

@@ -1,7 +1,7 @@
 import asyncio
 import logging.config
 from aiogram import Bot, Dispatcher
-
+from aiogram.fsm.storage.redis import RedisStorage
 from app.services.Bybit.functions.bybit_ws import get_or_create_event_loop, set_event_loop
 from app.telegram.database.models import async_main
 from app.telegram.handlers.handlers import router
@@ -17,8 +17,9 @@ from config import TOKEN_TG_BOT_1
 logging.config.dictConfig(LOGGING_CONFIG)
 logger = logging.getLogger("main")
 
+storage = RedisStorage.from_url("redis://localhost:6379/0")
 bot = Bot(token=TOKEN_TG_BOT_1)
-dp = Dispatcher()
+dp = Dispatcher(storage=storage)
 
 
 async def main() -> None:
@@ -37,7 +38,10 @@ async def main() -> None:
     dp.include_router(router_register_bybit_api)
     dp.include_router(router_functions_bybit_trade)
 
-    await dp.start_polling(bot)
+    try:
+        await dp.start_polling(bot)
+    except asyncio.CancelledError:
+        logger.info("Bot is off")
 
 
 if __name__ == '__main__':

app/services/Bybit/functions/Futures.py

@@ -262,6 +262,9 @@ async def handle_execution_message(message, msg):
                     float(martingale_factor) ** current_martingale_step
             )
             await rq.update_martingale_step(tg_id, current_martingale)
+            await message.answer(
+                f"❗️ Сделка закрылась в минус, открываю новую сделку с увеличенной ставкой.\n"
+            )
             await open_position(
                 tg_id,
                 message,
@@ -367,6 +370,10 @@ async def open_position(
 
         balance = await balance_g.get_balance(tg_id, message)
         if safe_float(balance) < total_budget:
+            logger.error(
+                f"Недостаточно средств для серии из {max_martingale_steps} шагов с текущими параметрами. "
+                f"Требуемый бюджет: {total_budget:.2f} USDT, доступно: {balance} USDT."
+            )
             await message.answer(
                 f"Недостаточно средств для серии из {max_martingale_steps} шагов с текущими параметрами. "
                 f"Требуемый бюджет: {total_budget:.2f} USDT, доступно: {balance} USDT.",
@@ -387,7 +394,7 @@ async def open_position(
             await error_max_risk(message)
             return
 
-        if max_martingale_steps == current_martingale:
+        if max_martingale_steps < current_martingale:
             await error_max_step(message)
             return
 
@@ -431,6 +438,11 @@ async def open_position(
 
         order_value = float(quantity) * price_for_calc
         if order_value < min_order_value:
+            logger.error(
+                f"Сумма ордера слишком мала: {order_value:.2f} USDT. "
+                f"Минимум для торговли — {min_order_value} USDT. "
+                f"Пожалуйста, увеличьте количество позиций."
+            )
             await message.answer(
                 f"Сумма ордера слишком мала: {order_value:.2f} USDT. "
                 f"Минимум для торговли — {min_order_value} USDT. "
@@ -581,6 +593,7 @@ async def open_position(
             await message.answer("Недостаточно средств для нового ордера",
                                  reply_markup=inline_markup.back_to_main)
         else:
+            logger.error("Ошибка при совершении сделки: %s", e)
             await message.answer(
                 "Недостаточно средств для размещения нового ордера с заданным количеством и плечом.",
                 reply_markup=inline_markup.back_to_main,
@@ -593,37 +606,6 @@ async def open_position(
         )
 
 
-async def trading_cycle(tg_id, message, side, margin_mode, symbol, starting_quantity):
-    """
-    Цикл торговой логики с учётом таймера пользователя.
-    """
-    try:
-        timer_data = await rq.get_user_timer(tg_id)
-        timer_min = 0
-        if isinstance(timer_data, dict):
-            timer_min = timer_data.get("timer_minutes") or timer_data.get("timer") or 0
-        else:
-            timer_min = timer_data or 0
-
-        timer_sec = timer_min * 60
-
-        if timer_sec > 0:
-            await asyncio.sleep(timer_sec)
-
-        await open_position(
-            tg_id,
-            message,
-            side=side,
-            margin_mode=margin_mode,
-            symbol=symbol,
-            quantity=starting_quantity,
-        )
-    except asyncio.CancelledError:
-        logger.info(
-            f"Торговый цикл для пользователя {tg_id} был отменён.", exc_info=True
-        )
-
-
 async def set_take_profit_stop_loss(
         tg_id: int,
         message,
@@ -643,7 +625,7 @@ async def set_take_profit_stop_loss(
             client.set_tp_sl_mode(symbol=symbol, category="linear", tpSlMode=tpsl_mode)
         except exceptions.InvalidRequestError as e:
             if "same tp sl mode" in str(e).lower():
-                logger.info(f"Режим TP/SL уже установлен для {symbol}")
+                logger.info("Режим TP/SL уже установлен для %s - пропускаем", symbol)
             else:
                 raise
 

app/services/Bybit/functions/functions.py

@@ -9,7 +9,7 @@ from logger_helper.logger_helper import LOGGING_CONFIG
 from app.services.Bybit.functions.Futures import (close_user_trade, set_take_profit_stop_loss, \
                                                   get_active_positions_by_symbol, get_active_orders_by_symbol,
                                                   get_active_positions, get_active_orders, cancel_all_tp_sl_orders,
-                                                  trading_cycle, open_position, close_trade_after_delay, safe_float,
+                                                  open_position, close_trade_after_delay, safe_float,
                                                   )
 from app.services.Bybit.functions.balance import get_balance
 import app.telegram.Keyboards.inline_keyboards as inline_markup
@@ -29,6 +29,8 @@ logger = logging.getLogger("functions")
 
 router_functions_bybit_trade = Router()
 
+user_trade_tasks = {}
+
 
 @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) -> None:
@@ -63,7 +65,6 @@ async def start_bybit_trade_message(message: Message) -> None:
     """
     balance = await get_balance(message.from_user.id, message)
     tg_id = message.from_user.id
-    await run_ws_for_user(tg_id, message)
 
     if balance:
         await run_ws_for_user(tg_id, message)
@@ -229,14 +230,41 @@ async def start_trading_process(callback: CallbackQuery) -> None:
         timer_minute = timer_data or 0
 
     if timer_minute > 0:
-        await message.answer(f"Торговля начнётся через {timer_minute} мин.")
-        await trading_cycle(tg_id, message, side=side, margin_mode=margin_mode, symbol=symbol,
-                            starting_quantity=starting_quantity)
-        await rq.update_user_timer(tg_id, minutes=0)
+        await message.answer(f"Торговля начнётся через {timer_minute} мин.", reply_markup=inline_markup.cancel_start)
+
+        async def delay_start():
+            try:
+                await asyncio.sleep(timer_minute * 60)
+                await open_position(tg_id, message, side, margin_mode, symbol=symbol, quantity=starting_quantity)
+                await rq.update_user_timer(tg_id, minutes=0)
+            except asyncio.exceptions.CancelledError:
+                logger.exception(f"Торговый цикл для пользователя {tg_id} был отменён.")
+                raise
+
+        task = asyncio.create_task(delay_start())
+        user_trade_tasks[tg_id] = task
     else:
         await open_position(tg_id, message, side, margin_mode, symbol=symbol, quantity=starting_quantity)
 
 
+@router_functions_bybit_trade.callback_query(F.data == "clb_cancel_start")
+async def cancel_start_trading(callback: CallbackQuery):
+    tg_id = callback.from_user.id
+    task = user_trade_tasks.get(tg_id)
+    if task and not task.done():
+        task.cancel()
+        try:
+            await task
+        except asyncio.CancelledError:
+            pass
+        user_trade_tasks.pop(tg_id, None)
+        await rq.update_user_timer(tg_id, minutes=0)
+        await callback.message.answer("Запуск торговли отменён.", reply_markup=inline_markup.back_to_main)
+        await callback.message.edit_reply_markup(reply_markup=None)
+    else:
+        await callback.answer("Нет запланированной задачи запуска.", show_alert=True)
+
+
 @router_functions_bybit_trade.callback_query(F.data == "clb_my_deals")
 async def show_my_trades(callback: CallbackQuery) -> None:
     """
@@ -443,7 +471,8 @@ async def process_close_delay(message: Message, state: FSMContext) -> None:
     symbol = data.get("symbol")
 
     delay = delay_minutes * 60
-    await message.answer(f"Закрытие сделки {symbol} запланировано через {delay_minutes} мин.", reply_markup=inline_markup.back_to_main)
+    await message.answer(f"Закрытие сделки {symbol} запланировано через {delay_minutes} мин.",
+                         reply_markup=inline_markup.back_to_main)
     await close_trade_after_delay(message.from_user.id, message, symbol, delay)
     await state.clear()
 
@@ -454,7 +483,7 @@ async def reset_martingale(callback: CallbackQuery) -> None:
     Сбрасывает шаги мартингейла пользователя.
     """
     tg_id = callback.from_user.id
-    await rq.update_martingale_step(tg_id, 0)
+    await rq.update_martingale_step(tg_id, 1)
     await callback.answer("Сброс шагов выполнен.")
     await main_settings_message(tg_id, callback.message)
 
@@ -511,7 +540,8 @@ async def process_stop_delay(message: Message, state: FSMContext):
     tg_id = message.from_user.id
     delay_seconds = delay_minutes * 60
 
-    await message.answer(f"Торговля будет остановлена через {delay_minutes} минут.", reply_markup=inline_markup.back_to_main)
+    await message.answer(f"Торговля будет остановлена через {delay_minutes} минут.",
+                         reply_markup=inline_markup.back_to_main)
     await asyncio.sleep(delay_seconds)
     await rq.update_trigger(tg_id, "Ручной")
     await message.answer("Автоматическая торговля остановлена.", reply_markup=inline_markup.back_to_main)

app/telegram/Keyboards/inline_keyboards.py

@@ -9,6 +9,10 @@ settings_markup = InlineKeyboardMarkup(inline_keyboard=[
     [InlineKeyboardButton(text="Запуск", callback_data='clb_start_trading')]
 ])
 
+cancel_start = InlineKeyboardMarkup(inline_keyboard=[
+    [InlineKeyboardButton(text="Отменить запуск", callback_data="clb_cancel_start")]
+])
+
 back_btn_list_settings = [InlineKeyboardButton(text="Назад",
                                                callback_data='clb_back_to_special_settings_message')]  # Кнопка для возврата к списку каталога настроек
 back_btn_list_settings_markup = InlineKeyboardMarkup(inline_keyboard=[[InlineKeyboardButton(text="Назад",
@@ -46,10 +50,8 @@ trading_markup = InlineKeyboardMarkup(inline_keyboard=[
 ])
 
 start_trading_markup = InlineKeyboardMarkup(inline_keyboard=[
-    [InlineKeyboardButton(text="На главную", callback_data='back_to_main')],
     [InlineKeyboardButton(text="Начать торговлю", callback_data="clb_start_chatbot_trading")],
-    [InlineKeyboardButton(text="Установить таймер", callback_data="clb_set_timer")],
-
+    [InlineKeyboardButton(text="На главную", callback_data='back_to_main')],
 ])
 
 cancel = InlineKeyboardMarkup(inline_keyboard=[
@@ -196,6 +198,7 @@ def create_close_limit_markup(symbol: str) -> InlineKeyboardMarkup:
 
 timer_markup = InlineKeyboardMarkup(inline_keyboard=[
     [InlineKeyboardButton(text="Установить таймер", callback_data="clb_set_timer")],
+    [InlineKeyboardButton(text="Удалить таймер", callback_data="clb_delete_timer")],
     back_btn_to_main
 ])
 

app/telegram/database/models.py

@@ -149,7 +149,7 @@ class User_Main_Settings(Base):
     size_leverage = mapped_column(Integer(), default=1)
     starting_quantity = mapped_column(Integer(), default=1)
     martingale_factor = mapped_column(Integer(), default=1)
-    martingale_step = mapped_column(Integer(), default=0)
+    martingale_step = mapped_column(Integer(), default=1)
     maximal_quantity = mapped_column(Integer(), default=10)
     entry_order_type = mapped_column(String(10), default='Market')
     limit_order_price = mapped_column(Numeric(18, 15), nullable=True)

app/telegram/functions/condition_settings/settings.py

@@ -29,7 +29,7 @@ async def main_settings_message(id, message):
     text = f""" <b>Условия запуска</b>
 
 <b>- Режим торговли:</b>  {trigger}
-<b>- Таймер: </b> установить таймер / остановить таймер  
+<b>- Таймер: </b> установить таймер / удалить таймер
 """
     await message.answer(text=text, parse_mode='html', reply_markup=inline_markup.condition_settings_markup)
 
@@ -68,7 +68,7 @@ async def timer_message(id, message: Message, state: FSMContext):
         return
 
     await message.answer(
-        f"Таймер: {timer_info['timer_minutes']} мин\n",
+        f"Таймер установлен на: {timer_info['timer_minutes']} мин\n",
         reply_markup=inline_markup.timer_markup
     )
 
@@ -76,7 +76,7 @@ async def timer_message(id, message: Message, state: FSMContext):
 @condition_settings_router.callback_query(F.data == "clb_set_timer")
 async def set_timer_callback(callback: CallbackQuery, state: FSMContext):
     await state.set_state(condition_settings.timer)  # состояние для ввода времени
-    await callback.message.answer("Введите время работы в минутах (например, 60):")
+    await callback.message.answer("Введите время работы в минутах (например, 60):", reply_markup=inline_markup.cancel)
     await callback.answer()
 
 
@@ -89,14 +89,21 @@ async def process_timer_input(message: Message, state: FSMContext):
             return
 
         await rq.update_user_timer(message.from_user.id, minutes)
-        await message.answer(f"Таймер установлен на {minutes} минут.\nНажмите кнопку 'Начать торговлю' для запуска.",
-                             reply_markup=inline_markup.start_trading_markup)
-
+        logger.info("Timer set for user %s: %s minutes", message.from_user.id, minutes)
+        await timer_message(message.from_user.id, message, state)
         await state.clear()
     except ValueError:
         await message.reply("Пожалуйста, введите корректное число.")
 
 
+@condition_settings_router.callback_query(F.data == "clb_delete_timer")
+async def delete_timer_callback(callback: CallbackQuery, state: FSMContext):
+    await state.clear()
+    await rq.update_user_timer(callback.from_user.id, 0)
+    logger.info("Timer deleted for user %s", callback.from_user.id)
+    await timer_message(callback.from_user.id, callback.message, state)
+    await callback.answer()
+
 
 async def filter_volatility_message(message, state):
     text = '''Фильтр волатильности

app/telegram/functions/main_settings/settings.py

@@ -67,7 +67,7 @@ async def main_settings_message(id, message):
     <b>- Размер кредитного плеча:</b> х{data['size_leverage']}
     <b>- Начальная ставка:</b> {data['starting_quantity']}
     <b>- Коэффициент мартингейла:</b> {data['martingale_factor']}
-    <b>- Количество ставок в серии:</b> {data['martingale_step']}
+    <b>- Текущий шаг:</b> {data['martingale_step']}
     <b>- Максимальное количество ставок в серии:</b> {data['maximal_quantity']}   
     
     <b>- Требуемый бюджет:</b> {total_budget:.2f} USDT

logger_helper/logger_helper.py

@@ -80,7 +80,7 @@ LOGGING_CONFIG = {
             "level": "DEBUG",
             "propagate": False,
         },
-        "conditions_settings": {
+        "condition_settings": {
             "handlers": ["console", "timed_rotating_file"],
             "level": "DEBUG",
             "propagate": False,

requirements.txt

@@ -6,22 +6,40 @@ aiosignal==1.4.0
 aiosqlite==0.21.0
 annotated-types==0.7.0
 attrs==25.3.0
+black==25.1.0
 certifi==2025.8.3
 charset-normalizer==3.4.3
+click==8.2.1
+colorama==0.4.6
 dotenv==0.9.9
+flake8==7.3.0
+flake8-bugbear==24.12.12
+flake8-pie==0.16.0
 frozenlist==1.7.0
 greenlet==3.2.4
 idna==3.10
+isort==6.0.1
 magic-filter==1.0.12
+mando==0.7.1
+mccabe==0.7.0
 multidict==6.6.4
+mypy_extensions==1.1.0
 nest-asyncio==1.6.0
+packaging==25.0
+pathspec==0.12.1
+platformdirs==4.4.0
 propcache==0.3.2
 pybit==5.11.0
+pycodestyle==2.14.0
 pycryptodome==3.23.0
 pydantic==2.11.7
 pydantic_core==2.33.2
+pyflakes==3.4.0
 python-dotenv==1.1.1
+radon==6.0.1
+redis==6.4.0
 requests==2.32.5
+six==1.17.0
 SQLAlchemy==2.0.43
 typing-inspection==0.4.1
 typing_extensions==4.14.1