refactor(bot): modularize bot code

This commit is contained in:
2026-07-22 18:44:44 +05:00
parent 9b8d388c3c
commit 04f17ff8e2
25 changed files with 1433 additions and 983 deletions
+1
View File
@@ -0,0 +1 @@
+118
View File
@@ -0,0 +1,118 @@
import aiohttp
import requests
from bot.config import ASF_PASSWORD, ASF_URL
IPC_BASE_URL = "http://127.0.0.1:1242"
def _headers() -> dict[str, str | None]:
return {"Authentication": ASF_PASSWORD}
def asf_get(path: str) -> requests.Response:
return requests.get(f"{IPC_BASE_URL}{path}", headers=_headers())
def asf_post(path: str, payload: dict | None = None) -> requests.Response:
return requests.post(f"{IPC_BASE_URL}{path}", headers=_headers(), json=payload)
def get_asf_status() -> requests.Response:
return requests.get(ASF_URL, headers=_headers()) # type: ignore[arg-type]
def get_all_bots() -> requests.Response:
return asf_get("/Api/Bot/ASF")
def get_bot(bot_name: str) -> requests.Response:
return asf_get(f"/Api/Bot/{bot_name}")
def get_plugins() -> requests.Response:
return asf_get("/Api/Plugins")
def send_command(command: str) -> requests.Response:
return asf_post("/Api/Command", {"Command": command})
def restart_asf() -> requests.Response:
return send_command("!restart")
async def get_bot_inventory(bot_name: str, appid: int, contextid: int) -> dict | None:
async with aiohttp.ClientSession() as session:
async with session.get(
f"{IPC_BASE_URL}/Api/Bot/{bot_name}/Inventory/{appid}/{contextid}",
headers=_headers(),
) as response:
if response.status != 200:
return None
data = await response.json()
if not data.get("Success"):
return None
return data.get("Result")
async def get_2fa_token(bot_name: str) -> str:
async with aiohttp.ClientSession() as session:
async with session.get(
f"{IPC_BASE_URL}/Api/Bot/{bot_name}/TwoFactorAuthentication/Token",
headers=_headers(),
) as response:
if response.status != 200:
return f"Ошибка HTTP {response.status}"
data = await response.json()
if not data.get("Success"):
return "Ошибка ASF"
try:
return data["Result"][bot_name]["Result"]
except Exception:
return "Не удалось получить код"
async def get_confirmations(bot_name: str) -> list:
async with aiohttp.ClientSession() as session:
async with session.get(
f"{IPC_BASE_URL}/Api/Bot/{bot_name}/TwoFactorAuthentication/Confirmations",
headers=_headers(),
) as response:
if response.status != 200:
return []
data = await response.json()
if not data.get("Success"):
return []
try:
return data["Result"][bot_name]["Result"]
except Exception:
return []
async def accept_confirmations(bot_name: str) -> int:
async with aiohttp.ClientSession() as session:
async with session.post(
f"{IPC_BASE_URL}/Api/Bot/{bot_name}/TwoFactorAuthentication/Confirmations/Accept",
headers=_headers(),
) as response:
return response.status
def save_bot_config(bot_name: str, config: dict) -> requests.Response:
return asf_post(f"/Api/Bot/{bot_name}", {"BotConfig": config})
async def redeem_key(bot_name: str, key: str) -> tuple[bool, str]:
async with aiohttp.ClientSession() as session:
async with session.post(
f"{IPC_BASE_URL}/Api/Command",
headers=_headers(),
json={"Command": f"!redeem {bot_name} {key}"},
) as response:
if response.status != 200:
return False, f"HTTP_ERROR_{response.status}"
data = await response.json()
if not data.get("Success"):
return False, "ASF_ERROR"
return True, str(data.get("Result", ""))