# see https://github.com/WolfgangFahl/gremlin-python-tutorial/blob/master/test_005_graphviz.py
from tutorial import remote
from graphviz import Digraph
import os.path
from gremlin_python.process.traversal import T

# initialize a remote traversal
g = remote.RemoteTraversal().g()

# test creating a graphviz graph from the tinkerpop graph
def test_createGraphvizGraph():
    # make sure we re-load the tinkerpop modern example
    remoteTraversal=remote.RemoteTraversal()
    remoteTraversal.load("tinkerpop-modern.xml")
    # start a graphviz
    dot = Digraph(comment='Modern')
    # get vertice properties including id and label as dicts
    for vDict in g.V().valueMap(True).toList():
        # uncomment to debug
        # print vDict
        # get id and label
        vId=vDict[T.id]
        vLabel=vDict[T.label]
        # greate a graphviz node label
        # name property is alway there
        gvLabel=r"%s\n%s\nname=%s" % (vId,vLabel,vDict["name"][0])
        # if there is an age property add it to the label
        if "age" in vDict:
            gvLabel=gvLabel+r"\nage=%s" % (vDict["age"][0])
        # create a graphviz node
        dot.node("node%d" % (vId),gvLabel)
    # loop over all edges
    for e in g.E():
        # get the detail information with a second call per edge (what a pitty to be so inefficient ...)
        eDict=g.E(e.id).valueMap(True).next()
        # uncomment if you'd like to debug
        # print (e,eDict)
        # create a graphviz label
        geLabel=r"%s\n%s\nweight=%s" % (e.id,e.label,eDict["weight"])
        # add a graphviz edge
        dot.edge("node%d" % (e.outV.id),"node%d" % (e.inV.id),label=geLabel)
    # modify the styling see http://www.graphviz.org/doc/info/attrs.html
    dot.edge_attr.update(arrowsize='2',penwidth='2')
    dot.node_attr.update(style='filled',fillcolor="#A8D0E4")
    # print the source code
    print (dot.source)
    # render without viewing - default is creating a pdf file
    dot.render('/tmp/modern.gv', view=False)
    # check that the pdf file exists
    assert os.path.isfile('/tmp/modern.gv.pdf')

# call the test
test_createGraphvizGraph()

# see https://github.com/WolfgangFahl/gremlin-python-tutorial/blob/master/test_005_graphviz.py
from tutorial import remote
from graphviz import Digraph
import os.path
from gremlin_python.process.traversal import T

# initialize a remote traversal
g = remote.RemoteTraversal().g()

# test creating a graphviz graph from the tinkerpop graph
def test_createGraphvizGraph():
    # make sure we re-load the tinkerpop modern example
    remoteTraversal=remote.RemoteTraversal()
    remoteTraversal.load("tinkerpop-modern.xml")
    # start a graphviz
    dot = Digraph(comment='Modern')
    # get vertice properties including id and label as dicts
    for vDict in g.V().valueMap(True).toList():
        # uncomment to debug
        # print vDict
        # get id and label
        vId=vDict[T.id]
        vLabel=vDict[T.label]
        # greate a graphviz node label
        # name property is alway there
        gvLabel=r"%s\n%s\nname=%s" % (vId,vLabel,vDict["name"][0])
        # if there is an age property add it to the label
        if "age" in vDict:
            gvLabel=gvLabel+r"\nage=%s" % (vDict["age"][0])
        # create a graphviz node
        dot.node("node%d" % (vId),gvLabel)
    # loop over all edges
    for e in g.E():
        # get the detail information with a second call per edge (what a pitty to be so inefficient ...)
        eDict=g.E(e.id).valueMap(True).next()
        # uncomment if you'd like to debug
        # print (e,eDict)
        # create a graphviz label
        geLabel=r"%s\n%s\nweight=%s" % (e.id,e.label,eDict["weight"])
        # add a graphviz edge
        dot.edge("node%d" % (e.outV.id),"node%d" % (e.inV.id),label=geLabel)
    # modify the styling see http://www.graphviz.org/doc/info/attrs.html
    dot.edge_attr.update(arrowsize='2',penwidth='2')
    dot.node_attr.update(style='filled',fillcolor="#A8D0E4")
    # print the source code
    print (dot.source)
    # render without viewing - default is creating a pdf file
    dot.render('/tmp/modern.gv', view=False)
    # check that the pdf file exists
    assert os.path.isfile('/tmp/modern.gv.pdf')

# call the test
test_createGraphvizGraph()

        file="data/tinkerpop-modern.xml";
        g.io(file).read().iterate()
        vCount=g.V().count().next()
        print ("%s has %d vertices" % (file,vCount))
        assert vCount==6

    '''
    Created on 2020-03-30
    
    @author: wf
    '''
    from gremlin_python.driver.driver_remote_connection import DriverRemoteConnection
    from gremlin_python.structure.graph import Graph
    from shutil import copyfile
    import os
    
    class RemoteGremlin(object):
        '''
        helper for remote gremlin connections
        '''
    
        def __init__(self, server, port=8182):
            '''
            construct me with the given server and port
            '''
            self.server=server
            self.port=port    
            
        def sharepoint(self,sharepoint,sharepath):
            '''
            set up the sharepoint
            '''
            self.sharepoint=sharepoint
            self.sharepath=sharepath
            
            
        def share(self,file):
            '''
            share the given file  and return the path as seen by the server
            '''
            fbase=os.path.basename(file)
            copyfile(file,self.sharepoint+fbase)
            return self.sharepath+fbase
                
        def open(self):
            '''
            open the remote connection
            '''
            self.graph = Graph()
            self.url='ws://%s:%s/gremlin' % (self.server,self.port)
            self.connection = DriverRemoteConnection(self.url, 'g')    
            # The connection should be closed on shut down to close open connections with connection.close()
            self.g = self.graph.traversal().withRemote(self.connection)
    
        def close(self):
            '''
            close the remote connection
            '''
            self.connection.close()

    '''
    Created on 2020-03-28
    
    @author: wf
    '''
    import unittest
    from tp.gremlin import RemoteGremlin
    
    class JanusGraphTest(unittest.TestCase):
        '''
        test access to a janus graph docker instance via the RemoteGremlin helper class
        '''
    
        def setUp(self):
            pass
    
    
        def tearDown(self):
            pass
        
        def test_loadGraph(self):
            # change to your server
            rg=RemoteGremlin("capri.bitplan.com")
            rg.open()
            # change to your shared path
            rg.sharepoint("/Volumes/bitplan/user/wf/graphdata/","/graphdata/")
            g=rg.g
            graphmlFile="air-routes-small.xml";
            shared=rg.share(graphmlFile)
            # drop the existing content of the graph
            g.V().drop().iterate()
            # read the content from the air routes example
            g.io(shared).read().iterate()
            vCount=g.V().count().next()
            print ("%s has %d vertices" % (shared,vCount))
            assert vCount==47
    
    
    if __name__ == "__main__":
        #import sys;sys.argv = ['', 'Test.testName']
        unittest.main()

def testJanusGraph(self):
        graph = Graph()
        connection = DriverRemoteConnection('ws://localhost:8182/gremlin', 'g')
        # The connection should be closed on shut down to close open connections with connection.close()
        g = graph.traversal().withRemote(connection)
        # Reuse 'g' across the application
        herculesAge = g.V().has('name', 'hercules').values('age').next()
        print('Hercules is {} years old.'.format(herculesAge))
        pass