def download_graphs()

in src/utils.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

def download_graphs(graph_url, data_dir = "./graph_data"):
    """
    Download graph .zip files from web URL
    
    :param graph_url: dict, with a format of {'graph_name': 'url'}
    :param data_dir: str, the directory path to store graph data
    """
    
    if not os.path.exists(data_dir):
        os.makedirs(data_dir)
        print("Created ./graph_data directory in local machine to store graph data.")
    
    for graph_name in graph_url.keys():
        url = graph_url[graph_name]
        with urlopen(url) as zr:
            with ZipFile(BytesIO(zr.read())) as zf:
                zf.extractall(data_dir)