-# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
-# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
-#
-from hashlib import md5
-
-from django.conf import settings
-from mailchimp3 import MailChimp
-from mailchimp3.mailchimpclient import MailChimpError
-
-
-def subscriber_hash(email):
- return md5(email.encode('utf-8')).hexdigest()
-
-
-def remove_from_groups(email, client):
- group_ids = []
- categories = client.lists.interest_categories.all(list_id=settings.MAILCHIMP_LIST_ID)['categories']
- for category in categories:
- groups = client.lists.interest_categories.interests.all(
- list_id=settings.MAILCHIMP_LIST_ID, category_id=category['id'])['interests']
- group_ids += [group['id'] for group in groups]
- interests = {group_id: False for group_id in group_ids}
- client.lists.members.update(
- settings.MAILCHIMP_LIST_ID, subscriber_hash(email),
- data={'interests': interests})
-
-
-def subscribe(email, mailing_lists=None):
- client = MailChimp(mc_api=settings.MAILCHIMP_API_KEY, timeout=10.0)
- try:
- member = client.lists.members.get(settings.MAILCHIMP_LIST_ID, subscriber_hash(email))
- except MailChimpError:
- pass
- else:
- if member['status'] != 'subscribed':
- remove_from_groups(email, client)
- mailing_lists = mailing_lists or [settings.MAILCHIMP_DEFAULT_GROUP]
- interests = {
- settings.MAILCHIMP_GROUP_IDS[mailing_list]: True
- for mailing_list in mailing_lists
- if mailing_list in settings.MAILCHIMP_GROUP_IDS
- }
- client.lists.members.create_or_update(
- settings.MAILCHIMP_LIST_ID, subscriber_hash(email),
- data={
- 'email_address': email,
- 'status_if_new': 'subscribed',
- 'status': 'subscribed',
- 'interests': interests,
- }
- )
-
-
-def unsubscribe(email, mailing_lists=None):
- client = MailChimp(mc_api=settings.MAILCHIMP_API_KEY, timeout=10.0)
- try:
- member = client.lists.members.get(settings.MAILCHIMP_LIST_ID, subscriber_hash(email))
- except MailChimpError:
- return
- else:
- if member['status'] != 'subscribed':
- return
- mailing_lists = mailing_lists or settings.MAILCHIMP_GROUP_IDS
- interests = {
- settings.MAILCHIMP_GROUP_IDS[mailing_list]: False
- for mailing_list in mailing_lists
- if mailing_list in settings.MAILCHIMP_GROUP_IDS
- }
- client.lists.members.update(
- settings.MAILCHIMP_LIST_ID, subscriber_hash(email),
- data={
- 'interests': interests,
- }
- )