import os import logging import httpx from aiogram import Bot, Dispatcher, types from aiogram.filters import Command, CommandObject from aiogram.client.default import DefaultBotProperties from aiogram.enums import ParseMode # Настройки TOKEN = os.getenv("TELEGRAM_TOKEN") API_URL = os.getenv("VNDB_API_URL", "https://api.vndb.org/kana") logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) bot = Bot( token=TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML) ) dp = Dispatcher() async def fetch_vndb(endpoint: str, filters: list, fields: str): payload = {"filters": filters, "fields": fields} async with httpx.AsyncClient(timeout=10.0) as client: try: response = await client.post(f"{API_URL}/{endpoint}", json=payload) if response.status_code == 200: return response.json().get("results", []) logger.error(f"VNDB API Error {response.status_code}: {response.text}") return None except Exception as e: logger.error(f"Request failed: {e}") return None @dp.message(Command("start", "help")) async def cmd_help(message: types.Message): help_text = ( "🤖 VNDB Bot\n\n" "Поиск:\n" "/search <name>\n" "/char <name>\n" "/release <name>\n\n" "Инфо по ID:\n" "/vn <id> (v17)\n" "/char_id <id> (c1)\n" "/rel_id <id> (r1)" ) await message.answer(help_text) # --- SEARCH HANDLERS --- @dp.message(Command("search")) async def search_vn(message: types.Message, command: CommandObject): if not command.args: return await message.answer("Пример: /search Steins;Gate") res = await fetch_vndb("vn", ["search", "=", command.args], "title") if not res: return await message.answer("❌ Ничего не найдено.") out = ["🔍 VN Results:"] for i in res[:10]: out.append(f"• {i['title']} ({i['id']})") await message.answer("\n".join(out)) @dp.message(Command("char")) async def search_char(message: types.Message, command: CommandObject): if not command.args: return await message.answer("Пример: /char Kurisu") res = await fetch_vndb("character", ["search", "=", command.args], "name") if not res: return await message.answer("❌ Персонаж не найден.") out = ["👤 Characters:"] for i in res[:10]: out.append(f"• {i['name']} ({i['id']})") await message.answer("\n".join(out)) @dp.message(Command("release")) async def search_release(message: types.Message, command: CommandObject): if not command.args: return await message.answer("Пример: /release Chaos;Head") res = await fetch_vndb("release", ["search", "=", command.args], "title") if not res: return await message.answer("❌ Релиз не найден.") out = ["💿 Releases:"] for i in res[:10]: out.append(f"• {i['title']} ({i['id']})") await message.answer("\n".join(out)) # --- DETAIL HANDLERS --- @dp.message(Command("vn")) async def detail_vn(message: types.Message, command: CommandObject): if not command.args: return await message.answer("Введите ID (v17)") res = await fetch_vndb("vn", ["id", "=", command.args], "id, title, original, released, rating, votecount") if not res: return await message.answer("❌ VN не найдена.") v = res[0] rating = f"{v['rating']/10} ⭐" if v.get('rating') else "N/A" text = (f"📖 {v['title']}\n" f"Original: {v.get('original', 'N/A')}\n" f"Released: {v.get('released', 'N/A')}\n" f"Rating: {rating} ({v.get('votecount', 0)} votes)\n" f"https://vndb.org/{v['id']}") await message.answer(text) @dp.message(Command("char_id")) async def detail_char(message: types.Message, command: CommandObject): if not command.args: return await message.answer("Введите ID (c1)") res = await fetch_vndb("character", ["id", "=", command.args], "id, name, original") if not res: return await message.answer("❌ Персонаж не найден.") c = res[0] text = (f"👤 {c['name']}\n" f"Original: {c.get('original', 'N/A')}\n" f"https://vndb.org/{c['id']}") await message.answer(text) @dp.message(Command("rel_id")) async def detail_rel(message: types.Message, command: CommandObject): if not command.args: return await message.answer("Введите ID (r1)") res = await fetch_vndb("release", ["id", "=", command.args], "id, title, released") if not res: return await message.answer("❌ Релиз не найден.") r = res[0] text = (f"💿 {r['title']}\n" f"Released: {r.get('released', 'N/A')}\n" f"https://vndb.org/{r['id']}") await message.answer(text) if __name__ == "__main__": dp.run_polling(bot)