diff --git a/api/main.py b/api/main.py index 7da84d0..9db52e6 100644 --- a/api/main.py +++ b/api/main.py @@ -10,6 +10,8 @@ from typing import List, Optional import mysql.connector from mysql.connector import Error import os +import csv +import io from contextlib import contextmanager # Configuration @@ -93,6 +95,18 @@ class Subscription(BaseModel): member_email: EmailStr active: bool = True +class BulkImportRequest(BaseModel): + csv_data: str + list_ids: List[int] + +class BulkImportResult(BaseModel): + total_rows: int + processed_rows: int + created_members: int + updated_members: int + subscriptions_added: int + errors: List[str] + # Routes @app.get("/") async def root(): @@ -365,6 +379,118 @@ async def unsubscribe_member(list_email: EmailStr, member_email: EmailStr, token cursor.close() return {"message": "Unsubscribed successfully"} +@app.post("/bulk-import", response_model=BulkImportResult) +async def bulk_import_members(bulk_request: BulkImportRequest, token: str = Depends(verify_token)): + """Bulk import members from CSV data and subscribe them to specified lists""" + + result = BulkImportResult( + total_rows=0, + processed_rows=0, + created_members=0, + updated_members=0, + subscriptions_added=0, + errors=[] + ) + + with get_db() as conn: + cursor = conn.cursor(dictionary=True) + + try: + # Verify all list_ids exist + if bulk_request.list_ids: + placeholders = ','.join(['%s'] * len(bulk_request.list_ids)) + cursor.execute(f"SELECT list_id FROM lists WHERE list_id IN ({placeholders})", bulk_request.list_ids) + existing_lists = [row['list_id'] for row in cursor.fetchall()] + + invalid_lists = set(bulk_request.list_ids) - set(existing_lists) + if invalid_lists: + raise HTTPException(status_code=400, detail=f"Invalid list IDs: {list(invalid_lists)}") + + # Parse CSV data + csv_reader = csv.DictReader(io.StringIO(bulk_request.csv_data)) + + # Validate CSV headers - we need at least Name and Email + if not csv_reader.fieldnames or 'Name' not in csv_reader.fieldnames or 'Email' not in csv_reader.fieldnames: + raise HTTPException(status_code=400, detail="CSV must contain 'Name' and 'Email' columns") + + for row_num, row in enumerate(csv_reader, start=1): + result.total_rows += 1 + + try: + name = row.get('Name', '').strip() + email = row.get('Email', '').strip() + + # Skip rows without email (name is optional) + if not email: + result.errors.append(f"Row {row_num}: Missing email address") + continue + + # Basic email validation + if '@' not in email or '.' not in email.split('@')[1]: + result.errors.append(f"Row {row_num}: Invalid email format: {email}") + continue + + # Use email as name if no name provided + if not name: + name = email.split('@')[0] # Use the part before @ as a default name + + # Check if member exists + cursor.execute("SELECT member_id FROM members WHERE email = %s", (email,)) + existing_member = cursor.fetchone() + + if existing_member: + # Update existing member's name if we have a better name (not auto-generated from email) + should_update_name = ( + row.get('Name', '').strip() and # We have a name in the CSV + name != email.split('@')[0] # It's not the auto-generated name + ) + if should_update_name: + cursor.execute("UPDATE members SET name = %s WHERE member_id = %s", (name, existing_member['member_id'])) + if cursor.rowcount > 0: + result.updated_members += 1 + member_id = existing_member['member_id'] + else: + # Create new member + cursor.execute( + "INSERT INTO members (name, email, active) VALUES (%s, %s, %s)", + (name, email, True) + ) + member_id = cursor.lastrowid + result.created_members += 1 + + # Subscribe to selected lists + for list_id in bulk_request.list_ids: + try: + cursor.execute( + "INSERT INTO list_members (list_id, member_id, active) VALUES (%s, %s, %s)", + (list_id, member_id, True) + ) + result.subscriptions_added += 1 + except Error as e: + if "Duplicate entry" in str(e): + # Member already subscribed to this list - not an error + pass + else: + result.errors.append(f"Row {row_num}: Failed to subscribe to list {list_id}: {str(e)}") + + result.processed_rows += 1 + + except Exception as e: + result.errors.append(f"Row {row_num}: {str(e)}") + continue + + # Commit all changes + conn.commit() + cursor.close() + + return result + + except HTTPException: + raise + except Exception as e: + cursor.close() + raise HTTPException(status_code=500, detail=f"Bulk import failed: {str(e)}") + if __name__ == "__main__": import uvicorn uvicorn.run(app, host="0.0.0.0", port=8000) diff --git a/web/index.html b/web/index.html index 30672aa..39e4d68 100644 --- a/web/index.html +++ b/web/index.html @@ -96,10 +96,16 @@ Click the "Lists" button next to any member to manage their subscriptions - +
+ + +
@@ -273,6 +279,189 @@
+ + +