diff --git a/diagrams/__init__.py b/diagrams/__init__.py index b46fa468..627b40ee 100644 --- a/diagrams/__init__.py +++ b/diagrams/__init__.py @@ -1,5 +1,6 @@ import contextvars import os +import re import uuid from pathlib import Path from typing import List, Union, Dict @@ -279,6 +280,7 @@ class Node: _icon_dir = None _icon = None + _icon_ext = ".png" _height = 1.9 @@ -394,7 +396,15 @@ class Node: else: o.connect(self, Edge(self, reverse=True)) return self - + + def _get_icon(self): + if self._icon: + return self._icon + + icon = re.sub(r'(?