Fixed

app/telegram/database/models.py

@@ -148,12 +148,14 @@ class User_Main_Settings(Base):
     switch_state = mapped_column(String(10), default='По направлению')
     size_leverage = mapped_column(Integer(), default=1)
     starting_quantity = mapped_column(Integer(), default=1)
+    base_quantity = mapped_column(Integer(), default=1)
     martingale_factor = mapped_column(Integer(), default=1)
     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)
     last_side = mapped_column(String(10), default='Buy')
+    trading_start_stop = mapped_column(Integer(), default=0)
 
 
 class User_Risk_Management_Settings(Base):

app/telegram/database/requests.py

@@ -320,6 +320,8 @@ async def get_user_main_settings(tg_id):
                 'limit_order_price': user.limit_order_price,
                 'martingale_step': user.martingale_step,
                 'last_side': user.last_side,
+                'trading_start_stop': user.trading_start_stop,
+                'base_quantity': user.base_quantity,
             }
             return data
 
@@ -368,15 +370,23 @@ async def update_size_leverange(tg_id, num):
 
 
 async def update_starting_quantity(tg_id, num):
-    """Обновить размер левеража пользователя."""
+    """Обновить размер начальной ставки пользователя."""
     async with async_session() as session:
         await session.execute(update(UMS).where(UMS.tg_id == tg_id).values(starting_quantity=num))
 
         await session.commit()
 
 
+async def update_base_quantity(tg_id, num):
+    """Обновить размер следующей ставки пользователя."""
+    async with async_session() as session:
+        await session.execute(update(UMS).where(UMS.tg_id == tg_id).values(base_quantity=num))
+
+        await session.commit()
+
+
 async def update_martingale_factor(tg_id, num):
-    """Обновить размер левеража пользователя."""
+    """Обновить шаг мартингейла пользователя."""
     async with async_session() as session:
         await session.execute(update(UMS).where(UMS.tg_id == tg_id).values(martingale_factor=num))
 
@@ -384,7 +394,7 @@ async def update_martingale_factor(tg_id, num):
 
 
 async def update_maximal_quantity(tg_id, num):
-    """Обновить размер левеража пользователя."""
+    """Обновить размер максимальной ставки пользователя."""
     async with async_session() as session:
         await session.execute(update(UMS).where(UMS.tg_id == tg_id).values(maximal_quantity=num))
 
@@ -582,4 +592,4 @@ async def set_last_series_info(tg_id: int, last_side: str):
                     last_side=last_side,
                 )
                 session.add(new_entry)
-        await session.commit()
+        await session.commit()