X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/26c45c9115e985280c9c0d8f6560ebce0c133242..965a0bef5bcde07b7df2ac1b07f73ce67105b8a6:/contact/management/commands/export_newsletter.py

diff --git a/contact/management/commands/export_newsletter.py b/contact/management/commands/export_newsletter.py
new file mode 100644
index 0000000..70ef2b4
--- /dev/null
+++ b/contact/management/commands/export_newsletter.py
@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# This file is part of EduMed, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from django.core.management.base import BaseCommand
+
+from contact.models import Contact
+from edumed import contact_forms
+
+FORMS = (
+    contact_forms.CooperateForm,
+    contact_forms.ContestForm,
+    contact_forms.WTEMForm,
+    contact_forms.TEMForm,
+    contact_forms.CybernauciForm,
+    contact_forms.WLEMForm,
+)
+
+
+class Command(BaseCommand):
+    help = 'Export contacts for newsletter.'
+
+    def handle(self, **options):
+        addresses = set(self.get_addresses())
+        for address in addresses:
+            print address
+
+    def get_addresses(self):
+        for form in FORMS:
+            tags = [form.form_tag] + form.old_form_tags
+            contacts = Contact.objects.filter(form_tag__in=tags)
+            for contact in contacts:
+                if contact.body.get(form.mailing_field):
+                    yield contact.contact