#!/usr/bin/env python3
"""Add dictionaries to the comprehensive enum database."""

import json

# AttributeSetInfo dictionary manually extracted (COMPLETE)
attribute_set_info = {
    "2": "Test",
    "4": "Carraida's Benediction",
    "5": "Noble Relic Set",
    "6": "Ancient Relic Set",
    "7": "Relic Alduressa Set",
    "8": "Shou-jen Set",
    "9": "Empyrean Rings Set",
    "10": "Arm, Mind, Heart Set",
    "11": "Coat of the Perfect Light Set",
    "12": "Leggings of Perfect Light Set",
    "13": "Soldier's Set",
    "14": "Adept's Set",
    "15": "Archer's Set",
    "16": "Defender's Set",
    "17": "Tinker's Set",
    "18": "Crafter's Set",
    "19": "Hearty Set",
    "20": "Dexterous Set",
    "21": "Wise Set",
    "22": "Swift Set",
    "23": "Hardenend Set",
    "24": "Reinforced Set",
    "25": "Interlocking Set",
    "26": "Flame Proof Set",
    "27": "Acid Proof Set",
    "28": "Cold Proof Set",
    "29": "Lightning Proof Set",
    "30": "Dedication Set",
    "31": "Gladiatorial Clothing Set",
    "32": "Ceremonial Clothing",
    "33": "Protective Clothing",
    "34": "Noobie Armor",
    "35": "Sigil of Defense",
    "36": "Sigil of Destruction",
    "37": "Sigil of Fury",
    "38": "Sigil of Growth",
    "39": "Sigil of Vigor",
    "40": "Heroic Protector Set",
    "41": "Heroic Destroyer Set",
    "42": "Olthoi Armor D Red",
    "43": "Olthoi Armor C Rat",
    "44": "Olthoi Armor C Red",
    "45": "Olthoi Armor F Red",
    "46": "Olthoi Armor K Red",
    "47": "Olthoi Armor M Red",
    "48": "Olthoi Armor B Red",
    "49": "Olthoi Armor B Rat",
    "50": "Olthoi Armor K Rat",
    "51": "Olthoi Armor M Rat",
    "52": "Olthoi Armor F Rat",
    "53": "Olthoi Armor D Rat"
}

# Load existing database
with open('comprehensive_enum_database_v2.json', 'r') as f:
    db = json.load(f)

# Add dictionaries section
if 'dictionaries' not in db:
    db['dictionaries'] = {}

db['dictionaries']['AttributeSetInfo'] = {
    'name': 'AttributeSetInfo',
    'description': 'Equipment set names',
    'values': attribute_set_info
}

# Save updated database
with open('comprehensive_enum_database_v2.json', 'w') as f:
    json.dump(db, f, indent=2)

print("Added AttributeSetInfo dictionary to comprehensive enum database")
print(f"Total equipment sets: {len(attribute_set_info)}")
print("Example sets:")
for set_id in ['13', '14', '16', '21']:
    if set_id in attribute_set_info:
        print(f"  {set_id} -> {attribute_set_info[set_id]}")