From 6f6bd530d2f877ae5a86bfff5f4617fc072cf7ba Mon Sep 17 00:00:00 2001
From: Raqueebuddin Aziz <raqueebuddin.aziz.314@gmail.com>
Date: Fri, 1 Dec 2023 08:02:30 +0530
Subject: [PATCH] update lifecycle scripts

---
 __init__.py  | 11 +++++++++++
 disable.py   | 16 +++-------------
 enable.py    | 20 +++++---------------
 install.py   | 20 +++++---------------
 uninstall.py | 16 +++-------------
 5 files changed, 27 insertions(+), 56 deletions(-)

diff --git a/__init__.py b/__init__.py
index ebb1123..312e8e3 100644
--- a/__init__.py
+++ b/__init__.py
@@ -6,6 +6,17 @@
 """
 
 from .wildcard import Wildcard
+import sys
+import os
+import folder_paths
+
+sys.path.append(os.path.dirname(__file__))
+
+module_js_directory = os.path.join(os.path.dirname(os.path.realpath(__file__)), "js")
+application_root_directory = os.path.dirname(folder_paths.__file__)
+application_web_extensions_directory = os.path.join(
+    application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
+)
 
 NODE_CLASS_MAPPINGS = {"SimpleWildcard": Wildcard}
 NODE_DISPLAY_NAME_MAPPINGS = {"SimpleWildcard": "SimpleWildcard"}
diff --git a/disable.py b/disable.py
index 6d12d72..21f0b08 100644
--- a/disable.py
+++ b/disable.py
@@ -1,14 +1,4 @@
-try:
-    import os, shutil
-    import folder_paths
+import shutil
+from . import application_web_extensions_directory
 
-    module_js_directory = os.path.join(
-        os.path.dirname(os.path.realpath(__file__)), "js"
-    )
-    application_root_directory = os.path.dirname(folder_paths.__file__)
-    application_web_extensions_directory = os.path.join(
-        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
-    )
-    shutil.rmtree(application_web_extensions_directory)
-except:
-    pass
+shutil.rmtree(application_web_extensions_directory)
diff --git a/enable.py b/enable.py
index 00ac25a..5a1efb1 100644
--- a/enable.py
+++ b/enable.py
@@ -1,16 +1,6 @@
-try:
-    import os, shutil
-    import folder_paths
+import shutil
+from . import application_web_extensions_directory, module_js_directory
 
-    module_js_directory = os.path.join(
-        os.path.dirname(os.path.realpath(__file__)), "js"
-    )
-    application_root_directory = os.path.dirname(folder_paths.__file__)
-    application_web_extensions_directory = os.path.join(
-        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
-    )
-    shutil.copytree(
-        module_js_directory, application_web_extensions_directory, dirs_exist_ok=True
-    )
-except:
-    pass
+shutil.copytree(
+    module_js_directory, application_web_extensions_directory, dirs_exist_ok=True
+)
diff --git a/install.py b/install.py
index 00ac25a..5a1efb1 100644
--- a/install.py
+++ b/install.py
@@ -1,16 +1,6 @@
-try:
-    import os, shutil
-    import folder_paths
+import shutil
+from . import application_web_extensions_directory, module_js_directory
 
-    module_js_directory = os.path.join(
-        os.path.dirname(os.path.realpath(__file__)), "js"
-    )
-    application_root_directory = os.path.dirname(folder_paths.__file__)
-    application_web_extensions_directory = os.path.join(
-        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
-    )
-    shutil.copytree(
-        module_js_directory, application_web_extensions_directory, dirs_exist_ok=True
-    )
-except:
-    pass
+shutil.copytree(
+    module_js_directory, application_web_extensions_directory, dirs_exist_ok=True
+)
diff --git a/uninstall.py b/uninstall.py
index 6d12d72..21f0b08 100644
--- a/uninstall.py
+++ b/uninstall.py
@@ -1,14 +1,4 @@
-try:
-    import os, shutil
-    import folder_paths
+import shutil
+from . import application_web_extensions_directory
 
-    module_js_directory = os.path.join(
-        os.path.dirname(os.path.realpath(__file__)), "js"
-    )
-    application_root_directory = os.path.dirname(folder_paths.__file__)
-    application_web_extensions_directory = os.path.join(
-        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
-    )
-    shutil.rmtree(application_web_extensions_directory)
-except:
-    pass
+shutil.rmtree(application_web_extensions_directory)