add 'before_hr callback' script callback

This commit is contained in:
hako-mikan
2023-06-28 18:37:08 +09:00
committed by GitHub
parent 394ffa7b0a
commit b0ec69b360
2 changed files with 17 additions and 0 deletions

View File

@@ -186,6 +186,11 @@ class Script:
return f'script_{tabname}{title}_{item_id}'
def before_hr(self, p ,*args):
"""
This function is called before hires fix start.
"""
pass
current_basedir = paths.script_path
@@ -548,6 +553,15 @@ class ScriptRunner:
self.scripts[si].args_to = args_to
def before_hr(self, p):
for script in self.alwayson_scripts:
try:
script_args = p.script_args[script.args_from:script.args_to]
script.before_hr(p, *script_args)
except Exception:
errors.report(f"Error running before_hr: {script.filename}", exc_info=True)
scripts_txt2img: ScriptRunner = None
scripts_img2img: ScriptRunner = None
scripts_postproc: scripts_postprocessing.ScriptPostprocessingRunner = None