def adjust_graph()

in scripts/generate_replica_metadata.py [0:0]

• 34 lines of code
• 14 McCabe index (conditional complexity)

def adjust_graph(graph, points, name):
    # manual fixes
    if name == 'apartment_0':
        upper_floor_nodes = list()
        for node in graph.nodes():
            if graph.nodes[node]['point'][1] > 0.5:
                upper_floor_nodes.append(node)
        graph.remove_nodes_from(upper_floor_nodes)
        graph.remove_nodes_from([0, 30, 31, 32, 47, 48, 49, 50, 67, 90, 173, 378, 509])
    elif name == 'apartment_2':
        graph.remove_nodes_from([0, 168, 187, 218, 237])
        graph.add_node(191, point=points[191])
        graph.add_edges_from([(173, 191), (191, 209)])
    elif name == 'frl_apartment_1':
        graph.remove_nodes_from([164, 177])
    elif name == 'frl_apartment_2':
        graph.remove_nodes_from([174])
    elif name == 'frl_apartment_5':
        graph.remove_nodes_from([9, 18, 26, 35])
    elif name == 'hotel_0':
        graph.remove_nodes_from([1])
    elif name == 'office_0':
        # check
        graph.remove_nodes_from([6, 7])
        graph.add_edges_from([(12, 21)])
    elif name == 'office_2':
        graph.remove_nodes_from([0, 11, 3, 9, 15, 10, 16])
    elif name == 'office_3':
        graph.remove_nodes_from([48, 82, 115])
        # graph.add_edges_from([(56, 69)])
    elif name == 'room_0':
        graph.remove_nodes_from([123, 124, 125, 126, 127, 118, 117, 102, 103, 111, 112, 120, 121])
        # graph.add_edges_from([(95, 103)])
    elif name == 'room_1':
        # graph.add_edges_from([(37, 45), (51, 59)])
        graph.remove_nodes_from([45, 51])
    else:
        return False
    return True