mirror of
https://github.com/Llloooggg/TextSouls.git
synced 2026-03-06 04:26:23 +03:00
106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
from aiogram import Router
|
||
from aiogram.filters import Command
|
||
from aiogram.fsm.state import StatesGroup, State
|
||
|
||
from textsouls.common import backend
|
||
from textsouls.keyboards.common import row_kb
|
||
|
||
|
||
router = Router()
|
||
|
||
available_races = {}
|
||
available_classes = {}
|
||
|
||
|
||
class CharachterCreation(StatesGroup):
|
||
choosing_race = State()
|
||
choosing_class = State()
|
||
choosing_name = State()
|
||
creation_ending = State()
|
||
|
||
|
||
@router.message(Command(commands=["createchar"]))
|
||
async def character_creation(message, state):
|
||
|
||
global available_races
|
||
result = backend.get("/character_races")
|
||
if not result["error"] and result["response"].status_code == 200:
|
||
for race in result["response"].json():
|
||
available_races[race["name"]] = race["id"]
|
||
|
||
await message.answer(
|
||
text="Выбери расу:",
|
||
reply_markup=row_kb(available_races.keys()),
|
||
)
|
||
await state.set_state(CharachterCreation.choosing_class)
|
||
|
||
|
||
@router.message(CharachterCreation.choosing_class)
|
||
async def race_chosen(message, state):
|
||
|
||
if not available_races.get(message.text):
|
||
await message.answer(
|
||
text="Хм, такую не знаю.\nПопробуй из вариантов ниже:",
|
||
reply_markup=row_kb(available_races),
|
||
)
|
||
else:
|
||
await state.update_data(chosen_race=message.text)
|
||
|
||
global available_classes
|
||
result = backend.get("/character_classes")
|
||
if not result["error"] and result["response"].status_code == 200:
|
||
for char_class in result["response"].json():
|
||
available_classes[char_class["name"]] = char_class["id"]
|
||
await message.answer(
|
||
text="А теперь выбери класс:",
|
||
reply_markup=row_kb(available_classes.keys()),
|
||
)
|
||
await state.set_state(CharachterCreation.choosing_name)
|
||
|
||
|
||
@router.message(CharachterCreation.choosing_name)
|
||
async def class_chosen(message, state):
|
||
|
||
if not available_classes.get(message.text):
|
||
await message.answer(
|
||
text="Хм, такой не знаю.\nПопробуй из вариантов ниже:",
|
||
reply_markup=row_kb(available_classes),
|
||
)
|
||
else:
|
||
await state.update_data(chosen_class=message.text)
|
||
|
||
await message.answer(
|
||
text=f"Ну, неплохо.\nА зовут-то тебя как, {message.text.lower()}?",
|
||
reply_markup="",
|
||
)
|
||
await state.set_state(CharachterCreation.creation_ending)
|
||
|
||
|
||
@router.message(CharachterCreation.creation_ending)
|
||
async def name_chosen(message, state):
|
||
|
||
name = message.text
|
||
|
||
user_elections = await state.get_data()
|
||
|
||
new_character = {
|
||
"owner": message.from_user.id,
|
||
"name": name,
|
||
"character_race": available_races[user_elections["chosen_race"]],
|
||
"character_class": available_classes[user_elections["chosen_class"]],
|
||
}
|
||
|
||
result = backend.post("/characters", new_character)
|
||
if not result["error"] and result["response"].status_code == 200:
|
||
await message.answer(
|
||
text="Так и запишем.\nТеперь постарайся не умереть",
|
||
reply_markup="",
|
||
)
|
||
await state.clear()
|
||
else:
|
||
await message.answer(
|
||
text="Что-то пошло не так!\nДавай-ка по-новой",
|
||
)
|
||
await state.set_state(CharachterCreation.choosing_race)
|
||
await character_creation(message, state)
|