Added commission_fee and order_type functions

This commit is contained in:
algizn97
2025-08-21 13:39:43 +05:00
parent 7f53caaac6
commit c8b0dad7c2

View File

@@ -1,4 +1,6 @@
import logging
from typing import Any
logger = logging.getLogger(__name__)
from app.telegram.database.models import async_session
@@ -181,11 +183,13 @@ async def get_user_risk_management_settings(tg_id):
price_profit = await session.scalar(select(URMS.price_profit).where(URMS.tg_id == tg_id))
price_loss = await session.scalar(select(URMS.price_loss).where(URMS.tg_id == tg_id))
max_risk_deal = await session.scalar(select(URMS.max_risk_deal).where(URMS.tg_id == tg_id))
commission_fee = await session.scalar(select(URMS.commission_fee).where(URMS.tg_id == tg_id))
data = {
'price_profit': price_profit,
'price_loss': price_loss,
'max_risk_deal': max_risk_deal
'max_risk_deal': max_risk_deal,
'commission_fee': commission_fee,
}
return data
@@ -273,3 +277,29 @@ async def update_max_risk_deal(tg_id, num):
await session.execute(update(URMS).where(URMS.tg_id == tg_id).values(max_risk_deal = num))
await session.commit()
async def update_entry_order_type(tg_id, order_type):
async with async_session() as session:
await session.execute(
update(UMS)
.where(UMS.tg_id == tg_id)
.values(entry_order_type=order_type)
)
await session.commit()
async def get_entry_order_type(tg_id: object) -> str | None | Any:
async with async_session() as session:
order_type = await session.scalar(
select(UMS.entry_order_type).where(UMS.tg_id == tg_id)
)
# Если в базе не установлен тип — возвращаем значение по умолчанию
return order_type or 'Market'
async def update_commission_fee(tg_id, num):
async with async_session() as session:
await session.execute(update(URMS).where(URMS.tg_id == tg_id).values(commission_fee = num))
await session.commit()