async def get_progress(cid: int, user: User = Depends(get_current_user), db: AsyncSession = Depends(get_db)): c = await _get_own(db, user, cid) return await _progress_payload(db, c) @router.get("/{cid}/stream") async def campaign_stream(cid: int, request: Request, user: User = Depends(get_current_user), db: AsyncSession = Depends(get_db)): """SSE live progress stream. Manual implementation (no sse-starlette): a Redis pub/sub subscription on campaign:{id}:progress plus periodic DB polling fallback. Emits event: progress with counters; event: done when campaign finishes.""" c = await _get_own(db, user, cid) import redis.asyncio as aioredis from app.db.session import AsyncSessionLocal async def event_gen(): r = aioredis.Redis.from_url(cfg.REDIS_URL, decode_responses=True) pubsub = r.pubsub() channel = f"campaign:{cid}:progress" await pubsub.subscribe(channel) heartbeat = 2.0 poll_interval = 5.0 last_payload = None last_poll = 0.0 import time try: yield "event: connected\ndata: {\"ok\": true}\n\n" while True: if await request.is_disconnected(): break # check for pub/sub tick (non-blocking) msg = await pubsub.get_message(ignore_subscribe_messages=True, timeout=0.1) now = time.monotonic() payload = None if msg is not None or now - last_poll >= poll_interval: # refresh counters from DB with a fresh session async with AsyncSessionLocal() as s: camp = await s.get(Campaign, cid) if camp: payload = await _progress_payload(s, camp) last_poll = now if payload is None and msg is None and now - last_poll < poll_interval: await asyncio.sleep(heartbeat) yield ": keepalive\n\n" continue if payload: data = json.dumps(payload) if data != last_payload or msg is not None: yield f"event: progress\ndata: {data}\n\n" last_payload = data if payload.get("status") in ("completed", "cancelled", "failed"): yield f"event: done\ndata: {json.dumps(payload)}\n\n" break await asyncio.sleep(0.2) finally: try: await pubsub.unsubscribe(channel) await pubsub.close() await r.aclose() except Exception: pass return StreamingResponse(event_gen(), media_type="text/event-stream", headers={"Cache-Control": "no-cache", "X-Accel-Buffering": "no", "Connection": "keep-alive"}) @router.get("/{cid}/recipients") async def campaign_recipients(cid: int, status: str = Query(""), search: str = Query(""), page: int = Query(1, ge=1), page_size: int = Query(50, ge=1, le=200), db: AsyncSession = Depends(get_db), user: User = Depends(get_current_user)): """Monitor table: per-recipient status, assigned account, subject/template, errors.""" c = await _get_own(db, user, cid) q = select(CampaignRecipient).where(CampaignRecipient.campaign_id == cid) if status: q = q.where(CampaignRecipient.status == status) if search: q = q.where(CampaignRecipient.email.ilike(f"%{search}%")) total = (await db.execute(select(func.count()).select_from(q.subquery()))).scalar() or 0 rows = (await db.execute(q.order_by(CampaignRecipient.seq) .offset((page - 1) * page_size).limit(page_size))).scalars().all() async def names(): acc_map, subj_map, tpl_map = {}, {}, {} for cr in rows: if cr.assigned_account_id and cr.assigned_account_id not in acc_map: a = await db.get(GmailAccount, cr.assigned_account_id) acc_map[cr.assigned_account_id] = a.email if a else "?" if cr.generated_subject_id and cr.generated_subject_id not in subj_map: s = await db.get(SubjectLine, cr.generated_subject_id) subj_map[cr.generated_subject_id] = (s.subject[:80] if s else "?") if cr.generated_template_id and cr.generated_template_id not in tpl_map: t = await db.get(EmailTemplate, cr.generated_template_id) tpl_map[cr.generated_template_id] = (t.name if t else "?") return acc_map, subj_map, tpl_map acc_map, subj_map, tpl_map = await names() out = [] for cr in rows: out.append({ "id": cr.id, "seq": cr.seq, "email": cr.email, "status": cr.status, "account": acc_map.get(cr.assigned_account_id, ""), "subject": subj_map.get(cr.generated_subject_id, ""), "template": tpl_map.get(cr.generated_template_id, ""), "invoice_number": cr.generated_invoice_number, "amount": cr.generated_amount, "shipping": cr.generated_shipping, "attempts": cr.attempt_count, "message_id": cr.msg_id, "thread_id": cr.thread_id, "sent_at": cr.sent_at.isoformat() if cr.sent_at else None, "last_error": (cr.last_error or "")[:200], }) return {"recipients": out, "total": total, "page": page, "page_size": page_size} @router.get("/{cid}/export") async def export_campaign(cid: int, db: AsyncSession = Depends(get_db), user: User = Depends(get_current_user)): """CSV export of campaign results (one row per campaign recipient).""" c = await _get_own(db, user, cid) rows = (await db.execute( select(CampaignRecipient).where(CampaignRecipient.campaign_id == cid) .order_by(CampaignRecipient.seq))).scalars().all() def generate(): buf = io.StringIO() w = csv.writer(buf) w.writerow(["seq", "email", "status", "account", "subject", "template", "invoice_number", "amount", "shipping", "attempts", "message_id", "thread_id", "sent_at", "last_error"]) yield buf.getvalue() buf.seek(0); buf.truncate(0) for cr in rows: w.writerow([cr.seq, cr.email, cr.status, cr.assigned_account_id, cr.generated_subject_id, cr.generated_template_id, cr.generated_invoice_number, cr.generated_amount or "", cr.generated_shipping or "", cr.attempt_count, cr.msg_id, cr.thread_id, cr.sent_at.isoformat() if cr.sent_at else "", (cr.last_error or "")[:300]]) yield buf.getvalue() buf.seek(0); buf.truncate(0) fname = re.sub(r"[^A-Za-z0-9_-]+", "_", c.name)[:60] or f"campaign_{cid}" return StreamingResponse(generate(), media_type="text/csv", headers={"Content-Disposition": f'attachment; filename="{fname}.csv"'}) # --------------------------------------------------------------------------- # Wizard options: what lists/subjects/templates/accounts/phones are selectable # --------------------------------------------------------------------------- @router.get("/wizard/options") async def wizard_options(db: AsyncSession = Depends(get_db), user: User = Depends(get_current_user)): """All selectable entities for the 12-step wizard in one call.""" lists = (await db.execute(select(RecipientList).where(RecipientList.user_id == user.id) .order_by(RecipientList.id.desc()))).scalars().all() subjects = (await db.execute(select(SubjectLine).where(SubjectLine.user_id == user.id)