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(+)

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