# This upgrader moves memory pools from the system object to the SE workload
# object.
def upgrader(cpt):
    systems = {}

    # Find sections with 'num_mem_pools' options, and assume those are system
    # objects which host MemPools.
    for sec in cpt.sections():
        num_mem_pools = cpt.get(sec, "num_mem_pools", fallback=None)
        if num_mem_pools is not None:
            systems[sec] = num_mem_pools

    for sec, num_mem_pools in systems.items():
        # Transfer num_mem_pools to the new location.
        cpt.remove_option(sec, "num_mem_pools")
        cpt.set(f"{sec}.workload", "num_mem_pools", num_mem_pools)

        for idx in range(int(num_mem_pools)):
            old_name = f"{sec}.memPool{idx}"
            new_name = f"{sec}.workload.memPool{idx}"

            # Create the new section.
            cpt.add_section(new_name)

            # Copy items from the old section into it.
            for item in cpt.items(old_name):
                cpt.set(new_name, *item)

            # Delete the old section.
            cpt.remove_section(old_name)


depends = "mempool-sections"