mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-04-16 09:59:00 +08:00
eef1f1e0c6
- Handled no negative prompt scenario - Include new lines in positive/negative prompts - Added links to the top to copy png info sections (all/positive/negative/settings)
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
import re
|
|
|
|
class PngParser:
|
|
re_top_level = None
|
|
re_top_level2 = None
|
|
re_extra_newline = None
|
|
re_parameters = None
|
|
|
|
def __init__(self, pnginfo_string):
|
|
PngParser.init_re()
|
|
|
|
self.valid = self.parse_pnginfo(pnginfo_string)
|
|
|
|
def parse_pnginfo(self, pnginfo_string):
|
|
try:
|
|
# separate positive, negative, and parameters
|
|
m = PngParser.re_top_level.search(pnginfo_string)
|
|
if m is None:
|
|
m = PngParser.re_top_level2.search(pnginfo_string)
|
|
if m is None:
|
|
return False
|
|
else:
|
|
self.positive = m.group(1)
|
|
self.negative = None
|
|
self.parameters = m.group(2)
|
|
else:
|
|
self.positive = m.group(1)
|
|
self.negative = m.group(2)
|
|
self.parameters = m.group(3)
|
|
|
|
self.extra = None
|
|
self.settings = None
|
|
|
|
# parse extra parameters (if they exist) by a newline outside of quotes
|
|
m = PngParser.re_extra_newline.search(self.parameters)
|
|
if m is not None:
|
|
s = m.span()
|
|
self.extra = self.parameters[s[1]:]
|
|
self.parameters = self.parameters[:s[0]]
|
|
|
|
# parse standard parameters
|
|
self.settings = PngParser.re_parameters.findall(self.parameters)
|
|
if self.settings is None:
|
|
return False
|
|
except Exception:
|
|
return False
|
|
|
|
return True
|
|
|
|
@classmethod
|
|
def init_re(cls):
|
|
if cls.re_top_level is None:
|
|
cls.re_top_level = re.compile(r'^(?P<positive>(?:.|\n)*)\nNegative prompt: (?P<negative>(?:.|\n)*)\n(?=Steps: )(?P<parameters>(?:.|\n)*)$')
|
|
cls.re_top_level2 = re.compile(r'^(?P<positive>(?:.|\n)*)\nSteps: (?P<parameters>(?:.|\n)*)$')
|
|
# cls.re_top_level2 = re.compile(r'^(?P<positive>(?:.|\n)*)\n(?=Steps: )(?P<parameters>(?:.|\n)*)$')
|
|
cls.re_extra_newline = re.compile(r'\n(?=(?:[^"]*"[^"]*")*[^"]*$)')
|
|
cls.re_parameters = re.compile(r'\s*(?P<param>[^:,]+):\s*(?P<quote>")?(?P<value>(?(2)(?:.)*?(?:(?<!\\)")|.*?))(?:\s*,\s*|$)')
|