Run a survey one question at a time on WhatsApp

from clickruntime import on_event, credentials, http, state, notify

PHONE_NUMBER_ID = "123456789012345"
QUESTIONS = [("How did you hear about us?", ["Search", "Friend", "Social"]),
             ("How likely are you to recommend us?", ["Very", "Maybe", "Not likely"]),
             ("Anything we could do better?", None)]

def send(to, q, options):
    msg = {"type": "text", "text": {"body": q}} if not options else {"type": "interactive", "interactive": {
        "type": "button", "body": {"text": q},
        "action": {"buttons": [{"type": "reply", "reply": {"id": o, "title": o}} for o in options]}}}
    http.post(f"https://graph.facebook.com/v21.0/{PHONE_NUMBER_ID}/messages",
              headers={"Authorization": f"Bearer {credentials.secret('WHATSAPP_TOKEN')}"},
              json={"messaging_product": "whatsapp", "to": to, **msg})

@on_event
def handle(event):
    value = event.json["entry"][0]["changes"][0]["value"]
    if "messages" not in value:
        return {"skipped": True}
    m = value["messages"][0]
    answer = m.get("text", {}).get("body") or m.get("interactive", {}).get("button_reply", {}).get("title")
    s = state.get(f"quiz:{m['from']}") or {"step": -1, "answers": []}
    if s["step"] >= 0:
        s["answers"].append(answer)
    s["step"] += 1
    if s["step"] < len(QUESTIONS):
        send(m["from"], *QUESTIONS[s["step"]])
        state.set(f"quiz:{m['from']}", s)
        return {"step": s["step"]}
    state.delete(f"quiz:{m['from']}")
    send(m["from"], "Thanks, that's everything!", None)
    notify.email_me(f"Questionnaire from +{m['from']}", "\n".join(f"{q}: {a}" for (q, _), a in zip(QUESTIONS, s["answers"])))
    return {"done": True}