Corin1998 commited on
Commit
777fd0c
·
verified ·
1 Parent(s): 2a333a1

Update export/qa_csv.py

Browse files
Files changed (1) hide show
  1. export/qa_csv.py +10 -12
export/qa_csv.py CHANGED
@@ -1,17 +1,15 @@
1
  # export/qa_csv.py
 
 
2
  import csv, os
3
- from typing import List, Dict
4
 
5
- def save_qa_csv(qa_list: List[Dict], links: List[str], path: str) -> str:
6
  os.makedirs(os.path.dirname(path), exist_ok=True)
7
- with open(path, "w", newline="", encoding="utf-8-sig") as f:
8
- w = csv.writer(f)
9
  w.writerow(["Q", "A"])
10
- for x in qa_list or []:
11
- w.writerow([x.get("q",""), x.get("a","")])
12
- if links:
13
- w.writerow([])
14
- w.writerow(["参考リンク"])
15
- for u in links:
16
- w.writerow([u])
17
- return path
 
1
  # export/qa_csv.py
2
+ from __future__ import annotations
3
+ from typing import List, Dict, Tuple
4
  import csv, os
 
5
 
6
+ def save_qa_csv(qa_list: List[Dict], links: List[str], path: str) -> None:
7
  os.makedirs(os.path.dirname(path), exist_ok=True)
8
+ with open(path, "w", encoding="utf-8-sig", newline="") as f:
9
+ w = csv.writer(f, quoting=csv.QUOTE_ALL)
10
  w.writerow(["Q", "A"])
11
+ for item in qa_list:
12
+ q = item.get("q") or item.get("Q") or ""
13
+ a = item.get("a") or item.get("A") or ""
14
+ # 改行はCSV内に残してExcelでも読めるようにする(QUOTE_ALLで安全)
15
+ w.writerow([q, a])