| """ |
| Loads the pretest/posttest email pools (study/stimuli/email_pool.json, |
| produced by src/data/extract_stimuli.py) and selects a balanced random subset |
| for each participant. |
| |
| Ground truth ("type") is deliberately NOT sent to the client -- the server |
| remembers which email IDs were shown to which code, and scoring against |
| ground truth happens later during analysis (Objective O5), not in real time. |
| This also means pretest/posttest give no immediate correct/incorrect |
| feedback, which is the right design for a pre/post measurement (see |
| study/consent_ethics/ for the participant-facing description). |
| """ |
| import json |
| import random |
| from pathlib import Path |
|
|
| POOL_FILE = Path(__file__).resolve().parents[1] / "study" / "stimuli" / "email_pool.json" |
|
|
| PRETEST_N_PHISHING = 3 |
| PRETEST_N_LEGIT = 2 |
| POSTTEST_N_PHISHING = 3 |
| POSTTEST_N_LEGIT = 2 |
|
|
| _pool = None |
|
|
|
|
| def _load_pool(): |
| global _pool |
| if _pool is None: |
| with open(POOL_FILE, encoding="utf-8") as f: |
| _pool = json.load(f) |
| return _pool |
|
|
|
|
| def _select(batch: list, n_phishing: int, n_legit: int) -> list: |
| phishing = [e for e in batch if e["type"] == "Phishing"] |
| legit = [e for e in batch if e["type"] == "Legitimate"] |
| selected = random.sample(phishing, min(n_phishing, len(phishing))) + \ |
| random.sample(legit, min(n_legit, len(legit))) |
| random.shuffle(selected) |
| return selected |
|
|
|
|
| def get_pretest_emails() -> list: |
| pool = _load_pool() |
| return _select(pool["pretest_batch"], PRETEST_N_PHISHING, PRETEST_N_LEGIT) |
|
|
|
|
| def get_posttest_emails() -> list: |
| pool = _load_pool() |
| return _select(pool["posttest_batch"], POSTTEST_N_PHISHING, POSTTEST_N_LEGIT) |
|
|
|
|
| def public_view(emails: list) -> list: |
| """ |
| Strips ground truth before sending to the client. |
| |
| Includes both "text" (unchanged flattened string -- sent to /classify for |
| the tool-interaction step, must stay identical to what T30/T31 were |
| computed against) and "display" (structured subject/from/to/date/body |
| fields, used only to render a realistic email-client UI). |
| """ |
| return [{"id": e["id"], "text": e["text"], "display": e.get("display")} for e in emails] |
|
|
