#!/usr/bin/env python3
"""Test if API is returning clean format"""
import json
import urllib.request

# Login
login_data = json.dumps({"username": "admin", "password": "admin123"}).encode()
login_req = urllib.request.Request(
    "http://localhost:8000/api/v1/auth/login",
    data=login_data,
    headers={"Content-Type": "application/json"}
)
with urllib.request.urlopen(login_req) as response:
    token = json.loads(response.read())["access_token"]

# Test action mappings export
export_req = urllib.request.Request(
    "http://localhost:8000/api/v1/configuration/action-mappings/export",
    headers={"Authorization": f"Bearer {token}"}
)
with urllib.request.urlopen(export_req) as response:
    data = json.loads(response.read())
print("Response keys:", list(data.keys()))
print()

if "action_mappings" in data:
    print("CLEAN FORMAT - Success!")
    print(f"Total mappings: {data['total_count']}")
    print(f"Total actions: {data['total_actions']}")
    print("\nFirst 3 mappings:")
    print(json.dumps(data["action_mappings"][:3], indent=2))
elif "success" in data:
    print("OLD FORMAT - Still returning raw data")
    print(f"Total mappings: {data['total_count']}")
    print("\nFirst mapping:")
    print(json.dumps(data["mappings"][0], indent=2))