Class PD_CaptionJSONParser

py/caption_json_editor.py:184–253

PD Caption JSON解析器 节点 功能：解析JSON格式的caption字符串，提取caption和lang字段

class PD_CaptionJSONParser:
    """
    PD Caption JSON解析器 节点
    功能：解析JSON格式的caption字符串，提取caption和lang字段
    """
    
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "json_string": ("STRING", {
                    "default": "",
                    "multiline": True,
                    "placeholder": '输入JSON字符串',
                    "forceInput": True
                }),
            }
        }
    
    RETURN_TYPES = ("STRING", "STRING")
    RETURN_NAMES = ("caption", "lang")
    FUNCTION = "parse_json"
    CATEGORY = "PDuse/Text"
    
    # 支持列表输入和输出
    INPUT_IS_LIST = {"json_string": True}
    OUTPUT_IS_LIST = (True, True)
    
    def parse_json(self, json_string):
        """
        解析Caption JSON字符串，提取caption和lang
        支持批处理：输入列表，输出列表
        """
        # 确保json_string是列表
        if not isinstance(json_string, list):
            json_string = [json_string]
        
        caption_list = []
        lang_list = []
        
        # 处理每个JSON字符串
        for idx, json_str in enumerate(json_string):
            try:
                # 类型转换
                if isinstance(json_str, list):
                    json_str = str(json_str[0]) if json_str else ""
                if not isinstance(json_str, str):
                    json_str = str(json_str)
                
                # 解析JSON
                data = json.loads(json_str)
                caption = data.get("caption", "")
                lang = data.get("lang", "en")
                
                caption_list.append(caption)
                lang_list.append(lang)
                
            except json.JSONDecodeError as e: