AutoFlatten extract_order

extract_order.py

def extract_order(self, structure):
    '''
    Description: 
    This function does a BFS traversal to obtain the order in which 
    the array type fields are to be exploded

    :param structure: [type: dict] contains the hierarchical mapping for array fields

    :return order: [type: list] contains the fields in order in which array explode has to take place
    '''
    q = [('', structure['json'])]
    order = []
    while q:
        key, a = q.pop(0)
        for x in a.keys():
            order.append(f"{key}.{x}")
            q.append((f"{key}.{x}", a[x]))
    return order