load_graph.py

import csv

from get_icon import *
import graph

"""
This function reads a CSV history file and loads each URL as a node in the graph
"""

def load_graph(filename=str()):
    # open File
    file = open(filename, 'r')
    # define graph object
    g = graph.Graph()
    # metadata stores URL : (Page Title, Icon path, Icon position)
    md = {}

    # read file
    reader = csv.reader(file, skipinitialspace=True)
    for row in reader:
        try:
            md[row[4]] = [row[2], get_icon(row[4]), (0, 0)]
            g.add_vertex(row[4])

        except:
            print("Could not add " + row[4])
            print(row[0])
            continue
    return g, md