Method edit_caption_json

py/caption_json_editor.py:66–181

编辑Caption JSON字符串 支持批处理：输入列表，输出列表

(self, json_string, new_caption="", add_prefix="", 
                         add_suffix="", replace_from="", replace_to="", new_lang="en")

    OUTPUT_IS_LIST = (True, True, True)
    
    def edit_caption_json(self, json_string, new_caption="", add_prefix="", 
                         add_suffix="", replace_from="", replace_to="", new_lang="en"):
        """
        编辑Caption JSON字符串
        支持批处理：输入列表，输出列表
        """
        # 确保json_string是列表
        if not isinstance(json_string, list):
            json_string = [json_string]
        
        json_output_list = []
        caption_list = []
        lang_list = []
        
        # 解析替换词列表（只解析一次，应用到所有项）
        from_words = []
        to_words = []
        if replace_from and (isinstance(replace_from, str) and replace_from.strip()):
            from_words = [line.strip() for line in replace_from.strip().split('\n') if line.strip()]
            to_words = [line.strip() for line in replace_to.strip().split('\n') if line.strip()] if replace_to else []
            # 确保替换列表长度一致
            while len(to_words) < len(from_words):
                to_words.append("")  # 如果替换词不够，用空字符串替换（即删除）
        
        # 处理每个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字符串
                try:
                    data = json.loads(json_str)
                    caption = data.get("caption", "")
                    lang = data.get("lang", "en")
                except json.JSONDecodeError as e:
                    print(f"PD CaptionJSONEditor: 第{idx+1}项JSON解析失败: {e}")
                    # 尝试直接作为普通文本处理
                    caption = json_str.strip()
                    lang = "en"
                
                # 处理caption内容
                # 1. 如果提供了新的caption，则完全替换
                if new_caption:
                    if isinstance(new_caption, list):
                        current_new_caption = new_caption[idx] if idx < len(new_caption) else ""
                    else:
                        current_new_caption = new_caption
                    
                    if isinstance(current_new_caption, str) and current_new_caption.strip():
                        caption = current_new_caption.strip()
                
                # 2. 执行词语替换
                for from_word, to_word in zip(from_words, to_words):
                    if from_word: