# -*- coding: utf-8 -*-
+import csv
+import json
+
from django.contrib import admin
-from .models import Contact
from django.utils.translation import ugettext as _
-from .forms import contact_forms, admin_list_width
from django.utils.safestring import mark_safe
from django.conf.urls import patterns, url
from django.http import HttpResponse, Http404
-from .utils import csv_prepare
+from edumed.utils import UnicodeCSVWriter
+from .forms import contact_forms, admin_list_width
+from .models import Contact
class ContactAdminMeta(admin.ModelAdmin.__class__):
return lambda obj: self.admin_list(obj, nr)
raise AttributeError(name)
- def change_view(self, request, object_id, from_url='', extra_context=None):
+ def change_view(self, request, object_id, form_url='', extra_context=None):
if object_id:
try:
instance = Contact.objects.get(pk=object_id)
pass
else:
# Create readonly fields from the body JSON.
- body_fields = ['body__%s' % k for k in instance.body.keys()]
attachments = list(instance.attachment_set.all())
- body_fields += ['body__%s' % a.tag for a in attachments]
- self.readonly_fields.extend(body_fields)
+ body_keys = instance.body.keys() + [a.tag for a in attachments]
# Find the original form.
try:
orig_fields = {}
# Try to preserve the original order.
- admin_fields = []
orig_keys = list(orig_fields.keys())
- while orig_keys:
- key = orig_keys.pop(0)
- key = "body__%s" % key
- if key in body_fields:
- admin_fields.append(key)
- body_fields.remove(key)
- admin_fields.extend(body_fields)
+ admin_keys = [key for key in orig_keys if key in body_keys] + \
+ [key for key in body_keys if key not in orig_keys]
+ admin_fields = ['body__%s' % key for key in admin_keys]
+
+ self.readonly_fields.extend(admin_fields)
self.fieldsets = [
(None, {'fields': self.fields}),
]
# Create field getters for fields and attachments.
+ def attach_getter(key, value):
+ def f(self):
+ return value
+ f.short_description = orig_fields[key].label if key in orig_fields else _(key)
+ setattr(self, "body__%s" % key, f)
+
for k, v in instance.body.items():
- f = (lambda v: lambda self: v)(Contact.pretty_print(v, for_html=True))
- f.short_description = orig_fields[k].label if k in orig_fields else _(k)
- setattr(self, "body__%s" % k, f)
+ attach_getter(k, Contact.pretty_print(v, for_html=True))
download_link = "<a href='%(url)s'>%(url)s</a>"
for attachment in attachments:
- k = attachment.tag
link = mark_safe(download_link % {
'url': attachment.get_absolute_url()})
- f = (lambda v: lambda self: v)(link)
- f.short_description = orig_fields[k].label if k in orig_fields else _(k)
- setattr(self, "body__%s" % k, f)
+ attach_getter(attachment.tag, link)
return super(ContactAdmin, self).change_view(
- request, object_id, from_url=from_url, extra_context=extra_context)
+ request, object_id, form_url=form_url, extra_context=extra_context)
def changelist_view(self, request, extra_context=None):
context = dict()
def extract_view(request, form_tag, extract_type_slug):
- toret = u''
contacts_by_spec = dict()
form = contact_forms.get(form_tag)
if form is None and extract_type_slug not in ('contacts', 'all'):
q = q.filter(created_at__month=at_month)
# Segregate contacts by body key sets
+ if form:
+ orig_keys = list(form().fields.keys())
+ else:
+ orig_keys = []
for contact in q.all():
if extract_type_slug == 'contacts':
keys = ['contact']
elif extract_type_slug == 'all':
keys = contact.body.keys() + ['contact']
+ keys = [key for key in orig_keys if key in keys] + [key for key in keys if key not in orig_keys]
else:
keys = form.get_extract_fields(contact, extract_type_slug)
contacts_by_spec.setdefault(tuple(keys), []).append(contact)
-
+
+ response = HttpResponse(content_type='text/csv')
+ csv_writer = UnicodeCSVWriter(response)
+
# Generate list for each body key set
for keys, contacts in contacts_by_spec.items():
- toret += u','.join(keys) + '\n'
+ csv_writer.writerow(keys)
for contact in contacts:
if extract_type_slug == 'contacts':
records = [dict(contact=contact.contact)]
for key in keys:
if key not in record:
record[key] = ''
- record[key] = csv_prepare(record[key])
- toret += u','.join([record[key] for key in keys]) + '\n'
- toret += '\n\n'
+ if isinstance(record[key], basestring):
+ pass
+ elif isinstance(record[key], bool):
+ record[key] = 'tak' if record[key] else 'nie'
+ elif isinstance(record[key], (list, tuple)) and all(isinstance(v, basestring) for v in record[key]):
+ record[key] = ', '.join(record[key])
+ else:
+ record[key] = json.dumps(record[key])
+
+ csv_writer.writerow([record[key] for key in keys])
+ csv_writer.writerow([])
- response = HttpResponse(toret, content_type='text/csv')
response['Content-Disposition'] = 'attachment; filename="kontakt.csv"'
return response
fnp / edumed.git / blobdiff