Saving player data to json file

services/torn_api.py

@@ -1,12 +1,68 @@
+# services/torn_api.py
 import aiohttp
-from config import TORN_API_KEY, ENEMY_FACTION_ID
+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 def fetch_enemy_members():
-    url = f"https://api.torn.com/v2/faction/{ENEMY_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("Torn faction fetch error:", resp.status)
-                return []
+                print(f"Torn faction fetch error: {resp.status}")
+                return False
             data = await resp.json()
-    return data.get("members", [])
+
+    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)