177 lines
5.5 KiB
Python
Executable File
177 lines
5.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
"""
|
||
将符合条件的CSV转为符合Payload需要的TOML格式
|
||
|
||
使用命令: python3 csv2payload.py <CSV路径> <生成TOML路径, 默认为文件名相同, 后缀为toml的TOML文件> [-r: 可选参数, 表示按照索引打乱顺序的随机整数种子]
|
||
|
||
转换规则:
|
||
1. `ident` 列用作 TOML 的 section 标题(`[ident]`)
|
||
2. 若某行的 `ident` 为空，则自动按顺序生成标识符，例如 `idx_1`、`idx_2` 等
|
||
3. 所有其他列(除 `ident` 外)都作为该 section 下的键值对
|
||
4. 所有列都是可选的，但 `ident` 为空时会自动生成
|
||
|
||
示例 CSV:
|
||
```csv
|
||
ident, content, meaning, ...
|
||
"Fox", "Fox", "狐狸(一种动物)", ...
|
||
"Dog", "Dog", "狗(一种比猫聪明的动物)", ...
|
||
"Cat", "Cat", "猫(一种不如狗聪明的动物)", ...
|
||
"Dolphin", "Dolphin", "一种很聪明的海洋哺乳动物", ...
|
||
, "Duck", "一种扁嘴水禽"
|
||
, "Meow", "猫发出的声音"
|
||
"Doge", "Doge", "神烦狗(一张搞笑狗狗表情包的代称)", ...
|
||
, "Woof", "狗发出的声音"
|
||
```
|
||
|
||
转换后的 TOML:
|
||
```toml
|
||
[Fox]
|
||
content = "Fox"
|
||
meaning = "狐狸(一种动物)"
|
||
|
||
[Dog]
|
||
content = "Dog"
|
||
meaning = "狗(一种比猫聪明的动物)"
|
||
|
||
[Cat]
|
||
content = "Cat"
|
||
meaning = "猫(一种不如狗聪明的动物)"
|
||
|
||
[Dolphin]
|
||
content = "Dolphin"
|
||
meaning = "一种很聪明的海洋哺乳动物"
|
||
|
||
[idx_1]
|
||
content = "Duck"
|
||
meaning = "一种扁嘴水禽"
|
||
|
||
[idx_2]
|
||
content = "Meow"
|
||
meaning = "猫发出的声音"
|
||
|
||
[Doge]
|
||
content = "Doge"
|
||
meaning = "神烦狗(一张搞笑狗狗表情包的代称)"
|
||
|
||
[idx_3]
|
||
content = "Woof"
|
||
meaning = "狗发出的声音"
|
||
```
|
||
|
||
补充说明：
|
||
- 自动生成的标识符使用 `idx_` 前缀加数字序列
|
||
- 生成序列基于原始 CSV 中 `ident` 为空的行出现的顺序
|
||
- 所有值都保留为字符串类型，符合 TOML 字符串格式要求
|
||
- 如果 CSV 包含更多列，它们也会以相同方式转换为键值对
|
||
- 支持 `-r` 参数指定随机种子来打乱 section 顺序
|
||
"""
|
||
import csv
|
||
import sys
|
||
import os
|
||
import random
|
||
import argparse
|
||
from pathlib import Path
|
||
|
||
def csv_to_toml(csv_path, toml_path=None, random_seed=None):
|
||
"""
|
||
将CSV文件转换为TOML格式
|
||
|
||
Args:
|
||
csv_path (str): 输入CSV文件路径
|
||
toml_path (str): 输出TOML文件路径，默认为相同目录下同名文件
|
||
random_seed (int): 随机种子，用于打乱section顺序，None表示不打乱
|
||
"""
|
||
# 检查CSV文件是否存在
|
||
csv_file = Path(csv_path)
|
||
if not csv_file.exists():
|
||
print(f"错误: CSV文件不存在 - {csv_path}")
|
||
sys.exit(1)
|
||
|
||
# 确定输出TOML文件路径
|
||
if toml_path is None:
|
||
toml_path = csv_file.with_suffix('.toml')
|
||
else:
|
||
toml_path = Path(toml_path)
|
||
|
||
# 读取CSV文件
|
||
try:
|
||
with open(csv_file, 'r', encoding='utf-8') as f:
|
||
reader = csv.DictReader(f)
|
||
rows = list(reader)
|
||
except Exception as e:
|
||
print(f"错误: 无法读取CSV文件 - {e}")
|
||
sys.exit(1)
|
||
|
||
# 检查CSV文件是否有数据
|
||
if not rows:
|
||
print("错误: CSV文件为空或格式不正确")
|
||
sys.exit(1)
|
||
|
||
# 如果指定了随机种子，设置随机种子并打乱行顺序
|
||
if random_seed is not None:
|
||
random.seed(random_seed)
|
||
random.shuffle(rows)
|
||
print(f"提示: 使用随机种子 {random_seed} 打乱了 section 顺序")
|
||
|
||
# 生成TOML内容
|
||
toml_content = []
|
||
idx_counter = 1
|
||
|
||
for row in rows:
|
||
# 处理ident列，为空时生成自动标识符
|
||
ident = row.get('ident', '').strip()
|
||
if not ident:
|
||
ident = f"idx_{idx_counter}"
|
||
idx_counter += 1
|
||
|
||
# 添加section标题
|
||
toml_content.append(f"[{ident}]")
|
||
|
||
# 添加所有其他列作为键值对（排除ident列）
|
||
for key, value in row.items():
|
||
if key == 'ident':
|
||
continue
|
||
|
||
# 确保值存在且不为空
|
||
if value is not None and str(value).strip() != '':
|
||
# 转义特殊字符并添加引号
|
||
escaped_value = str(value).replace('"', '\\"')
|
||
toml_content.append(f'"{key}" = "{escaped_value}"')
|
||
|
||
# section之间添加空行
|
||
toml_content.append("")
|
||
|
||
# 写入TOML文件
|
||
try:
|
||
with open(toml_path, 'w', encoding='utf-8') as f:
|
||
f.write('\n'.join(toml_content).strip())
|
||
print(f"成功: 已生成TOML文件 - {toml_path}")
|
||
except Exception as e:
|
||
print(f"错误: 无法写入TOML文件 - {e}")
|
||
sys.exit(1)
|
||
|
||
def main():
|
||
"""主函数"""
|
||
parser = argparse.ArgumentParser(
|
||
description='将CSV文件转换为TOML格式，支持随机打乱section顺序',
|
||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||
epilog='''
|
||
示例:
|
||
%(prog)s input.csv output.toml
|
||
%(prog)s input.csv # 自动生成input.toml
|
||
%(prog)s input.csv -r 42 # 使用种子42打乱顺序
|
||
%(prog)s input.csv -r 123 output.toml # 指定种子和输出路径
|
||
'''
|
||
)
|
||
|
||
parser.add_argument('csv_path', help='输入的CSV文件路径')
|
||
parser.add_argument('toml_path', nargs='?', help='输出的TOML文件路径，默认为CSV同名文件')
|
||
parser.add_argument('-r', '--random-seed', type=int,
|
||
help='随机种子，用于打乱TOML section的顺序')
|
||
|
||
args = parser.parse_args()
|
||
|
||
csv_to_toml(args.csv_path, args.toml_path, args.random_seed)
|
||
|
||
if __name__ == "__main__":
|
||
main() |