Follow up on campaign replies and clicks

"""Hand campaign repliers to the CRM and re-engage clickers with a second-touch campaign."""
import os
import time

import requests

BASE = "https://api.tryfuse.ai/api/v1/business"
AUTH = (os.environ["FUSE_API_KEY"], "")
CAMPAIGN_ID = os.environ["FUSE_CAMPAIGN_ID"]
SECOND_TOUCH_ID = os.environ.get("FUSE_SECOND_TOUCH_ID")  # set after the first run
CRM = "hubspot"


class FuseError(Exception):
    def __init__(self, status, code, message):
        super().__init__(f"{status} {code}: {message}")
        self.code = code


def call(method, path, **kwargs):
    """Send one request, wait out 429s, and raise FuseError on any other failure."""
    while True:
        response = requests.request(method, BASE + path, auth=AUTH, timeout=60, **kwargs)
        if response.status_code == 429:
            time.sleep(int(response.headers.get("Retry-After", 5)))
            continue
        if response.status_code >= 400:
            try:
                error = response.json().get("error", {})
            except ValueError:
                error = {}
            raise FuseError(response.status_code, error.get("code"), error.get("message"))
        return response.json() if response.content else None


def list_named(name):
    """Return the id of the contact list with this name, creating it if needed."""
    found = call("GET", "/lists", params={"search": name, "entityType": "contactList"})["data"]
    for existing in found:
        if existing["name"] == name:
            return existing["id"]
    return call("POST", "/lists", json={"name": name})["list"]["id"]


def add_members(list_id, contact_ids):
    for start in range(0, len(contact_ids), 5000):
        call("POST", f"/lists/{list_id}/members", json={"contactIds": contact_ids[start:start + 5000]})


# 2. Sort the leads into repliers and finished clickers
repliers, clickers, page = [], [], 1
while True:
    body = call("GET", f"/campaigns/{CAMPAIGN_ID}/leads", params={"limit": 100, "pageNum": page})
    for lead in body["leads"]:
        if lead["hasReplied"]:
            repliers.append(lead["contactId"])
        elif lead["hasClicked"] and lead["sentSteps"] == lead["totalSteps"]:
            clickers.append(lead["contactId"])
    if page >= body["pagination"]["totalPages"]:
        break
    page += 1
print(f"{len(repliers)} repliers, {len(clickers)} clickers who finished without replying")

# 3. Print what each replier said
for contact_id in repliers:
    timeline = call("GET", f"/contacts/{contact_id}/activities", params={"limit": 20})["activities"]
    replies = [a for a in timeline if a["status"] == "replied" and a["campaignId"] == CAMPAIGN_ID]
    if replies:
        print(f"\n{contact_id}: {replies[0]['subject']}\n{replies[0]['body']['content']}")

# 4. Hand repliers to the CRM
if repliers:
    replied_list = list_named("Q4 fintech - replied")
    add_members(replied_list, repliers)
    if CRM in call("GET", "/integrations")["integrations"]["connected"]:
        call("POST", f"/lists/{replied_list}/crm-push", json={"crm": CRM})

# 5-6. Put the clickers on their own list, then create the second touch once
if clickers:
    clicked_list = list_named("Q4 fintech - clicked, no reply")
    add_members(clicked_list, clickers)
    if not SECOND_TOUCH_ID:
        # The copy is ready at once and keeps every exclusion except excludeCompanyLists
        copy_id = call("POST", f"/campaigns/{CAMPAIGN_ID}/duplicate")["campaign"]["id"]
        call("PATCH", f"/campaigns/{copy_id}", json={
            "campaignName": "Q4 fintech - second touch",
            "listIds": [clicked_list],
            "excludeContacted": "none",
            "isDynamic": True,
        })
        print(f"Draft second touch {copy_id}: rewrite its steps, preview, then approve.")
        print(f"Set FUSE_SECOND_TOUCH_ID={copy_id} so later runs only add members.")