From f34635226a7eacc674edd5f0ee0304b2cf83fe5a Mon Sep 17 00:00:00 2001
From: lcolok <425311101@qq.com>
Date: Mon, 28 Oct 2024 14:39:52 +0800
Subject: [PATCH] improve path handling using folder_paths module

---
 __init__.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/__init__.py b/__init__.py
index caf99dd..240d335 100644
--- a/__init__.py
+++ b/__init__.py
@@ -2,22 +2,22 @@
 import glob
 import os
 import sys
-import __main__
 import filecmp
 import shutil
+import folder_paths
 
 NODE_CLASS_MAPPINGS = {}
 NODE_DISPLAY_NAME_MAPPINGS = {}
 
 python = sys.executable
-extentions_folder = os.path.join(os.path.dirname(os.path.realpath(__main__.__file__)),
-                                 "web" + os.sep + "extensions" + os.sep + "dzNodes")
+base_path = os.path.dirname(folder_paths.base_path)
+extentions_folder = os.path.join(base_path, "web", "extensions", "dzNodes")
 javascript_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "js")
 outdate_file_list = ['comfy_shared.js', 'debug.js', 'mtb_widgets.js', 'parse-css.js', 'dz_widgets.js']
 
 if not os.path.exists(extentions_folder):
     print('# 😺dzNodes: Making the "web\extensions\dzNodes" folder')
-    os.mkdir(extentions_folder)
+    os.makedirs(extentions_folder, exist_ok=True)
 else:
     for i in outdate_file_list:
         outdate_file = os.path.join(extentions_folder, i)