U td(@sZdZddlZddlZddlZddlZdZdZGdddZGdddZ Gd d d Z dS) zGClasses that help igraph communicate with Gephi (http://www.gephi.org).N)GephiConnectionGephiGraphStreamerGephiGraphStreamingAPIFormatzrestructuredtext enc@sdeZdZdZdddZdd Zd d Zd d ZddZe ddZ e j ddZ ddZ ddZ dS)rz=Object that represents a connection to a Gephi master server.N 127.0.0.1cCs$g|_d|_|p|||||_dS)aConstructs a connection to a Gephi master server. The connection object can be constructed either by specifying the ``url`` directly, or by specifying the ``host``, ``port`` and ``workspace`` arguments. The latter three are evaluated only if ``url`` is None; otherwise the ``url`` will take precedence. The ``url`` argument does not have to include the operation (e.g., ``?operation=updateGraph``); the connection will take care of it. E.g., if you wish to connect to workspace 2 in a local Gephi instance on port 7341, the correct form to use for the ``url`` is as follows:: http://localhost:7341/workspace0 iN)_pending_operations_autoflush_threshold_construct_default_urlurl)selfr hostport workspacerL/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/remote/gephi.py__init__szGephiConnection.__init__cCs*z |Wntjjk r$YnXdS)N)closeurlliberrorURLErrorr rrr__del__%s zGephiConnection.__del__cCsd|||fS)Nzhttp://%s:%d/workspace%dr)r r rrrrrr ,sz&GephiConnection._construct_default_urlcCs |dS)ZFlushes all the pending operations to the Gephi master server in a single request.N)flushrrrrr/szGephiConnection.closecCs,d|j}g|_tjj|j|d}|S)r)data)joinrrrequesturlopen _update_urlread)r rconnrrrr4s zGephiConnection.flushcCs|jS)z;The URL of the Gephi workspace where the data will be sent.) _url_rootrrrrr <szGephiConnection.urlcCs"||_|jd|_|jd|_dS)Nz?operation=getGraphz?operation=updateGraph)r#Z_get_urlr )r valuerrrr As cCs(|j|t|j|jkr$|dS)z^Sends the given raw data to the Gephi streaming master server in an HTTP POST request.N)rappendlenr r)r rrrrwriteGs zGephiConnection.writecCsd|jj|jfS)Nz %s(url=%r)) __class____name__r rrrr__repr__NszGephiConnection.__repr__)Nrrr)r) __module__ __qualname____doc__rrr rrpropertyr setterr'r*rrrrrs   rc@sHeZdZdZifddZifddZddZdd Zd d Zd d Z dS)rzObject that implements the Gephi graph streaming API format and returns Python objects corresponding to the events defined in the API. cCs d||iiS)aGenerates a Python object corresponding to the event that adds a node with the given identifier and attributes in the Gephi graph streaming API. Example:: >>> api = GephiGraphStreamingAPIFormat() >>> api.get_add_node_event("spam") {'an': {'spam': {}}} >>> api.get_add_node_event("spam", dict(ham="eggs")) {'an': {'spam': {'ham': 'eggs'}}} Zanrr identifier attributesrrrget_add_node_eventWs z/GephiGraphStreamingAPIFormat.get_add_node_eventcCs0t|}||d<||d<t||d<d||iiS)zGenerates a Python object corresponding to the event that adds an edge with the given source, target, directednessr and attributes in the Gephi graph streaming API. sourcetargetdirectedZae)dictbool)r r1r4r5r6r2resultrrrget_add_edge_eventes  z/GephiGraphStreamingAPIFormat.get_add_edge_eventcCs d||iiS)aAGenerates a Python object corresponding to the event that changes the attributes of some node in the Gephi graph streaming API. The given attributes are merged into the existing ones; use C{None} as the attribute value to delete a given attribute. Example:: >>> api = GephiGraphStreamingAPIFormat() >>> api.get_change_node_event("spam", dict(ham="eggs")) {'cn': {'spam': {'ham': 'eggs'}}} >>> api.get_change_node_event("spam", dict(ham=None)) {'cn': {'spam': {'ham': None}}} Zcnrr0rrrget_change_node_eventpsz2GephiGraphStreamingAPIFormat.get_change_node_eventcCs d||iiS)aAGenerates a Python object corresponding to the event that changes the attributes of some edge in the Gephi graph streaming API. The given attributes are merged into the existing ones; use C{None} as the attribute value to delete a given attribute. Example:: >>> api = GephiGraphStreamingAPIFormat() >>> api.get_change_edge_event("spam", dict(ham="eggs")) {'ce': {'spam': {'ham': 'eggs'}}} >>> api.get_change_edge_event("spam", dict(ham=None)) {'ce': {'spam': {'ham': None}}} Zcerr0rrrget_change_edge_eventsz2GephiGraphStreamingAPIFormat.get_change_edge_eventcCs d|iiiS)a1Generates a Python object corresponding to the event that deletes a node with the given identifier in the Gephi graph streaming API. Example:: >>> api = GephiGraphStreamingAPIFormat() >>> api.get_delete_node_event("spam") {'dn': {'spam': {}}} dnrr r1rrrget_delete_node_events z2GephiGraphStreamingAPIFormat.get_delete_node_eventcCs d|iiiS)a:Generates a Python object corresponding to the event that deletes an edge with the given identifier in the Gephi graph streaming API. Example:: >>> api = GephiGraphStreamingAPIFormat() >>> api.get_delete_edge_event("spam:ham") {'de': {'spam:ham': {}}} derr>rrrget_delete_edge_events z2GephiGraphStreamingAPIFormat.get_delete_edge_eventN) r)r+r,r-r3r:r;r<r?rArrrrrRs   rc@s6eZdZdZd ddZddZd ddZdd d ZdS)raClass that produces JSON event objects that stream an igraph graph to Gephi using the Gephi Graph Streaming API. The Gephi graph streaming format is a simple JSON-based format that can be used to post mutations to a graph (i.e. node and edge additions, removals and updates) to a remote component. For instance, one can open up Gephi (http://www.gephi.org), install the Gephi graph streaming plugin and then send a graph from igraph straight into the Gephi window by using `GephiGraphStreamer` with the appropriate URL where Gephi is listening. Example:: >>> from cStringIO import StringIO >>> from igraph import Graph >>> buf = StringIO() >>> streamer = GephiGraphStreamer() >>> graph = Graph.Formula("A --> B, B --> C") >>> streamer.post(graph, buf) >>> print(buf.getvalue()) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE {"an": {"igraph:...:v:0": {"name": "A"}}} {"an": {"igraph:...:v:1": {"name": "B"}}} {"an": {"igraph:...:v:2": {"name": "C"}}} {"ae": {"igraph:...:e:0:1": {...}}} {"ae": {"igraph:...:e:1:2": {...}}} NcCs|ptjdd|_t|_dS)a"Constructs a Gephi graph streamer that will post graphs to a given file-like object or a Gephi connection. ``encoder`` must either be ``None`` or an instance of ``json.JSONEncoder`` and it must contain the JSON encoder to be used when posting JSON objects. T) ensure_asciiN)json JSONEncoderencoderrformat)r rErrrrszGephiGraphStreamer.__init__ccsdtt|f}|jj}|jD]}|d||jf|Vq |jj}|}|j D]<}|d||j |j fd||j fd||j f||VqVdS)a Iterates over the JSON objects that build up the graph using the Gephi graph streaming API. The objects returned from this function are Python objects; they must be formatted with ``json.dumps`` before sending them to the destination. z igraph:%sz%s:v:%dz %s:e:%d:%dN) hexidrFr3vsindexr2r:Z is_directedesr4r5)r graphZ id_prefixadd_nodeZvertexZadd_edger6edgerrr iterjsonobjs    zGephiGraphStreamer.iterjsonobjcCs8|p|j}||D]}|j|||ddq|dS)aCPosts the given graph to the destination of the streamer using the given JSON encoder. When ``encoder`` is ``None``, it falls back to the default JSON encoder of the streamer in the `encoder` property. ``destination`` must be a file-like object or an instance of `GephiConnection`. F)rErN)rErO send_eventr)r rL destinationrEZjsonobjrrrposts zGephiGraphStreamer.postTcCs:|p|j}|||d|d|r6|dS)aSends a single JSON event to the given destination using the given JSON encoder. When ``encoder`` is ``None``, it falls back to the default JSON encoder of the streamer in the `encoder` property. ``destination`` must be a file-like object or an instance of `GephiConnection`. The method flushes the destination after sending the event. If you want to avoid this (e.g., because you are sending many events), set ``flush`` to ``False``. zutf-8s N)rEr'encoder)r eventrQrErrrrrPs  zGephiGraphStreamer.send_event)N)N)NT)r)r+r,r-rrOrRrPrrrrrs   r) r-rC urllib.errorr urllib.parseurllib.request__all__ __docformat__rrrrrrrsDW