Diagramme de séquence, flow graph

Bonjour.
je cherche à faire en Python ce qu'on appelle des diagrammes de séquence en UML ou des flow graph avec l'analyseur de traces Wireshark, plus précisément une représentation graphique d'une trace réseau entre différents équipements où chaque équipement est représenté en haut du diagramme par une boîte avec, en dessous, une ligne verticale et les messages échangés sont représentés par des flèches horizontales ou obliques entre l'équipement émetteur et le récepteur.
Ci-dessous un lien exemple:
http://jakarta.apache.org/cactus/how_it_works_uml.html

Je veux faire quelque chose d'un peu plus simple que l'exemple ci-dessus. Quelle bibliothèque 2D Python utiliser?

Merci.
Cordialement,