Search code examples
pythongraphvizpygraph

Setting graph attributes in pygraph


I found in the documentation of pygraph how to change the attributes of the nodes and the edges, but I found no help on how to change the attributes of the graphs.

I tried without luck:

gr = graph()
gr.__setattr__('aspect',2)

What do you recommend?

Thanks!

[update] I also tried:

gr = graph()
gr.__setattr__('rotate',90)
gr.rotate = 90
gr.color = 'red'
setattr(gr,'bgcolor','red')

[update 2] Example code from the website with the different ideas to change the attributes:

#!/usr/bin/env python

# Copyright (c) 2007-2008 Pedro Matiello <pmatiello@gmail.com>
# License: MIT (see COPYING file)

# Import graphviz
import sys
sys.path.append('..')
sys.path.append('/usr/lib/graphviz/python/')
sys.path.append('/usr/lib64/graphviz/python/')
import gv

# Import pygraph
from pygraph.classes.graph import graph
from pygraph.classes.digraph import digraph
from pygraph.algorithms.searching import breadth_first_search
from pygraph.readwrite.dot import write

# Graph creation
gr = graph()

gr.__setattr__('rotate',90)
gr.rotate = 90
gr.color = 'red'
setattr(gr,'bgcolor','red')

# Add nodes and edges
gr.add_nodes(["Portugal","Spain","France","Germany","Belgium","Netherlands","Italy"])
gr.add_nodes(["Switzerland","Austria","Denmark","Poland","Czech Republic","Slovakia","Hungary"])
gr.add_nodes(["England","Ireland","Scotland","Wales"])

gr.add_edge(("Portugal", "Spain"))
gr.add_edge(("Spain","France"))
gr.add_edge(("France","Belgium"))
gr.add_edge(("France","Germany"))
gr.add_edge(("France","Italy"))
gr.add_edge(("Belgium","Netherlands"))
gr.add_edge(("Germany","Belgium"))
gr.add_edge(("Germany","Netherlands"))
gr.add_edge(("England","Wales"))
gr.add_edge(("England","Scotland"))
gr.add_edge(("Scotland","Wales"))
gr.add_edge(("Switzerland","Austria"))
gr.add_edge(("Switzerland","Germany"))
gr.add_edge(("Switzerland","France"))
gr.add_edge(("Switzerland","Italy"))
gr.add_edge(("Austria","Germany"))
gr.add_edge(("Austria","Italy"))
gr.add_edge(("Austria","Czech Republic"))
gr.add_edge(("Austria","Slovakia"))
gr.add_edge(("Austria","Hungary"))
gr.add_edge(("Denmark","Germany"))
gr.add_edge(("Poland","Czech Republic"))
gr.add_edge(("Poland","Slovakia"))
gr.add_edge(("Poland","Germany"))
gr.add_edge(("Czech Republic","Slovakia"))
gr.add_edge(("Czech Republic","Germany"))
gr.add_edge(("Slovakia","Hungary"))

# Draw as PNG
dot = write(gr)
gvv = gv.readstring(dot)
gv.layout(gvv,'dot')
gv.render(gvv,'png','europe.png')

Solution

  • So, doing more research on my own, I found that pygraphviz is offering what I need. Here is the pygraph example using pygraphviz and accepting attributes. It is also shorter, as you don't have to specify the nodes if all of them are connected via edges.

    #!/usr/bin/env python
    
    # Copyright (c) 2007-2008 Pedro Matiello <pmatiello@gmail.com>
    # License: MIT (see COPYING file)
    
    # Import pygraphviz
    import pygraphviz as pgv
    
    # Graph creation and setting of attributes
    gr = pgv.AGraph(rotate='90',bgcolor='lightgray')
    
    # Add nodes and edges
    gr.add_edge(("Portugal", "Spain"))
    gr.add_edge(("Spain","France"))
    gr.add_edge(("France","Belgium"))
    gr.add_edge(("France","Germany"))
    gr.add_edge(("France","Italy"))
    gr.add_edge(("Belgium","Netherlands"))
    gr.add_edge(("Germany","Belgium"))
    gr.add_edge(("Germany","Netherlands"))
    gr.add_edge(("England","Wales"))
    gr.add_edge(("England","Scotland"))
    gr.add_edge(("Scotland","Wales"))
    gr.add_edge(("Switzerland","Austria"))
    gr.add_edge(("Switzerland","Germany"))
    gr.add_edge(("Switzerland","France"))
    gr.add_edge(("Switzerland","Italy"))
    gr.add_edge(("Austria","Germany"))
    gr.add_edge(("Austria","Italy"))
    gr.add_edge(("Austria","Czech Republic"))
    gr.add_edge(("Austria","Slovakia"))
    gr.add_edge(("Austria","Hungary"))
    gr.add_edge(("Denmark","Germany"))
    gr.add_edge(("Poland","Czech Republic"))
    gr.add_edge(("Poland","Slovakia"))
    gr.add_edge(("Poland","Germany"))
    gr.add_edge(("Czech Republic","Slovakia"))
    gr.add_edge(("Czech Republic","Germany"))
    gr.add_edge(("Slovakia","Hungary"))
    
    # Draw as PNG
    gr.layout(prog='dot')
    gr.draw('europe.png')