Use _icon_dir as absolute path when it starts with /

5 years ago · d6b8a8cdb1
parent 8768d6888a
commit d6b8a8cdb1
1 changed files with 6 additions and 2 deletions
--- a/diagrams/init.py
+++ b/diagrams/init.py
@ -414,8 +414,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: