Run an AQL and get a Python network object in return
pip install arangodb-pythongraph
Duh.
This package is based on pyintergraph and thus supports extraction to NetworkX, python-IGraph and Graph-Tools graph objects. However, these libraries are not defined as requirements for this package and if you want to extract to each of them you are required to install the necessary package accordingly.
All queries must return path objects.
from arangodb_pythongraph import execute_to_pygraph
db = ... # ArangoDB connection (use python-arango package)
example_query = '''
FOR v0 in vertex_collection
FOR e, v, p IN OUTBOUND v0 edge_collection
RETURN p
'''
python_graph = execute_to_pygraph(db, query)
nx_graph = python_graph.to_networkx()
gt_graph = python_graph.to_graph_tool()
ig_graph = python_graph.to_igraph()
If you want to export the graph (for example to use it with Gephi),
you might run into trouble if you have nested or complex attributes in your graph.
To overcome this, use the cleanup argument:
python_graph = execute_to_pygraph(db, query, cleanup=True)
The functionality might be missing some use cases so if you encounter problems while exporting the graph to file, please open an Issue describing the error you're getting together with a portion of the data you're trying to export.
For a neater use, run arangodb_pythongraph.register()
Before:
python_graph = execute_to_pythongraph(db, query)
After:
python_graph = db.aql.execute_to_pythongraph(query)
Contributions are more than welcome :)
Note that the project is managed with poetry, so make sure you use poetry to update the pypoject file.
To run the tests you must have a running instance of ArangoDB. If you don't have a connection to an existing DB you can use docker to run it easily.
Once you have a running ArangoDB connection, create a DB named 'test' (or any other name you choose), and run tests with pytest
ARANGODB_PASS=<pass_goes_here> ARANGODB_NAME=test poetry run pytest