54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
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 文件写入（确保表头、追加行）
|
||
""" |