211 lines
9.7 KiB
Python
211 lines
9.7 KiB
Python
from aiogram import F, Router
|
||
|
||
from app.services.Bybit.functions import Futures, func_min_qty
|
||
from app.services.Bybit.functions.Futures import open_market_order, open_limit_order
|
||
from app.services.Bybit.functions.balance import get_balance
|
||
import app.telegram.Keyboards.inline_keyboards as inline_markup
|
||
|
||
import app.telegram.database.requests as rq
|
||
from aiogram.types import Message, CallbackQuery
|
||
|
||
# FSM - Механизм состояния
|
||
from aiogram.fsm.state import State, StatesGroup
|
||
from aiogram.fsm.context import FSMContext
|
||
|
||
router_functions_bybit_trade = Router()
|
||
|
||
class state_update_symbol(StatesGroup):
|
||
symbol = State()
|
||
|
||
class state_update_entry_type(StatesGroup):
|
||
entry_type = State()
|
||
|
||
@router_functions_bybit_trade.callback_query(F.data.in_(['clb_start_trading', 'clb_back_to_main']))
|
||
async def clb_start_bybit_trade_message(callback: CallbackQuery, state: FSMContext):
|
||
api = await rq.get_bybit_api_key(callback.from_user.id)
|
||
secret = await rq.get_bybit_secret_key(callback.from_user.id)
|
||
balance = await get_balance(callback.from_user.id, callback.message)
|
||
|
||
if balance:
|
||
symbol = await rq.get_symbol(callback.from_user.id)
|
||
|
||
text = f'''💎 Торговля на Bybit
|
||
|
||
⚖️ Ваш баланс (USDT): {balance}
|
||
📊 Текущая торговая пара: {symbol}
|
||
|
||
Как начать торговлю?
|
||
|
||
1️⃣ Проверьте и тщательно настройте все параметры в вашем профиле.
|
||
2️⃣ Нажмите ниже кнопку 'Указать торговую пару' и введите торговую пару заглавными буквами, без лишних символов (например: BTCUSDT).
|
||
'''
|
||
await callback.message.edit_text(text=text, parse_mode='html', reply_markup=inline_markup.trading_markup)
|
||
|
||
async def start_bybit_trade_message(message, state):
|
||
api = await rq.get_bybit_api_key(message.from_user.id)
|
||
secret = await rq.get_bybit_secret_key(message.from_user.id)
|
||
balance = await get_balance(message.from_user.id, message)
|
||
|
||
if balance:
|
||
symbol = await rq.get_symbol(message.from_user.id)
|
||
|
||
text = f'''💎 Торговля на Bybit
|
||
|
||
⚖️ Ваш баланс (USDT): {balance}
|
||
📊 Текущая торговая пара: {symbol}
|
||
|
||
Как начать торговлю?
|
||
|
||
1️⃣ Проверьте и тщательно настройте все параметры в вашем профиле.
|
||
2️⃣ Нажмите ниже кнопку 'Указать торговую пару' и введите торговую пару заглавными буквами, без лишних символов (например: BTCUSDT).
|
||
'''
|
||
|
||
await message.answer(text=text, parse_mode='html', reply_markup=inline_markup.trading_markup)
|
||
|
||
@router_functions_bybit_trade.callback_query(F.data == 'clb_update_trading_pair')
|
||
async def update_symbol_for_trade_message(callback: CallbackQuery, state: FSMContext):
|
||
await state.set_state(state_update_symbol.symbol)
|
||
|
||
await callback.message.answer(text='Укажите торговую пару заглавными буквами без пробелов и лишних символов (пример: BTCUSDT): ')
|
||
|
||
@router_functions_bybit_trade.message(state_update_symbol.symbol)
|
||
async def update_symbol_for_trade(message: Message, state: FSMContext):
|
||
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 start_bybit_trade_message(message, state)
|
||
|
||
await state.clear()
|
||
|
||
|
||
@router_functions_bybit_trade.callback_query(F.data == 'clb_update_entry_type')
|
||
async def update_entry_type_message(callback: CallbackQuery, state: FSMContext):
|
||
await state.set_state(state_update_entry_type.entry_type)
|
||
await callback.message.answer("Выберите тип входа в позицию:", reply_markup=inline_markup.entry_order_type_markup)
|
||
await callback.answer()
|
||
|
||
|
||
@router_functions_bybit_trade.callback_query(lambda c: c.data and c.data.startswith('entry_order_type:'))
|
||
async def entry_order_type_callback(callback: CallbackQuery, state: FSMContext):
|
||
order_type = callback.data.split(':')[1]
|
||
|
||
if order_type not in ['Market', 'Limit']:
|
||
await callback.answer("Ошибка выбора", show_alert=True)
|
||
return
|
||
|
||
await state.update_data(entry_order_type=order_type)
|
||
await rq.update_entry_order_type(callback.from_user.id, order_type)
|
||
|
||
if order_type == 'Limit':
|
||
await callback.answer("Вы выбрали Limit. Введите цену для лимитного ордера.")
|
||
await callback.message.answer("Введите цену лимитного ордера:")
|
||
await state.update_data(awaiting_limit_price=True)
|
||
else:
|
||
await callback.answer("Вы выбрали Market. Нажмите кнопку ниже, чтобы открыть сделку.")
|
||
await callback.message.answer("Нажмите кнопку, чтобы открыть сделку.",
|
||
reply_markup=inline_markup.open_deal_markup)
|
||
await callback.message.delete()
|
||
|
||
|
||
@router_functions_bybit_trade.message()
|
||
async def process_limit_price_message(message: Message, state: FSMContext):
|
||
data = await state.get_data()
|
||
if not data.get('awaiting_limit_price'):
|
||
return
|
||
|
||
try:
|
||
price = float(message.text)
|
||
if price <= 0:
|
||
raise ValueError()
|
||
except ValueError:
|
||
await message.answer("Ошибка: введите корректное положительное число для цены.")
|
||
return
|
||
|
||
await state.update_data(limit_price=price, awaiting_limit_price=False)
|
||
|
||
await message.answer(f"Цена лимитного ордера установлена: {price}. Нажмите кнопку ниже, чтобы открыть сделку.",
|
||
reply_markup=inline_markup.open_deal_markup)
|
||
|
||
|
||
@router_functions_bybit_trade.callback_query(F.data == "clb_open_deal")
|
||
async def open_deal(callback: CallbackQuery, state: FSMContext):
|
||
data_main_stgs = await rq.get_user_main_settings(callback.from_user.id)
|
||
data = await state.get_data()
|
||
order_type = await rq.get_entry_order_type(callback.from_user.id)
|
||
api = await rq.get_bybit_api_key(callback.from_user.id)
|
||
secret = await rq.get_bybit_secret_key(callback.from_user.id)
|
||
qty = data_main_stgs['starting_quantity']
|
||
qty_min = await func_min_qty.get_min_qty(callback.from_user.id, callback.message)
|
||
|
||
if qty < qty_min:
|
||
await callback.message.edit_text(f"Количество вашей ставки ({qty}) меньше минимального количества ({qty_min}) для данной торговой пары")
|
||
await callback.answer()
|
||
return
|
||
|
||
if order_type == 'Market':
|
||
await open_market_order(callback.from_user.id, callback.message, api_key=api, secret_key=secret)
|
||
await rq.update_user_trades(callback.from_user.id, symbol=data.get('symbol'), side=order_type)
|
||
elif order_type == 'Limit':
|
||
price = data.get('limit_price')
|
||
if not price:
|
||
await callback.answer("Цена для лимитного ордера не задана. Введите сначала цену.")
|
||
return
|
||
await open_limit_order(callback.from_user.id, callback.message, price, api_key=api, secret_key=secret)
|
||
else:
|
||
await callback.answer("Неизвестный тип ордера.")
|
||
|
||
await callback.message.edit_reply_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
|
||
trades = await rq.get_user_trades(tg_id)
|
||
|
||
if not trades:
|
||
await callback.message.answer("У вас ещё нет сделок.")
|
||
await callback.answer()
|
||
return
|
||
|
||
grouped = {}
|
||
for trade in trades:
|
||
symbol = trade['symbol'] if isinstance(trade, dict) else trade.symbol
|
||
grouped.setdefault(symbol, []).append(trade)
|
||
|
||
text_response = "Ваши сделки по валютным парам:\n\n"
|
||
for symbol, trade_list in grouped.items():
|
||
text_response += f"<b>{symbol}</b>\n"
|
||
for t in trade_list:
|
||
side = t['side'] if isinstance(t, dict) else t.side
|
||
text_response += f" - {side}\n"
|
||
text_response += "\n"
|
||
|
||
await callback.message.answer(text_response, parse_mode='html')
|
||
await callback.answer()
|
||
|
||
|
||
# @router_functions_bybit_trade.callback_query(F.data == 'clb_open_deal')
|
||
# async def make_deal_bybit (callback: CallbackQuery):
|
||
# data_main_stgs = await rq.get_user_main_settings(callback.from_user.id)
|
||
#
|
||
# trade_mode = data_main_stgs['trading_mode']
|
||
# qty = data_main_stgs['starting_quantity']
|
||
# margin_mode = data_main_stgs['margin_type']
|
||
qty_min = await func_min_qty.get_min_qty(callback.from_user.id, callback.message)
|
||
#
|
||
# if qty < qty_min:
|
||
# await callback.message.edit_text(f"Количество вашей ставки ({qty}) меньше минимального количества ({qty_min}) для данной торговой пары")
|
||
# else:
|
||
# match trade_mode:
|
||
# case 'Long':
|
||
# await Futures.contract_long(callback.from_user.id, callback.message, margin_mode)
|
||
# case 'Short':
|
||
# await Futures.contract_short(callback.from_user.id, callback.message, margin_mode)
|
||
# case 'Switch':
|
||
# await callback.message.edit_text('Режим Switch пока недоступен')
|
||
# case 'Smart':
|
||
# await callback.message.edit_text('Режим Smart пока недоступен') |