Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-30 19:21:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: reduce complexity (C901)

Browse Source
This commit is contained in:
Bingsu
2023-10-07 21:53:20 +09:00
parent 44453dcff1
commit e682eae794
1 changed files with 23 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
scripts/!adetailer.py
View File

@@ -5,7 +5,7 @@ import platform
import re import re
import sys import sys
import traceback import traceback
from contextlib import contextmanager from contextlib import contextmanager, suppress
from copy import copy, deepcopy from copy import copy, deepcopy
from functools import partial from functools import partial
from pathlib import Path from pathlib import Path
@@ -366,9 +366,23 @@ class AfterDetailerScript(scripts.Script):
) )
def write_params_txt(self, p) -> None: def write_params_txt(self, p) -> None:
i = p._ad_idx
lenp = len(p.all_prompts)
if i % lenp != lenp - 1:
return
prev = None
if hasattr(p, "_ad_orig_steps"):
prev = p.steps
p.steps = p._ad_orig_steps
infotext = self.infotext(p) infotext = self.infotext(p)
params_txt = Path(data_path, "params.txt") params_txt = Path(data_path, "params.txt")
params_txt.write_text(infotext, encoding="utf-8") with suppress(Exception):
params_txt.write_text(infotext, encoding="utf-8")
if hasattr(p, "_ad_orig_steps"):
p.steps = prev
def script_filter(self, p, args: ADetailerArgs): def script_filter(self, p, args: ADetailerArgs):
script_runner = copy(p.scripts) script_runner = copy(p.scripts)
@@ -546,12 +560,16 @@ class AfterDetailerScript(scripts.Script):
@staticmethod @staticmethod
def need_call_process(p) -> bool: def need_call_process(p) -> bool:
if p.scripts is None:
return False
i = p._ad_idx i = p._ad_idx
bs = p.batch_size bs = p.batch_size
return i % bs == bs - 1 return i % bs == bs - 1
@staticmethod @staticmethod
def need_call_postprocess(p) -> bool: def need_call_postprocess(p) -> bool:
if p.scripts is None:
return False
i = p._ad_idx i = p._ad_idx
bs = p.batch_size bs = p.batch_size
return i % bs == 0 return i % bs == 0
@@ -668,15 +686,11 @@ class AfterDetailerScript(scripts.Script):
if getattr(p, "_ad_disabled", False) or not self.is_ad_enabled(*args_): if getattr(p, "_ad_disabled", False) or not self.is_ad_enabled(*args_):
return return
if hasattr(p, "_ad_orig_steps"):
p.steps = p._ad_orig_steps
del p._ad_orig_steps
p._ad_idx = getattr(p, "_ad_idx", -1) + 1 p._ad_idx = getattr(p, "_ad_idx", -1) + 1
init_image = copy(pp.image) init_image = copy(pp.image)
arg_list = self.get_args(p, *args_) arg_list = self.get_args(p, *args_)
if p.scripts is not None and self.need_call_postprocess(p): if self.need_call_postprocess(p):
dummy = Processed(p, [], p.seed, "") dummy = Processed(p, [], p.seed, "")
with preseve_prompts(p): with preseve_prompts(p):
p.scripts.postprocess(copy(p), dummy) p.scripts.postprocess(copy(p), dummy)
@@ -693,17 +707,11 @@ class AfterDetailerScript(scripts.Script):
p, init_image, condition="ad_save_images_before", suffix="-ad-before" p, init_image, condition="ad_save_images_before", suffix="-ad-before"
) )
if p.scripts is not None and self.need_call_process(p): if self.need_call_process(p):
with preseve_prompts(p): with preseve_prompts(p):
p.scripts.process(copy(p)) p.scripts.process(copy(p))
try: self.write_params_txt(p)
ia = p._ad_idx
lenp = len(p.all_prompts)
if ia % lenp == lenp - 1:
self.write_params_txt(p)
except Exception:
pass
def on_after_component(component, **_kwargs): def on_after_component(component, **_kwargs):