Function getSimilarModPositions

gui/fitCommands/helpers.py:396–426

(mods, mainMod)

def getSimilarModPositions(mods, mainMod):
    sMkt = Market.getInstance()
    mainGroupID = getattr(sMkt.getGroupByItem(mainMod.item), 'ID', None)
    mainMktGroupID = getattr(sMkt.getMarketGroupByItem(mainMod.item), 'ID', None)
    mainEffects = set(getattr(mainMod.item, 'effects', ()))
    positions = []
    for position, mod in enumerate(mods):
        if mod.isEmpty:
            continue
        # Always include selected module itself
        if mod is mainMod:
            positions.append(position)
            continue
        if mod.itemID is None:
            continue
        # Modules which have the same item ID
        if mod.itemID == mainMod.itemID:
            positions.append(position)
            continue
        # And modules from the same group and market group too
        modGroupID = getattr(sMkt.getGroupByItem(mod.item), 'ID', None)
        modMktGroupID = getattr(sMkt.getMarketGroupByItem(mod.item), 'ID', None)
        modEffects = set(getattr(mod.item, 'effects', ()))
        if (
            modGroupID is not None and modGroupID == mainGroupID and
            modMktGroupID is not None and modMktGroupID == mainMktGroupID and
            modEffects == mainEffects
        ):
            positions.append(position)
            continue
    return positions


def getSimilarFighters(fighters, mainFighter):