69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
# services/torn_api.py
|
|
import aiohttp
|
|
import json
|
|
from pathlib import Path
|
|
from config import TORN_API_KEY, ENEMY_FACTION_ID, YOUR_FACTION_ID
|
|
from .ffscouter import fetch_batch_stats
|
|
|
|
ENEMY_FILE = Path("data/enemy_faction.json")
|
|
FRIENDLY_FILE = Path("data/friendly_faction.json")
|
|
|
|
|
|
async def fetch_and_save_faction(faction_id: int, file_path: Path) -> bool:
|
|
"""
|
|
Fetches faction members from Torn, fetches their estimated BS from FFScouter,
|
|
and saves everything to a JSON file.
|
|
"""
|
|
url = f"https://api.torn.com/v2/faction/{faction_id}?selections=members&key={TORN_API_KEY}"
|
|
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(url) as resp:
|
|
if resp.status != 200:
|
|
print(f"Torn faction fetch error: {resp.status}")
|
|
return False
|
|
data = await resp.json()
|
|
|
|
members_list = data.get("members", [])
|
|
if not members_list:
|
|
return False
|
|
|
|
# Build list of IDs (Torn uses 'id', not 'player_id')
|
|
member_ids = [info.get("id") for info in members_list if "id" in info]
|
|
if not member_ids:
|
|
return False
|
|
|
|
# Fetch batch FFScouter stats
|
|
ff_data = await fetch_batch_stats(member_ids) # returns dict keyed by player_id
|
|
|
|
# Build final faction data
|
|
faction_data = []
|
|
for info in members_list:
|
|
pid = info.get("id")
|
|
if pid is None:
|
|
continue
|
|
|
|
est = ff_data.get(str(pid), {}).get("bs_estimate_human", "?")
|
|
member = {
|
|
"id": pid,
|
|
"name": info.get("name", "Unknown"),
|
|
"level": info.get("level", 0),
|
|
"estimate": est
|
|
}
|
|
faction_data.append(member)
|
|
|
|
# Save to file
|
|
file_path.parent.mkdir(exist_ok=True, parents=True) # ensure folder exists
|
|
with open(file_path, "w", encoding="utf-8") as f:
|
|
json.dump(faction_data, f, indent=2)
|
|
|
|
return True
|
|
|
|
|
|
# Wrappers for clarity
|
|
async def update_enemy_faction() -> bool:
|
|
return await fetch_and_save_faction(ENEMY_FACTION_ID, ENEMY_FILE)
|
|
|
|
|
|
async def update_friendly_faction() -> bool:
|
|
return await fetch_and_save_faction(YOUR_FACTION_ID, FRIENDLY_FILE)
|