Comment créer de beaux diagrammes d’architecture avec Python ? (~7min)

Paramètres avancés

Maintenant que l’on sait utiliser les diagrammes, les clusters, les edges et les nodes, intéressons-nous à la personnalisation. Il y a deux objets personnalisables : les nodes et les edges.

Personnalisation des Edges

Regardons en premier lieu comment personnaliser les edges. Il y a 3 paramètres de personnalisation : la couleur, le style et le label. La couleur par défaut est le gris, mais si vous souhaitez paramétrer une autre couleur, les couleurs sont celles utilisées par le package matplotlib que vous pouvez retrouver ici</a >.

Ensuite, il est possible de jouer sur le style et il y en a 4 de disponibles :

• • Le style par défaut qui est une ligne continue
  • Bold, une ligne continue en gras
  • Dashed, la ligne est faite de tirets
  • Dotted, la ligne est faite de points

Il n’est pas possible de combiner des styles différents, i.e. avoir une ligne faite de tirets en gras.

Enfin, il est possible de rajouter un label sur un Edge si on souhaite expliciter ce que représente cet Edge.

Voyons voir ce que cela donne en code.

    from diagrams import Diagram, Cluster, Edge
    from diagrams.aws.analytics import Glue, Quicksight
    from diagrams.aws.database import RDS
    from diagrams.aws.management import Cloudwatch
    from diagrams.aws.storage import S3
    
    def arrow(color='black', style='line', label=None):    
        """    
            Function to define the edge between the part of the diagram    
            :param color: the color of the edge, could be any color    
            :type color: str    
            :param style: the style of the edge, could be dashed, dotted, bold or line (default)    
            :type style: str    
            :param label: the text you want to show on the edge    
            :type label: str    
            :return: Edge object with the different parameters we set up    
            :type: Edge()    
        """    
        return Edge(color=color, style=style, label=label)
        
    with Diagram('Pipeline - Global Overview', filename='Diagramm/Pipeline_GO', show=True, direction="LR"):    
        with Cluster('RDS'):        
            db = RDS('PostgreSQL BDD\n stored in RDS')    
            
        with Cluster('AWS Glue (ETL)'):        
            jobs = Glue('Data Engineering\n (Filter,\n join,\n rename..)')    
            
        with Cluster('Cloudwatch'):        
            log = Cloudwatch('Monitoring Scripts')    
            
        with Cluster('S3'):        
            bucket = S3('S3 Buckets\n to store\n AWS Glue outputs')    
            
        with Cluster('Quicksight'):        
            dashboard = Quicksight('Dashboard\n for monitoring')    
            
        db_mycharlotte >> arrow(color='red', style='bold') >> jobs >> arrow(color='red', style='bold') >> \    
        bucket >> arrow(color='red', style='bold') >> dashboard    
        
        jobs >> arrow(color='hotpink', style='dashed') >> log

Ici, j’ai créé une fonction, arrow(), qui produit par défaut une flèche noire en gras, que je préfère à la flèche par défaut du package. J’utilise ensuite cette fonction pour définir les différentes flèches que je souhaite avoir dans mon graphique. Quand on souhaite personnaliser un Edge, il faut explicitement le marquer dans le schéma du diagramme entre les deux nœuds concernés. Ici j’ai souhaité que les Edges du pipeline soient en rouge et en gras, sauf la flèche pour les logs qui est en rose et en pointillés. On peut voir cela dans les deux dernières lignes du code.