Method parse_json

py/caption_json_editor.py:212–253

解析Caption JSON字符串，提取caption和lang 支持批处理：输入列表，输出列表

(self, json_string)

    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:
                print(f"PD CaptionJSONParser: 第{idx+1}项JSON解析失败: {e}")
                # 如果解析失败，将整个字符串作为caption返回
                caption_list.append(json_str.strip())
                lang_list.append("en")
            except Exception as e:
                print(f"PD CaptionJSONParser: 解析第{idx+1}项时出错: {e}")
                caption_list.append(f"解析出错: {str(e)}")
                lang_list.append("en")
        
        print(f"PD CaptionJSONParser: 批量解析完成，共处理 {len(caption_list)} 项")
        
        return (caption_list, lang_list)


# 注册节点