157 lines
7.7 KiB
Python
157 lines
7.7 KiB
Python
from aiogram import Router
|
||
import app.telegram.Keyboards.inline_keyboards as inline_markup
|
||
import app.telegram.Keyboards.reply_keyboards as reply_markup
|
||
|
||
import app.telegram.database.requests as rq
|
||
from aiogram.types import Message, CallbackQuery
|
||
|
||
# FSM - Механизм состояния
|
||
from aiogram.fsm.state import State, StatesGroup
|
||
|
||
router_risk_management_settings = Router()
|
||
|
||
class update_risk_management_settings(StatesGroup):
|
||
price_profit = State()
|
||
price_loss = State()
|
||
max_risk_deal = State()
|
||
commission_fee = State()
|
||
|
||
async def reg_new_user_default_risk_management_settings(id, message):
|
||
tg_id = id
|
||
|
||
await rq.set_new_user_default_risk_management_settings(tg_id)
|
||
|
||
async def main_settings_message(id, message, state):
|
||
data = await rq.get_user_risk_management_settings(id)
|
||
|
||
text = f"""<b>Риск менеджмент</b>,
|
||
|
||
<b>- Процент изменения цены для фиксации прибыли:</b> {data.get('price_profit', 0)}%
|
||
<b>- Процент изменения цены для фиксации убытков:</b> {data.get('price_loss', 0)}%
|
||
<b>- Максимальный риск на сделку (в % от баланса):</b> {data.get('max_risk_deal', 0)}%
|
||
<b>- Комиссия биржи для расчета процента фиксации прибыли:</b> {data.get('commission_fee', 0)}%
|
||
"""
|
||
await message.answer(text=text, parse_mode='html', reply_markup=inline_markup.risk_management_settings_markup)
|
||
|
||
async def price_profit_message(message, state):
|
||
await state.set_state(update_risk_management_settings.price_profit)
|
||
|
||
text = 'Введите число изменения цены для фиксации прибыли: '
|
||
|
||
await message.answer(text=text, parse_mode='html', reply_markup=None)
|
||
|
||
@router_risk_management_settings.message(update_risk_management_settings.price_profit)
|
||
async def state_price_profit(message: Message, state):
|
||
await state.update_data(price_profit = message.text)
|
||
|
||
data = await state.get_data()
|
||
data_settings = await rq.get_user_risk_management_settings(message.from_user.id)
|
||
|
||
if data['price_profit'].isdigit() and int(data['price_profit']) <= 100:
|
||
await message.answer(f"✅ Изменено: {data_settings['price_profit']}% → {data['price_profit']}%")
|
||
|
||
await rq.update_price_profit(message.from_user.id, data['price_profit'])
|
||
await main_settings_message(message.from_user.id, message, state)
|
||
|
||
await state.clear()
|
||
else:
|
||
await message.answer(f'⛔️ Ошибка: ваше значение ({data['price_profit']}%) или выше лимита (100) или вы вводите неверные символы')
|
||
|
||
await main_settings_message(message.from_user.id, message, state)
|
||
|
||
async def price_loss_message(message, state):
|
||
await state.set_state(update_risk_management_settings.price_loss)
|
||
|
||
text = 'Введите число изменения цены для фиксации убытков: '
|
||
|
||
await message.answer(text=text, parse_mode='html', reply_markup=None)
|
||
|
||
@router_risk_management_settings.message(update_risk_management_settings.price_loss)
|
||
async def state_price_loss(message: Message, state):
|
||
await state.update_data(price_loss = message.text)
|
||
|
||
data = await state.get_data()
|
||
data_settings = await rq.get_user_risk_management_settings(message.from_user.id)
|
||
|
||
if data['price_loss'].isdigit() and int(data['price_loss']) <= 100:
|
||
new_price_loss = int(data['price_loss'])
|
||
old_price_loss = int(data_settings.get('price_loss', 0))
|
||
|
||
current_price_profit = data_settings.get('price_profit')
|
||
# Пробуем перевести price_profit в число, если это возможно
|
||
try:
|
||
current_price_profit_num = int(current_price_profit)
|
||
except Exception:
|
||
current_price_profit_num = 0
|
||
|
||
# Флаг, если price_profit изначально равен 0 или совпадает со старым стоп-лоссом
|
||
should_update_profit = (current_price_profit_num == 0) or (current_price_profit_num == abs(old_price_loss))
|
||
|
||
# Обновляем стоп-лосс
|
||
await rq.update_price_loss(message.from_user.id, new_price_loss)
|
||
|
||
# Если нужно, меняем тейк-профит
|
||
if should_update_profit:
|
||
new_price_profit = abs(new_price_loss)
|
||
await rq.update_price_profit(message.from_user.id, new_price_profit)
|
||
await message.answer(f"✅ Стоп-лосс изменён: {old_price_loss}% → {new_price_loss}%\n"
|
||
f"Тейк-профит автоматически установлен в: {new_price_profit}%")
|
||
else:
|
||
await message.answer(f"✅ Стоп-лосс изменён: {old_price_loss}% → {new_price_loss}%")
|
||
|
||
await main_settings_message(message.from_user.id, message, state)
|
||
await state.clear()
|
||
else:
|
||
await message.answer(
|
||
f'⛔️ Ошибка: ваше значение ({data["price_loss"]}%) выше лимита (100) или содержит неверные символы')
|
||
await main_settings_message(message.from_user.id, message, state)
|
||
|
||
async def max_risk_deal_message(message, state):
|
||
await state.set_state(update_risk_management_settings.max_risk_deal)
|
||
|
||
text = 'Введите число (процент от баланса) для изменения максимального риска на сделку: '
|
||
|
||
await message.answer(text=text, parse_mode='html', reply_markup=None)
|
||
|
||
@router_risk_management_settings.message(update_risk_management_settings.max_risk_deal)
|
||
async def state_max_risk_deal(message: Message, state):
|
||
await state.update_data(max_risk_deal = message.text)
|
||
|
||
data = await state.get_data()
|
||
data_settings = await rq.get_user_risk_management_settings(message.from_user.id)
|
||
|
||
if data['max_risk_deal'].isdigit() and int(data['max_risk_deal']) <= 100:
|
||
await message.answer(f"✅ Изменено: {data_settings['max_risk_deal']}% → {data['max_risk_deal']}%")
|
||
|
||
await rq.update_max_risk_deal(message.from_user.id, data['max_risk_deal'])
|
||
await main_settings_message(message.from_user.id, message, state)
|
||
|
||
await state.clear()
|
||
else:
|
||
await message.answer(f'⛔️ Ошибка: ваше значение ({data['max_risk_deal']}%) или выше лимита (100) или вы вводите неверные символы')
|
||
|
||
await main_settings_message(message.from_user.id, message, state)
|
||
|
||
|
||
async def commission_fee_message(message, state):
|
||
await state.set_state(update_risk_management_settings.commission_fee)
|
||
await message.answer(text="Введите процент комиссии биржи (например, 0.1):", parse_mode='html', reply_markup=None)
|
||
|
||
@router_risk_management_settings.message(update_risk_management_settings.commission_fee)
|
||
async def state_commission_fee(message: Message, state):
|
||
await state.update_data(commission_fee=message.text)
|
||
data = await state.get_data()
|
||
data_settings = await rq.get_user_risk_management_settings(message.from_user.id)
|
||
|
||
try:
|
||
val = float(data['commission_fee'])
|
||
if val < 0 or val > 100:
|
||
raise ValueError()
|
||
except Exception:
|
||
await message.answer("⛔️ Ошибка: введите корректный процент комиссии от 0 до 100")
|
||
return await commission_fee_message(message, state)
|
||
|
||
await rq.update_commission_fee(message.from_user.id, val)
|
||
await message.answer(f"✅ Изменено: {data_settings['commission_fee']}% → {data['commission_fee']}%")
|
||
await main_settings_message(message.from_user.id, message, state)
|
||
await state.clear() |