local changes from server

[prawokultury.git] / contact / management / commands / contact_csv.py

diff --git a/contact/management/commands/contact_csv.py b/contact/management/commands/contact_csv.py
new file mode 100644 (file)
index 0000000..882ad04
--- /dev/null
+++ b/contact/management/commands/contact_csv.py
@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+
+from optparse import make_option
+from django.core.management.base import BaseCommand
+
+class Command(BaseCommand):
+    option_list = BaseCommand.option_list + (
+        make_option('-e', '--exclude-fields', dest='exclude', metavar='name,...', default='',
+            help='Exclude specific body fields by name'),
+        make_option('-m', '--mailto', dest='mailto', metavar='email,...', default='',
+            help='Send by mail'),
+    )
+    help = 'Dump form entries in a csv file.'
+    def handle(self, *args, **kwargs):
+        from csv import writer
+        from sys import stdout
+        from django.conf import settings
+        from django.core.mail import EmailMessage
+        from contact.models import Contact
+        from StringIO import StringIO
+
+        excludes = set(kwargs['exclude'].split(','))
+
+        buffer = StringIO()
+        w = writer(buffer)
+
+        lines = []
+        headers = set()
+        for c in Contact.objects.filter(form_tag__in=args):
+            headers |= set(c.body.keys())
+            lines.append(c)
+        headers = sorted(headers - excludes)
+        w.writerow([u'email'] + headers)
+        for c in lines:
+            row = [c.contact.encode('utf-8')]
+            for h in headers:
+                v = c.body.get(h)
+                if isinstance(v, unicode):
+                    v = v.encode('utf-8') 
+                row.append(v)
+            w.writerow(row)
+
+        mailto = kwargs['mailto']
+        if mailto:
+            mailto = mailto.split(',')
+            message = EmailMessage('Contacts CSV', 'Contacts CSV', settings.CONTACT_EMAIL, mailto)
+            message.attach('contacts.csv', buffer.getvalue(), 'text/csv')
+            message.send()
+        else:
+            print buffer.getvalue()
+