Update pipelines/anonymize.py

pipelines/anonymize.py

@@ -2,63 +2,47 @@ import re
 from typing import Tuple, Dict
 from reportlab.lib.pagesizes import A4
 from reportlab.pdfgen import canvas
-from reportlab.lib.units import mm
-import io
+from io import BytesIO
 
-# 簡易匿名化（メール・電話・氏名行のマスク）
+# ごく簡易な匿名化（メール/電話/氏名候補っぽい行）
 EMAIL_RE = re.compile(r"[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Za-z]{2,}")
 PHONE_RE = re.compile(r"(?:\+?\d{1,3}[ -]?)?(?:\(\d{2,4}\)[ -]?)?\d{2,4}[ -]?\d{2,4}[ -]?\d{3,4}")
-NAME_LINE_RE = re.compile(r"^(?:氏名|Name)[:：]?\s*([^\n\r]+)$", re.MULTILINE)
+NAME_HINT = re.compile(r"^\s*(氏名|Name)\s*[:：]?\s*(.+)$")
 
 def anonymize_text(text: str) -> Tuple[str, Dict[str, str]]:
     mapping = {}
-    # メール
     def _mask_email(m):
-        key = f"EMAIL_{len(mapping)+1}"
-        mapping[m.group(0)] = key
-        return key + "@masked"
-    text = EMAIL_RE.sub(_mask_email, text)
-
-    # 電話
+        val = m.group(0); k = f"EMAIL_{len(mapping)+1}"
+        mapping[val] = k; return k + "@masked"
     def _mask_phone(m):
-        ph = m.group(0)
-        # 短すぎるノイズはスキップ
-        if len(re.sub(r"\D", "", ph)) < 7:
-            return ph
-        key = f"TEL_{len(mapping)+1}"
-        mapping[ph] = key
-        return key
-    text = PHONE_RE.sub(_mask_phone, text)
-
-    # 氏名行
-    def _mask_name(m):
-        val = m.group(1).strip()
-        key = f"NAME_{len(mapping)+1}"
-        mapping[val] = key
-        return m.group(0).replace(val, key)
-    text = NAME_LINE_RE.sub(_mask_name, text)
+        val = m.group(0); k = f"TEL_{len(mapping)+1}"
+        mapping[val] = k; return k
 
-    return text, mapping
+    # line単位で氏名候補をマスク
+    lines = []
+    for line in text.splitlines():
+        nm = NAME_HINT.search(line)
+        if nm:
+            full = nm.group(0)
+            mapping[full] = "NAME_MASKED"
+            lines.append("氏名: NAME_MASKED")
+        else:
+            lines.append(line)
+    masked = "\n".join(lines)
+    masked = EMAIL_RE.sub(_mask_email, masked)
+    masked = PHONE_RE.sub(_mask_phone, masked)
+    return masked, mapping
 
 def render_anonymized_pdf(text: str) -> bytes:
-    buf = io.BytesIO()
+    buf = BytesIO()
     c = canvas.Canvas(buf, pagesize=A4)
     width, height = A4
-
-    # 素朴なテキスト描画（自動改頁）
-    margin_x = 15 * mm
-    margin_y = 15 * mm
-    y = height - margin_y
-    c.setFont("Helvetica", 10)
-
+    x, y = 40, height - 40
     for line in text.splitlines():
-        if y < margin_y:
+        if y < 40:
             c.showPage()
-            c.setFont("Helvetica", 10)
-            y = height - margin_y
-        c.drawString(margin_x, y, line[:180])  # 超長行は素直に切る
-        y -= 12
-
-    c.showPage()
+            y = height - 40
+        c.drawString(x, y, line[:1000])  # 1行制限（超長行ガード）
+        y -= 14
     c.save()
     return buf.getvalue()