mirror of
https://github.com/vladmandic/automatic
synced 2026-08-30 09:01:01 +02:00
major refactoring of modules
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from modules import scripts, scripts_postprocessing, shared
|
||||
from modules import scripts_manager, scripts_postprocessing, shared
|
||||
|
||||
|
||||
class ScriptPostprocessingForMainUI(scripts.Script):
|
||||
class ScriptPostprocessingForMainUI(scripts_manager.Script):
|
||||
def __init__(self, script_postproc):
|
||||
self.script: scripts_postprocessing.ScriptPostprocessing = script_postproc
|
||||
self.postprocessing_controls = None
|
||||
@@ -10,7 +10,7 @@ class ScriptPostprocessingForMainUI(scripts.Script):
|
||||
return self.script.name
|
||||
|
||||
def show(self, is_img2img): # pylint: disable=unused-argument
|
||||
return scripts.AlwaysVisible
|
||||
return scripts_manager.AlwaysVisible
|
||||
|
||||
def ui(self, is_img2img): # pylint: disable=unused-argument
|
||||
self.postprocessing_controls = self.script.ui()
|
||||
@@ -28,9 +28,9 @@ class ScriptPostprocessingForMainUI(scripts.Script):
|
||||
def create_auto_preprocessing_script_data():
|
||||
res = []
|
||||
for name in shared.opts.postprocessing_enable_in_main_ui:
|
||||
script = next(iter([x for x in scripts.postprocessing_scripts_data if x.script_class.name == name]), None)
|
||||
script = next(iter([x for x in scripts_manager.postprocessing_scripts_data if x.script_class.name == name]), None)
|
||||
if script is None:
|
||||
continue
|
||||
constructor = lambda s=script: ScriptPostprocessingForMainUI(s.script_class()) # pylint: disable=unnecessary-lambda-assignment
|
||||
res.append(scripts.ScriptClassData(script_class=constructor, path=script.path, basedir=script.basedir, module=script.module))
|
||||
res.append(scripts_manager.ScriptClassData(script_class=constructor, path=script.path, basedir=script.basedir, module=script.module))
|
||||
return res
|
||||
|
||||
Reference in New Issue
Block a user