fork download
  1. #!/usr/bin/env python
  2.  
  3. import argparse
  4. import csv
  5. import semanticnet as sn
  6. import sys
  7. sys.path.append('C:\Python34')
  8.  
  9. def add_edge_ifn(graph, src, dst):
  10. # get the edges between src and dst
  11. edges = graph.get_edges_between(src, dst)
  12. # if there are no edges, add one and return it
  13. if not edges:
  14. return graph.add_edge(src, dst)
  15. # if there is, return the first edge's ID
  16. return edges.items()[0][0]
  17.  
  18. if __name__ == "__main__":
  19. parser = argparse.ArgumentParser("ssv.py")
  20. parser.add_argument("input_filename", type=str)
  21. parser.add_argument("output_filename", type=str)
  22. args = parser.parse_args()
  23.  
  24. g = sn.DiGraph()
  25. with open(args.input_filename, "dataSmall") as infile:
  26. reader = csv.reader(infile, delimiter=',')
  27. for row in reader:
  28. previous = None
  29. for elem in row:
  30. current = g.get_or_add_node(elem, {"label": elem})
  31. if previous != None:
  32. add_edge_ifn(g, previous['id'], current['id'])
  33. previous = current
  34.  
  35. g.save_json(args.output_filename)
Runtime error #stdin #stdout #stderr 0.03s 9512KB
stdin
Standard input is empty
stdout
Standard output is empty
stderr
Traceback (most recent call last):
  File "prog.py", line 5, in <module>
ImportError: No module named semanticnet