Hamza

Created by: osergeek

visits/admin.py

+
 """Visits admin panel configuration."""
 
 from django import forms
@@ -122,7 +123,7 @@ class ParticipationInline(admin.TabularInline):
 def accept_selected_participations(modeladmin, request, queryset):
     """Accept selected participations in list view."""
     for obj in queryset:
-        obj.accepted = True
+        obj.accepted = 1
         obj.save()
     count = queryset.count()
     s = pluralize(count)
@@ -137,12 +138,12 @@ accept_selected_participations.short_description = (
 def reject_selected_participations(modeladmin, request, queryset):
     """Reject selected participations in list view."""
     for obj in queryset:
-        obj.accepted = False
+        obj.accepted = 0
         obj.save()
     count = queryset.count()
     s = pluralize(count)
     messages.add_message(request, messages.SUCCESS,
-                         f'{count} participation{s} rejetée{s} avec succès.') ## rejeté place accepté
+    f'{count} participation{s} rejetée{s} avec succès.')  # rejeté place accepté
 
 
 reject_selected_participations.short_description = (
@@ -162,6 +163,21 @@ def wait_selected_participations(modeladmin, request, queryset):
 wait_selected_participations.short_description = (
     'Mettre en attentes les participations sélectionnées')
 
+def wait_selected_participations(modeladmin, request, queryset):
+    """Reject selected participations in list view."""
+    for obj in queryset:
+        obj.accepted = 2  # in wait
+        obj.save()
+    count = queryset.count()
+    s = pluralize(count)
+    messages.add_message(request, messages.SUCCESS,
+                         f'{count} participation{s} en attente{s} avec succès.')
+
+
+wait_selected_participations.short_description = (
+    'Mettre en attentes les participations sélectionnées')
+
+
 @admin.register(Participation)
 class ParticipationAdmin(admin.ModelAdmin):
     """Admin panel for visit participations."""
@@ -195,7 +211,6 @@ class ParticipationAdmin(admin.ModelAdmin):
         response.write(codecs.BOM_UTF8)  # force response to be UTF-8
         writer = csv.writer(response, delimiter=';')
 
-
         writer.writerow(['first_name', 'last_name', 'school', 'grade',
                          'phone_number', 'scholarship'] + field_names)
 
@@ -211,7 +226,6 @@ class ParticipationAdmin(admin.ModelAdmin):
             row = writer.writerow([name[0]['first_name'], name[0]['last_name'], school[0]['school'], school[0]['grade'], name[0]['phone_number'], school[0]['scholarship']] + [getattr(obj, field)
                                                                                                                                                                                for field in field_names])
 
-
             nb_user += 1
         return response