64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
from flask import Blueprint, request, jsonify
|
|
from datetime import datetime
|
|
from pymongo.errors import DuplicateKeyError
|
|
from database import subscribers_collection
|
|
|
|
subscription_bp = Blueprint('subscription', __name__)
|
|
|
|
|
|
@subscription_bp.route('/api/subscribe', methods=['POST'])
|
|
def subscribe():
|
|
"""Subscribe a user to the newsletter"""
|
|
data = request.json
|
|
email = data.get('email', '').strip().lower()
|
|
|
|
if not email or '@' not in email:
|
|
return jsonify({'error': 'Invalid email address'}), 400
|
|
|
|
try:
|
|
subscriber_doc = {
|
|
'email': email,
|
|
'subscribed_at': datetime.utcnow(),
|
|
'status': 'active'
|
|
}
|
|
|
|
# Try to insert, if duplicate key error, subscriber already exists
|
|
try:
|
|
subscribers_collection.insert_one(subscriber_doc)
|
|
return jsonify({'message': 'Successfully subscribed!'}), 201
|
|
except DuplicateKeyError:
|
|
# Check if subscriber is active
|
|
existing = subscribers_collection.find_one({'email': email})
|
|
if existing and existing.get('status') == 'active':
|
|
return jsonify({'message': 'Email already subscribed'}), 200
|
|
else:
|
|
# Reactivate if previously unsubscribed
|
|
subscribers_collection.update_one(
|
|
{'email': email},
|
|
{'$set': {'status': 'active', 'subscribed_at': datetime.utcnow()}}
|
|
)
|
|
return jsonify({'message': 'Successfully re-subscribed!'}), 200
|
|
|
|
except Exception as e:
|
|
return jsonify({'error': str(e)}), 500
|
|
|
|
|
|
@subscription_bp.route('/api/unsubscribe', methods=['POST'])
|
|
def unsubscribe():
|
|
"""Unsubscribe a user from the newsletter"""
|
|
data = request.json
|
|
email = data.get('email', '').strip().lower()
|
|
|
|
try:
|
|
result = subscribers_collection.update_one(
|
|
{'email': email},
|
|
{'$set': {'status': 'inactive'}}
|
|
)
|
|
|
|
if result.matched_count > 0:
|
|
return jsonify({'message': 'Successfully unsubscribed'}), 200
|
|
else:
|
|
return jsonify({'error': 'Email not found in subscribers'}), 404
|
|
except Exception as e:
|
|
return jsonify({'error': str(e)}), 500
|