version 1 STCS

app/services/Bybit/functions/Add_Bybit_API.py

@@ -0,0 +1,73 @@
+from aiogram import F, Router
+
+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_register_bybit_api = Router()
+
+class state_reg_bybit_api(StatesGroup):
+    api_key = State()
+    secret_key = State()
+
+@router_register_bybit_api.callback_query(F.data == 'clb_new_user_connect_bybit_api_message')
+async def info_for_bybit_api_message(callback: CallbackQuery):
+    text = '''<b>Подключение Bybit аккаунта</b>
+    
+<b>1. Зарегистрируйтесь или войдите в свой аккаунт на Bybit (https://www.bybit.com/).</b>
+<b>2. В личном кабинете выберите раздел API. </b>  
+<b>3. Создание нового API ключа</b>  
+   - Нажмите кнопку Create New Key (Создать новый ключ).
+   - Выберите системно-сгенерированный ключ.
+   - Укажите название API ключа (любое).  
+   - Выберите права доступа для торговли (Trade).  
+   - Можно ограничить доступ по IP для безопасности.
+<b>4. Подтверждение создания</b>  
+   - Подтвердите создание ключа.
+   - Отправьте чат-роботу.
+
+<b>Важно: сохраните отдельно API Key и Secret Key в надежном месте. Secret ключ отображается только один раз. </b>            
+    '''
+
+    await callback.message.answer(text=text, parse_mode='html', reply_markup=inline_markup.connect_bybit_api_markup)
+
+    await callback.answer()
+
+@router_register_bybit_api.callback_query(F.data == 'clb_new_user_connect_bybit_api')
+async def add_api_key_message(callback: CallbackQuery, state: FSMContext):   
+    await state.set_state(state_reg_bybit_api.api_key)
+
+    text = 'Отправьте KEY_API ниже: '
+
+    await callback.message.answer(text=text)
+
+@router_register_bybit_api.message(state_reg_bybit_api.api_key)
+async def add_api_key_and_message_for_secret_key(message: Message, state: FSMContext):
+    await state.update_data(api_key = message.text)
+    
+    text = 'Отправьте SECRET_KEY ниже'
+
+    await message.answer(text=text)
+
+    await state.set_state(state_reg_bybit_api.secret_key)
+
+@router_register_bybit_api.message(state_reg_bybit_api.secret_key)
+async def add_secret_key(message: Message, state: FSMContext):
+    await state.update_data(secret_key = message.text)
+
+    data = await state.get_data()
+
+    await rq.update_api_key(message.from_user.id, data['api_key'])
+    await rq.update_secret_key(message.from_user.id, data['secret_key'])
+    await rq.set_new_user_symbol(message.from_user.id)
+    
+    await state.clear()
+
+    await message.answer('Данные добавлены, нажмите на профиль и начните торговлю!')
+
+