mirror of
https://github.com/jwinterm/modoboa.git
synced 2026-10-18 05:51:26 +02:00
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
"""Export related views."""
|
|
|
|
import csv
|
|
from io import StringIO
|
|
|
|
from django.contrib.auth.decorators import (
|
|
login_required, permission_required, user_passes_test
|
|
)
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import render
|
|
from django.urls import reverse
|
|
from django.utils.translation import ugettext as _
|
|
|
|
from ..forms import ExportDataForm
|
|
from ..lib import get_domains, get_identities
|
|
|
|
|
|
def _export(content, filename):
|
|
"""Export a csv file's content
|
|
|
|
:param content: the content to export (string)
|
|
:param filename: the name that will appear into the response
|
|
:return: an ``HttpResponse`` object
|
|
"""
|
|
resp = HttpResponse(content)
|
|
resp["Content-Type"] = "text/csv"
|
|
resp["Content-Length"] = len(content)
|
|
resp["Content-Disposition"] = 'attachment; filename="{}"'.format(filename)
|
|
return resp
|
|
|
|
|
|
@login_required
|
|
@user_passes_test(
|
|
lambda u: u.has_perm("core.add_user") or u.has_perm("admin.add_alias")
|
|
)
|
|
def export_identities(request):
|
|
ctx = {
|
|
"title": _("Export identities"),
|
|
"action_label": _("Export"),
|
|
"action_classes": "submit",
|
|
"formid": "exportform",
|
|
"action": reverse("admin:identity_export"),
|
|
}
|
|
|
|
if request.method == "POST":
|
|
form = ExportDataForm(request.POST)
|
|
form.is_valid()
|
|
fp = StringIO()
|
|
csvwriter = csv.writer(fp, delimiter=form.cleaned_data["sepchar"])
|
|
identities = get_identities(
|
|
request.user, **request.session["identities_filters"])
|
|
for ident in identities:
|
|
ident.to_csv(csvwriter)
|
|
content = fp.getvalue()
|
|
fp.close()
|
|
return _export(content, "modoboa-identities.csv")
|
|
|
|
ctx["form"] = ExportDataForm()
|
|
return render(request, "common/generic_modal_form.html", ctx)
|
|
|
|
|
|
@login_required
|
|
@permission_required("admin.add_domain")
|
|
def export_domains(request):
|
|
ctx = {
|
|
"title": _("Export domains"),
|
|
"action_label": _("Export"),
|
|
"action_classes": "submit",
|
|
"formid": "exportform",
|
|
"action": reverse("admin:domain_export"),
|
|
}
|
|
|
|
if request.method == "POST":
|
|
form = ExportDataForm(request.POST)
|
|
form.is_valid()
|
|
fp = StringIO()
|
|
csvwriter = csv.writer(fp, delimiter=form.cleaned_data["sepchar"])
|
|
for dom in get_domains(request.user,
|
|
**request.session["domains_filters"]):
|
|
dom.to_csv(csvwriter)
|
|
content = fp.getvalue()
|
|
fp.close()
|
|
return _export(content, "modoboa-domains.csv")
|
|
|
|
ctx["form"] = ExportDataForm()
|
|
return render(request, "common/generic_modal_form.html", ctx)
|