From 4c2478a68a4f11959fe4887d38e0436eac19f97e Mon Sep 17 00:00:00 2001
From: DepFA <35278260+dfaker@users.noreply.github.com>
Date: Sat, 1 Oct 2022 18:30:53 +0100
Subject: add script reload method

---
 modules/scripts.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'modules/scripts.py')

diff --git a/modules/scripts.py b/modules/scripts.py
index 7c3bd5e7..3c14b9e3 100644
--- a/modules/scripts.py
+++ b/modules/scripts.py
@@ -165,3 +165,12 @@ class ScriptRunner:
 
 scripts_txt2img = ScriptRunner()
 scripts_img2img = ScriptRunner()
+
+def reload_scripts(basedir):
+  global scripts_txt2img,scripts_img2img
+
+  scripts_data.clear()
+  load_scripts(basedir)
+
+  scripts_txt2img = ScriptRunner()
+  scripts_img2img = ScriptRunner()
-- 
cgit v1.2.1


From cf33268d686986a24f2e04eb615f01ed53bfe308 Mon Sep 17 00:00:00 2001
From: DepFA <35278260+dfaker@users.noreply.github.com>
Date: Sun, 2 Oct 2022 01:18:42 +0100
Subject: add script body only refresh

---
 modules/scripts.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'modules/scripts.py')

diff --git a/modules/scripts.py b/modules/scripts.py
index 3c14b9e3..788397f5 100644
--- a/modules/scripts.py
+++ b/modules/scripts.py
@@ -162,10 +162,33 @@ class ScriptRunner:
 
         return processed
 
+    def reload_sources(self):
+      for si,script in list(enumerate(self.scripts)):
+        with open(script.filename, "r", encoding="utf8") as file:
+            args_from = script.args_from
+            args_to   = script.args_to 
+            filename  = script.filename
+            text = file.read()
+
+            from types import ModuleType
+            compiled = compile(text, filename, 'exec')
+            module = ModuleType(script.filename)
+            exec(compiled, module.__dict__)
+
+            for key, script_class in module.__dict__.items():
+                if type(script_class) == type and issubclass(script_class, Script):
+                    self.scripts[si] = script_class()
+                    self.scripts[si].filename = filename
+                    self.scripts[si].args_from = args_from
+                    self.scripts[si].args_to = args_to
 
 scripts_txt2img = ScriptRunner()
 scripts_img2img = ScriptRunner()
 
+def reload_script_body_only():
+  scripts_txt2img.reload_sources()
+  scripts_img2img.reload_sources()
+
 def reload_scripts(basedir):
   global scripts_txt2img,scripts_img2img
 
-- 
cgit v1.2.1


From e808096cf641d868f88465515d70d40fc46125d4 Mon Sep 17 00:00:00 2001
From: DepFA <35278260+dfaker@users.noreply.github.com>
Date: Sun, 2 Oct 2022 19:26:06 +0100
Subject: correct indent

---
 modules/scripts.py | 48 +++++++++++++++++++++++++-----------------------
 1 file changed, 25 insertions(+), 23 deletions(-)

(limited to 'modules/scripts.py')

diff --git a/modules/scripts.py b/modules/scripts.py
index 788397f5..45230f9a 100644
--- a/modules/scripts.py
+++ b/modules/scripts.py
@@ -163,37 +163,39 @@ class ScriptRunner:
         return processed
 
     def reload_sources(self):
-      for si,script in list(enumerate(self.scripts)):
-        with open(script.filename, "r", encoding="utf8") as file:
-            args_from = script.args_from
-            args_to   = script.args_to 
-            filename  = script.filename
-            text = file.read()
+        for si, script in list(enumerate(self.scripts)):
+            with open(script.filename, "r", encoding="utf8") as file:
+                args_from = script.args_from
+                args_to = script.args_to
+                filename = script.filename
+                text = file.read()
 
-            from types import ModuleType
-            compiled = compile(text, filename, 'exec')
-            module = ModuleType(script.filename)
-            exec(compiled, module.__dict__)
+                from types import ModuleType
 
-            for key, script_class in module.__dict__.items():
-                if type(script_class) == type and issubclass(script_class, Script):
-                    self.scripts[si] = script_class()
-                    self.scripts[si].filename = filename
-                    self.scripts[si].args_from = args_from
-                    self.scripts[si].args_to = args_to
+                compiled = compile(text, filename, 'exec')
+                module = ModuleType(script.filename)
+                exec(compiled, module.__dict__)
+
+                for key, script_class in module.__dict__.items():
+                    if type(script_class) == type and issubclass(script_class, Script):
+                        self.scripts[si] = script_class()
+                        self.scripts[si].filename = filename
+                        self.scripts[si].args_from = args_from
+                        self.scripts[si].args_to = args_to
 
 scripts_txt2img = ScriptRunner()
 scripts_img2img = ScriptRunner()
 
 def reload_script_body_only():
-  scripts_txt2img.reload_sources()
-  scripts_img2img.reload_sources()
+    scripts_txt2img.reload_sources()
+    scripts_img2img.reload_sources()
+
 
 def reload_scripts(basedir):
-  global scripts_txt2img,scripts_img2img
+    global scripts_txt2img, scripts_img2img
 
-  scripts_data.clear()
-  load_scripts(basedir)
+    scripts_data.clear()
+    load_scripts(basedir)
 
-  scripts_txt2img = ScriptRunner()
-  scripts_img2img = ScriptRunner()
+    scripts_txt2img = ScriptRunner()
+    scripts_img2img = ScriptRunner()
-- 
cgit v1.2.1