diff options
| author | random-thoughtss <116161560+random-thoughtss@users.noreply.github.com> | 2022-10-30 00:30:18 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-30 00:30:18 -0700 |
| commit | 15468c993974ed30fb204792c9598dfa62088bbb (patch) | |
| tree | 313bbeb9d55c752cacfaa08b769c873c3566ebaf /modules/scripts.py | |
| parent | 39f55c3c35873bc7dd9792cb2155746a1c3d4292 (diff) | |
| parent | 17a2076f72562b428052ee3fc8c43d19c03ecd1e (diff) | |
Merge branch 'AUTOMATIC1111:master' into master
Diffstat (limited to 'modules/scripts.py')
| -rw-r--r-- | modules/scripts.py | 24 |
1 files changed, 21 insertions, 3 deletions
diff --git a/modules/scripts.py b/modules/scripts.py index a7f36012..96e44bfd 100644 --- a/modules/scripts.py +++ b/modules/scripts.py @@ -64,7 +64,16 @@ class Script: def process(self, p, *args):
"""
This function is called before processing begins for AlwaysVisible scripts.
- scripts. You can modify the processing object (p) here, inject hooks, etc.
+ You can modify the processing object (p) here, inject hooks, etc.
+ args contains all values returned by components from ui()
+ """
+
+ pass
+
+ def postprocess(self, p, processed, *args):
+ """
+ This function is called after processing ends for AlwaysVisible scripts.
+ args contains all values returned by components from ui()
"""
pass
@@ -289,13 +298,22 @@ class ScriptRunner: return processed
- def run_alwayson_scripts(self, p):
+ def process(self, p):
for script in self.alwayson_scripts:
try:
script_args = p.script_args[script.args_from:script.args_to]
script.process(p, *script_args)
except Exception:
- print(f"Error running alwayson script: {script.filename}", file=sys.stderr)
+ print(f"Error running process: {script.filename}", file=sys.stderr)
+ print(traceback.format_exc(), file=sys.stderr)
+
+ def postprocess(self, p, processed):
+ for script in self.alwayson_scripts:
+ try:
+ script_args = p.script_args[script.args_from:script.args_to]
+ script.postprocess(p, processed, *script_args)
+ except Exception:
+ print(f"Error running postprocess: {script.filename}", file=sys.stderr)
print(traceback.format_exc(), file=sys.stderr)
def reload_sources(self, cache):
|