gremlin export example

gremlin-exporter.py

import sys
import csv

from gremlin_python.structure.graph import Graph
from gremlin_python.driver.driver_remote_connection import DriverRemoteConnection

URL = 'ws://localhost:8182/gremlin'

def main(argv):

    import pdb; pdb.set_trace()

    graph = Graph()
    g = graph.traversal().withRemote(DriverRemoteConnection(URL, 'g'))

    if len(argv) < 2:
        print('Usage: # python %s LABEL' % argv[0])
        quit()

    for item in argv[1:]:
        with open(item + ".csv", "w", newline="") as outf:
            writer = csv.writer(outf, delimiter=",", quotechar='"', quoting=csv.QUOTE_MINIMAL)
            lines = g.V().hasLabel(item).toList()
            key_list = g.V(lines[0]).properties().key().toList()
            key_list.append('label')
            writer.writerow(key_list)
            for line in lines:
#                print(g.V(l).values().toList())
                tmp = g.V(line).values().toList()
                tmp.append(item)
                writer.writerow(tmp)

if __name__ == "__main__":
    main(sys.argv)