from jhplot import *

g=HGraph().buildWeightedPseudograph()
v1,v2,v3 = "v1","v2","v3"
g.addVertex(v1), g.addVertex(v2), g.addVertex(v3) 

factory = g.getEdgeFactory()
e = factory.createEdge(v1,v2)
g.setEdgeWeight(e, 5) 
g.addEdge(v1, v2, e)

e = factory.createEdge(v2,v3)
g.setEdgeWeight(e, 8.8)
g.addEdge(v2, v3, e)

e = factory.createEdge(v2,v1)
g.setEdgeWeight(e, 0.1)
g.addEdge(v2, v1, e)

HGraph().showGraph(g) # show the graph in GUI 

name='csvfile.csv'
# write to CVS with weights
with open(name,'wb') as file:
    for e in g.edgeSet():
        line=str(g.getEdgeSource(e))+","+str(g.getEdgeTarget(e))+","+str(g.getEdgeWeight(e))
        file.write(line)
        file.write('\n')


# reading from CVS
gg=HGraph().buildWeightedPseudograph()
vertex=set()
graph=[]
with open(name) as f:
  alist = f.read().splitlines()
  for l in alist:
     words = l.split(',')
     v1,v2=str(words[0]),str(words[1]) 
     vertex.add(v1)
     vertex.add(v2)
     w=float(words[2])
     graph.append([v1,v2,w])

# add vertex
for v in vertex:
    gg.addVertex(v)
for j in graph:
    v1,v2,w=j[0],j[1],j[2] 
    e = factory.createEdge(v1,v2)
    gg.setEdgeWeight(e, w)
    gg.addEdge(v1, v2, e)
print "New graph was created from file ",name
HGraph().showGraph(gg) # show the graph in GUI