Merge d6b8a8cdb1 into 1e26cfedaa

2 months ago · 23b6df550b
parent 1e26cfedaa d6b8a8cdb1
commit 23b6df550b
1 changed files with 6 additions and 2 deletions
--- a/diagrams/init.py
+++ b/diagrams/init.py
@ -431,8 +431,12 @@ class Node:
        return uuid.uuid4().hex

    def _load_icon(self):
-        basedir = Path(os.path.abspath(os.path.dirname(__file__)))
-        return os.path.join(basedir.parent, self._icon_dir, self._icon)
+        if self._icon_dir is None or not self._icon_dir.startswith(os.path.sep):
+            basedir = Path(os.path.abspath(os.path.dirname(__file__)))
+            return os.path.join(basedir.parent, self._icon_dir, self._icon)
+        else:
+            return os.path.join(self._icon_dir, self._icon)
+


 class Edge: