init commit
This commit is contained in:
54
utils/io.py
Normal file
54
utils/io.py
Normal file
@@ -0,0 +1,54 @@
|
||||
import json
|
||||
import os
|
||||
import csv
|
||||
|
||||
def load_keywords_from_file(path):
|
||||
"""逐行读取关键词文件,忽略空行,返回列表"""
|
||||
arr = []
|
||||
try:
|
||||
with open(path, 'r', encoding='utf-8') as f:
|
||||
for line in f:
|
||||
s = line.strip()
|
||||
if s:
|
||||
arr.append(s)
|
||||
except Exception:
|
||||
arr = []
|
||||
return arr
|
||||
|
||||
def write_json(path, obj):
|
||||
"""以 UTF-8 写入 JSON,使用非 ASCII 保留与缩进"""
|
||||
with open(path, 'w', encoding='utf-8') as f:
|
||||
json.dump(obj, f, ensure_ascii=False, indent=2)
|
||||
|
||||
def read_json(path):
|
||||
"""读取 JSON 文件,失败时返回空对象"""
|
||||
try:
|
||||
with open(path, 'r', encoding='utf-8') as f:
|
||||
return json.load(f)
|
||||
except Exception:
|
||||
return {}
|
||||
|
||||
def ensure_csv_header(path, headers):
|
||||
"""若 CSV 不存在则创建并写入表头;为空路径直接返回"""
|
||||
if not path:
|
||||
return
|
||||
if not os.path.exists(path):
|
||||
with open(path, 'w', newline='', encoding='utf-8') as wf:
|
||||
w = csv.writer(wf)
|
||||
w.writerow(headers)
|
||||
|
||||
def append_csv_rows(path, rows):
|
||||
"""向 CSV 追加多行,行元素按列表给出;为空路径直接返回"""
|
||||
if not path:
|
||||
return
|
||||
with open(path, 'a', newline='', encoding='utf-8') as af:
|
||||
w = csv.writer(af)
|
||||
for r in rows:
|
||||
w.writerow(r)
|
||||
"""通用 IO 工具
|
||||
|
||||
提供:
|
||||
- 关键词文件加载
|
||||
- JSON 读写
|
||||
- CSV 文件写入(确保表头、追加行)
|
||||
"""
|
||||
Reference in New Issue
Block a user