当前位置：首页>python>别再手动画架构图了!Python一行代码自动生成

别再手动画架构图了!Python一行代码自动生成

2026-02-03 23:01:07

你是不是也有这样的困扰？

每次要画系统架构图，都得打开Visio或者Draw.io，一个个拖组件、画连线，折腾半天。

好不容易画完了，架构一改又得重来。

说实话，我之前也是这样。

直到我发现了一个神器——Python的diagrams库，从此告别手动画图。

这个库有多好用？

简单来说，就是用代码生成架构图。

你只需要几行Python代码，就能生成专业的云服务架构图。

不用再手动拖拽组件，不用再调整布局，代码写完，图就出来了。

就像写文档一样自然。

代码就是最好的架构文档。

怎么开始？

先安装两个东西：Python和GraphViz。

安装命令：

# macOS / Linuxsudo apt-get install graphviz        # Ubuntu/Debianbrew install graphviz                # macOS# Windows → 下载 MSI：https://graphviz.org/download/pip install -U diagrams graphviz

验证是否安装成功：

dot -V>>> dot - graphviz version 8.1.0 (20230707.0739)

核心概念速览

diagrams库有几个核心概念，一学就会：

概念	说明	代码示例
`Diagram`	整张图画布	`with Diagram(...)`
`Node`	组件节点	`EC2("web")`
`Cluster`	逻辑分组	`with Cluster("DMZ"):`
`Edge`	连接线	`>>` 或 `-`
`Custom`	自定义图标	`Custom("CDN", "./cdn.png")`

不用记，用多了就熟了。

来个实战例子

咱们画一个微服务架构图，包含API网关、用户服务和订单服务。

代码是这样的：

from diagrams import Diagram, Clusterfrom diagrams.aws.compute import ECSfrom diagrams.aws.network import APIGateway, ELBfrom diagrams.aws.database import RDS, ElastiCachefrom diagrams.aws.storage import S3from diagrams.aws.security import WAFwith Diagram("Microservice Architecture", show=True, direction="LR", filename="micro"):    waf = WAF("WAF")    api = APIGateway("API Gateway")    with Cluster("User Service"):        user_lb = ELB("User-LB")        user_ecs = ECS("User-API")        user_db = RDS("User-DB")        user_cache = ElastiCache("Redis")        user_lb >> user_ecs >> user_cache        user_ecs >> user_db    with Cluster("Order Service"):        order_lb = ELB("Order-LB")        order_ecs = ECS("Order-API")        order_db = RDS("Order-DB")        order_lb >> order_ecs >> order_db    storage = S3("Static/Avatar")    user_ecs >> storage    order_ecs >> storage    waf >> api >> user_lb    waf >> api >> order_lb

运行一下：

python 00_microservice.py# 生成 micro.png（高清，4000×1800）

你看，就这么简单。

一个包含WAF、API网关、两个微服务集群（各自有负载均衡、服务实例、数据库、缓存）以及共享存储的架构图就生成了。

还有这些玩法

事件驱动架构

用EventBridge、SQS和Lambda组合，画一个事件驱动的架构：

from diagrams import Diagramfrom diagrams.aws.compute import Lambdafrom diagrams.aws.database import DynamoDBfrom diagrams.aws.integration import SQS, Eventbridgewith Diagram("Event-Driven", show=True, direction="TB"):    bus = Eventbridge("EventBridge")    queue = SQS("Order Queue")    producer = Lambda("Producer")    consumer = Lambda("Consumer")    db = DynamoDB("Order DB")    producer >> bus >> queue >> consumer >> db    db << producer  # 双向连接

混合云架构

AWS、GCP、Azure三家云服务放在一起：

from diagrams import Diagram, Cluster, Edgefrom diagrams.aws.compute import EC2from diagrams.gcp.compute import ComputeEnginefrom diagrams.azure.compute import VirtualMachinesfrom diagrams.onprem.network import Internetwith Diagram("Hybrid-Cloud", show=True):    inet = Internet("Internet")    with Cluster("AWS"):        aws_vpc = EC2("EC2-VPC")    with Cluster("GCP"):        gcp_vpc = ComputeEngine("GCE-VPC")    with Cluster("Azure"):        az_vpc = VirtualMachines("VM-VNet")    inet >> [aws_vpc, gcp_vpc, az_vpc]    aws_vpc - Edge(label="IPSec VPN") - gcp_vpc    gcp_vpc - Edge(label="专线") - az_vpc

Serverless架构

API Gateway + Cognito + Lambda + DynamoDB：

from diagrams import Diagramfrom diagrams.aws.compute import Lambdafrom diagrams.aws.database import DynamoDBfrom diagrams.aws.network import APIGatewayfrom diagrams.aws.security import Cognitowith Diagram("Serverless Backend", show=True):    api = APIGateway("API")    auth = Cognito("Auth")    func = Lambda("Lambda")    db = DynamoDB("DynamoDB")    api >> auth >> func >> db

还有CI/CD流水线、自定义图标、多层架构等等，玩法太多写不完。