Any simple way to hook into WW body part assignment?

[Witchbrew]

Hi, wondering if any modders have ever got their mods working nicely with WW body part assignment/selection, if there's any examples or references I can follow.
Making my own simplified 'omegaverse' mod (I know there's an existing one). Basically there's alpha/beta/omega, and I want female alphas to automatically be futa (have a penis) without having to manually adjust body for every sim. Already have python file working for auto-assignment of the ABO traits. I looked in the WW tuning .package file for what traits I might be able to auto-assign alongside it, and I saw "WickedWhims_Trait_Nudity_Core_HasPenis"  took that decimal and put it in the python script, but there's no body changes to the females assigned even if I undress them. 

 

Wondering if I'm looking at the right trait and doing it wrong, or if it's all more complicated than that and I shouldn't bother. I'd prefer to have a mod that requires WW & does it automatically vs a standalone mod that you still have to pick parts manually but 😅 how difficult it would be to get auto-assignment working determines that. 
 

Here's a basic snippet of what im doing

WW_TRAIT_HAS_PENIS = 3851100925

def get_trait(trait_id):
    trait_manager = get_instance_manager(Types.TRAIT)
    return trait_manager.get(trait_id)

def try_assign_abo_trait(sim_info):
# logic blah blah blah about whether alpha is female here then tldr:
	ww_trait = get_trait(WW_TRAIT_HAS_PENIS)
	sim_info.add_trait(ww_trait)