From 7baa3ed4a981c84d259062bdb9614f233d7c5520 Mon Sep 17 00:00:00 2001 From: AbyssYuan0 Date: Tue, 28 Nov 2023 18:01:56 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=87=E6=9C=AC=E5=88=97=E8=A1=A8=E8=BD=AC?= =?UTF-8?q?=E6=95=B4=E4=B8=AA=E6=96=87=E6=9C=AC=E4=B8=B2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- __init__.py | 33 +++++++++++++++++++++++++++++---- 1 file changed, 29 insertions(+), 4 deletions(-) diff --git a/__init__.py b/__init__.py index 7ee2fdc..ef33a01 100644 --- a/__init__.py +++ b/__init__.py @@ -317,7 +317,7 @@ class ImageScaleToSide: cls = cls.movedim(1, -1) return (cls,) -class NovelToFizz: +class StringToFizz: def __init__(self): pass @@ -327,10 +327,10 @@ class NovelToFizz: RETURN_TYPES = ("STRING", "INT",) - FUNCTION = "novelToFizz" + FUNCTION = "stringToFizz" CATEGORY = "badger" - def novelToFizz(self, text): + def stringToFizz(self, text): textA = text.split("\n") lines = 0 outText = "" @@ -342,6 +342,30 @@ class NovelToFizz: outText = outText[:-2] return (outText, lines, ) +class TextListToString: + def __init__(self): + pass + + @classmethod + def INPUT_TYPES(s): + return {"required": {"texts": ("STRING", {"multiline": True})}} + + RETURN_TYPES = ("STRING",) + + INPUT_IS_LIST = True + + FUNCTION = "textListToString" + CATEGORY = "badger" + + def textListToString(self, texts): + fullString = "" + if len(texts) <= 1: + return (texts, ) + else: + for text in texts: + fullString += text + "\n" + return (fullString, ) + NODE_CLASS_MAPPINGS = { @@ -351,7 +375,8 @@ NODE_CLASS_MAPPINGS = { "FloatToString-badger": FloatToString, "ImageNormalization-badger": ImageNormalization, "ImageScaleToSide-badger": ImageScaleToSide, - "NovelToFizz-badger": NovelToFizz + "StringToFizz-badger": StringToFizz, + "TextListToString-badger": TextListToString } NODE_DISPLAY_NAME_MAPPINGS = {