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

|
||
|
||
> 直接拖入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 节点的翻译功能相同
|