parse_stuff.py

# Built-ins
import logging
import operator

# Third-party
from unstdlib.standard import listify

logger = logging.getLogger(__name__)

CLASSES = ["Witch", "Shadow", "Ranger", "Duelist", "Marauder", "Templar", "Scion"]


@listify
def evaluate_skill_gems(gem_list, class_, missing, quest_data, skill_data):
    for gem_name in gem_list:
        # Vendor recipe skill gems
        if gem_name == "Block Chance Reduction":
            yield (
                gem_name,
                "Vendor Puncture + any dexterity based shield with 20% quality",
                int(skill_data[gem_name]["lvl"]),
                skill_data[gem_name]["colour"],
            )
            gem_name = "Puncture"  # Rename to allow the base gem to be added as well
        elif gem_name == "Mirror Arrow":
            yield (
                gem_name,
                "Vendor Blink Arrow + 1 Orb of Alteration",
                int(skill_data[gem_name]["lvl"]),
                skill_data[gem_name]["colour"],
            )
            gem_name = "Blink Arrow"  # Rename to allow the base gem to be added as well
        # Vaal skill gems
        elif gem_name.startswith("Vaal"):
            base_gem_name = find_corresponding_non_vaal_skill_gem_name(gem_name)
            yield (
                gem_name,
                "Drop-only",
                int(skill_data[base_gem_name]["lvl"]),
                skill_data[base_gem_name]["colour"],
            )
            gem_name = base_gem_name  # Rename to allow the base gem to be added as well

        try:
            for quest in quest_data:
                if gem_name in quest_data[quest][class_]:
                    yield (
                        gem_name,
                        quest,
                        int(skill_data[gem_name]["lvl"]),
                        skill_data[gem_name]["colour"],
                    )
                    break
            else:
                missing.append(gem_name)
        except KeyError:
            logger.debug(f"{gem_name} is missing from skill data.")


def find_corresponding_non_vaal_skill_gem_name(skill_gem_name):
    dct = {
        "Vaal Breach": "Portal",
        "Vaal Impurity of Ice": "Purity of Ice",
        "Vaal Impurity of Fire": "Purity of Fire",
        "Vaal Impurity of Lightning": "Purity of Lightning",
    }
    return dct.get(skill_gem_name, skill_gem_name.partition("Vaal ")[2])


def sort_by_quest(tpl):
    class_, character_values = tpl
    character_values.sort(key=operator.itemgetter(2, 1, 0))
    return class_, character_values


@listify
def parse(build, quest_data, skill_data):
    class_ = build.class_name
    missing_skill_gems = []
    gem_list = evaluate_skill_gems(
        (skill_gem.name for skill_gem in build.skill_gems),
        class_,
        missing_skill_gems,
        quest_data,
        skill_data,
    )
    class_skill_gems = (class_, gem_list)
    yield sort_by_quest(class_skill_gems)
    other_classes = [i for i in CLASSES if i != class_]
    other_class_skill_gems = [
        (
            class_,
            evaluate_skill_gems(missing_skill_gems, class_, [], quest_data, skill_data),
        )
        for class_ in other_classes
    ]
    # Sort by number of skill gems available to class
    other_class_skill_gems.sort(key=lambda x: len(x[1]))
    for other in other_class_skill_gems:
        yield sort_by_quest(other)