From 9a49666f4fd830a70cc0224050a0be0592381931 Mon Sep 17 00:00:00 2001
From: Claude Paroz <claude@2xlibre.net>
Date: Thu, 8 Jun 2023 10:59:42 +0200
Subject: [PATCH] Ajout action exportation contacts

---
 requirements.txt |  2 +-
 stages/admin.py  | 30 ++++++++++++++++++++++++++++++
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index fa99ee0..9282e73 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-Django~=4.1
+Django~=4.1.0
 tabimport==0.5.1
 xlrd<2  # Required by tabimport
 openpyxl==3.0.5
diff --git a/stages/admin.py b/stages/admin.py
index 2f9efdc..b99e336 100644
--- a/stages/admin.py
+++ b/stages/admin.py
@@ -253,6 +253,7 @@ class CorpContactAdmin(admin.ModelAdmin):
               ('iban', 'bank', 'clearing' ),
               ('qualification', 'fields_of_interest'),
              )
+    actions = ['export_contacts']
     formfield_overrides = {
         models.ManyToManyField: {'widget': forms.CheckboxSelectMultiple},
     }
@@ -266,6 +267,35 @@ class CorpContactAdmin(admin.ModelAdmin):
         qs, distinct = super().get_search_results(request, qs, term)
         return qs.exclude(archived=True), distinct
 
+    def export_contacts(self, request, queryset):
+        """
+        Export all CorpContact in Excel file.
+        """
+        field_names = (
+            'ext_id', 'civility', 'last_name', 'first_name', 'birth_date',
+            'street', 'pcode', 'city', 'etat_civil', 'etat_depuis', 'nation',
+            'tel', 'email', 'corporation', 'is_main', 'always_cc', 'role',
+            'permis_sejour', 'date_validite', 'avs', 'bank', 'clearing', 'iban',
+            'qualification', 'fields_of_interest',
+        )
+        fields = [CorpContact._meta.get_field(fname) for fname in field_names]
+        export_fields = {
+            getattr(f, 'verbose_name', f.name):
+            ('corporation__name' if f.name == 'corporation' else f.name) for f in fields
+        }
+
+        export = OpenXMLExport('Exportation')
+        export.write_line(export_fields.keys(), bold=True)
+        for corp in queryset.values_list(*export_fields.values()):
+            values = []
+            for value, field_name in zip(corp, export_fields.values()):
+                if field_name in ['is_main', 'always_cc']:
+                    value = 'Oui' if value else ''
+                values.append(value)
+            export.write_line(values)
+        return export.get_http_response('contacts_export')
+    export_contacts.short_description = 'Exportation Excel'
+
 
 class ContactInline(admin.StackedInline):
     model = CorpContact