# ComfyUI 字符串自定义节点

ComfyUI的字符串相关的自定义节点，其中包括 String Formatter、StringList、StringListFromCSV、StringListToCSV、StringConverter、StringMatcher、TimeFormatter、ShowTranslateString 节点，以提高在处理字符串时的效率和灵活性。

![image](workflow.png)

> 直接拖入ComfyUI即可使用

## 安装方法

1. 将此仓库克隆到你的 `ComfyUI/custom_nodes` 目录下：
```bash
cd custom_nodes
git clone https://github.com/liuqianhonga/ComfyUI-String-Helper.git
```
2. 重启 ComfyUI

## String Formatter 节点使用说明

字符串格式化节点允许你使用Python的f-string语法来格式化字符串。你可以在模板字符串中引用最多10个输入参数（arg1到arg10）且自由定制模板。对于复杂的语法，需要了解Python的f-string格式化语法。

### 节点参数

- **template** (必需)：使用f-string语法的模板字符串
- **arg1** - **arg10** (可选)：可在模板中引用的输入参数

### 使用示例

1. 基础文本格式化：
```python
arg1: "小明"
arg2: "ComfyUI"

模板: "你好 {arg1}，欢迎来到 {arg2}！"
输出: "你好 小明，欢迎来到 ComfyUI！"
```

2. 数字使用示例：
```python
arg1: 10
arg2: 20
arg3: 30

模板: "计算结果: {arg1} + {arg2} = {arg3}"
输出: "计算结果: 10 + 20 = 30"
```

3. 对象使用示例：
```python
arg1: {"名称": "模型_v1", "类型": "checkpoint"}

模板: "模型信息: {arg1}"
输出: "模型信息: {'名称': '模型_v1', '类型': 'checkpoint'}"
```


### 注意事项

- 未提供的参数（None值）将被忽略
- 节点支持任何类型的输入（字符串、数字、列表、字典等）
- 同一个参数可以在模板中多次使用
- 支持Python f-string的所有格式化特性

### 错误处理

如果模板或格式化过程中出现错误，节点将返回错误信息字符串：
```python
当引用了未提供的参数时：
模板: "你好 {arg1}，{arg2}"
arg1: "世界"
输出: "Format Error: name 'arg2' is not defined"
```

## StringList 节点使用说明

StringList 节点允许你灵活地选择和翻译字符串。你可以通过手动输入、随机选择、顺序循环或指定编号选择字符串，并可选地将其翻译为英文。

### 节点参数

- **p1_select_by_numbers** (可选)：最高优先级，通过输入序号（如 "1,3,5"）选择指定位置的字符串
- **p2_select_sequential** (可选)：第二优先级，布尔值，默认为 `False`。启用后，每次调用按顺序返回下一个字符串，到达列表末尾后自动从头开始
- **p3_select_random_count** (必需)：最低优先级，随机选择字符串的数量。特殊值:
  - `-1`：返回所有输入字符串
  - `0`：返回空列表
  - `1-10`：随机选择指定数量的字符串
- **translate_output** (可选)：布尔值，默认为 `False`。启用后，将选定的字符串从自动检测的语言翻译为英文
- **string1** - **string10** (必需)：多行字符串输入字段，用于手动输入字符串
- **string_list** (可选)：额外的字符串列表输入，用于合并外部字符串列表

### 使用示例

1. 按序号选择字符串（最高优先级）:
```python
p1_select_by_numbers: "1,3,5"
string1: "你好"
string2: "世界"
string3: "欢迎"
string4: "使用"
string5: "ComfyUI"
输出: ["你好", "欢迎", "ComfyUI"]
```

2. 顺序循环选择（第二优先级）:
```python
p2_select_sequential: True
string1: "第一个"
string2: "第二个"
string3: "第三个"
第一次输出: ["第一个"]
第二次输出: ["第二个"]
第三次输出: ["第三个"]
第四次输出: ["第一个"]  # 自动从头开始
```

3. 随机选择字符串（最低优先级）:
```python
p3_select_random_count: 3
string1: "你好"
string2: "世界"
string3: "欢迎"
string4: "使用"
string5: "ComfyUI"
输出: 随机选择3个字符串
```

4. 启用翻译:
```python
p1_select_by_numbers: "1,2"
translate_output: True
string1: "你好"
string2: "世界"
输出: ["hello", "world"]
```

## StringListFromCSV 节点使用说明

StringListFromCSV 节点允许你从 CSV 文件中读取和选择字符串。这个节点提供了与 StringList 节点类似的功能，但数据源来自 CSV 文件。节点支持从预定义的模板格式 CSV 文件中读取原始字符串或其翻译版本。

### 节点参数

- **csv_file** (必需)：CSV 文件路径，支持相对路径和绝对路径
- **use_translated** (必需)：布尔值，默认为 `False`。启用后，使用 `translate_string` 列的内容
- **random_select_count** (必需)：随机选择字符串的数量。特殊值:
  - `-1`：返回所有字符串
  - `0`：返回空列表
  - `>0`：随机选择指定数量的字符串
- **selected_numbers** (必需)：指定要选择的字符串编号，多个编号用逗号分隔，如"1,3,5"。当此字段有值时，`random_select_count` 失效
- **translate_output** (必需)：布尔值，默认为 `False`。启用后，将选定的字符串翻译为英文
- **reuse_last_result** (必需)：布尔值，默认为 `False`。控制是否重用上次的随机结果：
  - `False`：每次执行都重新随机选择
  - `True`：保持使用上一次的随机结果
- **string_list** (可选)：额外的字符串列表输入，用于合并外部字符串列表

### CSV 文件格式要求

CSV 文件必须包含以下列：
- `string` 列：原始字符串
- `translate_string` 列：对应的中文翻译
- 文件编码支持：UTF-8（推荐）、GBK、GB2312、GB18030、BIG5

### 使用示例

1. 随机选择并保持结果：
```python
csv_file: "template/string_list.csv"
random_select_count: 3
reuse_last_result: True  # 保持使用上次的随机结果
```

2. 使用翻译版本：
```python
csv_file: "template/string_list.csv"
use_translated: True     # 使用 translate_string 列的内容
random_select_count: 2
reuse_last_result: False # 每次执行重新随机
```

3. 获取新的随机结果并保持：
```python
# 步骤 1：获取新的随机结果
reuse_last_result: False # 临时设为 False 获取新的随机结果

# 步骤 2：保持新的随机结果
reuse_last_result: True  # 改回 True 保持新的结果
```

### 注意事项

- CSV 文件必须包含 `string` 和 `translate_string` 两列
- 当 `reuse_last_result` 为 True 时，节点会保持使用上一次的随机结果，直到你将其设为 False 以获取新的随机结果
- 如果 CSV 文件不存在或格式不正确，节点将返回空列表

## String List To CSV 节点使用说明

String List To CSV 节点允许你将字符串列表保存到 CSV 文件中，可选择是否翻译字符串以及是追加还是覆写文件。生成的 CSV 文件遵循与模板文件相同的格式。

### 节点参数

必需参数：
- **csv_file**：CSV 文件保存路径，默认为 `output/string_list_output.csv`
- **translate**：布尔值，默认为 `False`。设置为 `True` 时会将输入的字符串翻译为中文，作为 `translate_string` 列的值
- **append_mode**：布尔值，默认为 `True`。设置为 `True` 时会追加到现有文件，否则会覆写文件

可选参数：
- **string**：单个输入字符串
- **string_list**：字符串列表输入。如果同时提供了 string 和 string_list，两者会被合并处理

### 返回值

- **processed_strings**：成功写入到 CSV 文件的字符串列表
- **skipped_strings**：在追加模式下，由于已存在而被跳过的字符串列表

### 使用示例

1. 使用单个字符串输入：
```python
string: "Hello World"
csv_file: "output/my_strings.csv"
translate: True
append_mode: False
输出文件内容:
string,translate_string
Hello World,你好世界
返回值:
processed_strings: ["Hello World"]
skipped_strings: []
```

2. 使用字符串列表输入：
```python
string_list: ["Hello World", "Good Morning"]
csv_file: "output/my_strings.csv"
translate: True
append_mode: False
输出文件内容:
string,translate_string
Hello World,你好世界
Good Morning,早安
返回值:
processed_strings: ["Hello World", "Good Morning"]
skipped_strings: []
```

3. 同时使用单个字符串和字符串列表：
```python
string: "Hello World"
string_list: ["Good Morning", "Artificial Intelligence"]
csv_file: "output/my_strings.csv"
translate: True
append_mode: False
输出文件内容:
string,translate_string
Good Morning,早安
Artificial Intelligence,人工智能
Hello World,你好世界
返回值:
processed_strings: ["Good Morning", "Artificial Intelligence", "Hello World"]
skipped_strings: []
```

### 注意事项

- 如果使用相对路径，文件会保存在项目根目录下
- 如果目标目录不存在，会自动创建
- 追加模式下，会自动检查并跳过已存在的字符串，避免重复
- 如果不启用翻译，`translate_string` 列将为空
- 如果没有提供任何输入（string 和 string_list 都为空），节点将返回空列表
- 如果提供的字符串列表为空，节点也将返回空列表

## String Matcher 节点使用说明

String Matcher 节点用于从一组条件匹配规则中找到与输入值匹配的结果。每个条件规则都是以 "条件:值" 的格式定义的。

### 节点参数

必需参数：
- **condition_list**：多行文本，每行包含一个匹配规则，格式为 "条件:值"
- **target_type**：返回值的类型，可选值：
  - STRING：字符串（默认）
  - INT：整数
  - FLOAT：浮点数
  - BOOL：布尔值
  - LIST：列表（逗号分隔）
  - DICT：字典（JSON格式）

可选参数：
- **match_value**：要匹配的值，支持任意类型（会被转换为字符串进行匹配）
- **default_value**：当没有找到匹配项时返回的默认值，默认为空字符串

### 功能特点:
- 支持条件列表匹配
- 返回第一个匹配条件对应的值

### 正则表达式支持:
条件部分现在支持正则表达式模式，实现更灵活的匹配：
```
# 基础模式
.*蓝色.*:包含蓝色
\d+:包含数字
^测试:以测试开头
\.png$:以.png结尾

# 常用场景
.*LoRA.*:LoRA模型
E:\\.*:Windows路径
.*\.safetensors$:Safetensors文件
^v\d+\.\d+:版本号
[Bb]atch.*:以batch或Batch开头
```

### 示例:

基础匹配：
```
红色:这是红色
蓝色:这是蓝色
绿色:这是绿色
```

正则表达式匹配：
```
# 文件路径
.*\\train\\.*:训练数据
.*\\test\\.*:测试数据

# 文件类型
.*\.(jpg|png|webp)$:图片文件
.*\pth$:PyTorch模型

# 版本模式
v\d\.\d\.\d:语义版本
beta.*:测试版本
```

类型转换：
```
true:1
false:0
```

### 输出类型:
- STRING (默认)
- INT (整数)
- FLOAT (浮点数)
- BOOL (布尔值)
- LIST (逗号分隔的列表)
- DICT (JSON格式的字典)

## TimeFormatter 节点使用说明

TimeFormatter 节点允许你使用Python的日期时间格式化语法来格式化当前时间。你可以自由定制输出格式，支持年、月、日、时、分、秒等时间组件。

### 节点参数

- **format_string** (必需)：使用Python的strftime格式化语法的模板字符串，默认为 "%Y-%m-%d %H:%M:%S"

### 使用示例

1. 基础日期格式化：
```python
format_string: "%Y-%m-%d"
输出: "2024-12-21"  # 仅显示年月日
```

2. 完整时间格式化：
```python
format_string: "%Y-%m-%d %H:%M:%S"
输出: "2024-12-21 23:35:08"  # 显示年月日时分秒
```

3. 自定义格式：
```python
format_string: "现在是%Y年%m月%d日 %H点%M分"
输出: "现在是2024年12月21日 23点35分"
```

### 常用格式代码

- %Y：四位数的年份（如 2024）
- %m：两位数的月份（01-12）
- %d：两位数的日期（01-31）
- %H：24小时制的小时（00-23）
- %M：分钟（00-59）
- %S：秒（00-59）

## StringConverter 节点使用说明

StringConverter 节点允许你将字符串转换为其他数据类型。支持多种常用数据类型的转换，并会在转换失败时抛出异常。

### 节点参数

- **input_string** (必需)：要转换的输入字符串
- **target_type** (必需)：目标数据类型，可选值:
  - INT：转换为整数
  - FLOAT：转换为浮点数
  - BOOL：转换为布尔值
  - LIST：转换为列表（逗号分隔）
  - DICT：转换为字典（JSON格式）

### 使用示例

1. 转换为整数：
```python
input_string: "123"
target_type: "INT"
输出: 123
```

2. 转换为浮点数：
```python
input_string: "123.45"
target_type: "FLOAT"
输出: 123.45
```

3. 转换为布尔值：
```python
input_string: "true"  # 或 "1", "yes", "y", "on"
target_type: "BOOL"
输出: True
```

4. 转换为列表：
```python
input_string: "a, b, c"
target_type: "LIST"
输出: ["a", "b", "c"]
```

5. 转换为字典：
```python
input_string: '{"name": "test", "value": 123}'
target_type: "DICT"
输出: {"name": "test", "value": 123}
```

### 错误处理

- 如果转换失败，节点会抛出异常并显示详细的错误信息
- 不会返回默认值，确保数据的准确性
- 对于LIST类型，空字符串会返回空列表
- 对于DICT类型，输入必须是有效的JSON格式字符串

## String Translate 节点使用说明

String Translate 节点提供单独的字符串翻译功能，可以将输入的字符串翻译为英文。

### 节点参数

- **string** (必需)：要翻译的字符串
- **translate_output** (必需)：布尔值，默认为 `False`。控制是否执行翻译:
  - `False`：直接返回原字符串
  - `True`：将字符串翻译为英文

### 使用示例

1. 基础翻译:
```python
string: "你好，世界"
translate_output: True
输出: "Hello, world"
```

2. 不执行翻译:
```python
string: "你好，世界"
translate_output: False
输出: "你好，世界"
```

### 特点

- 支持自动语言检测
- 使用必应翻译服务
- 适合单个字符串的快速翻译
- 与 StringList/StringListFromCSV 节点的翻译功能相同