pre release 20240903

rog/admin.py

@@ -6,11 +6,16 @@ from leaflet.admin import LeafletGeoAdminMixin
 from leaflet_admin_list.admin import LeafletAdminListMixin
 from .models import RogUser, Location, SystemSettings, JoinedEvent, Favorite, TravelList, TravelPoint, ShapeLayers, Event, Location_line, Location_polygon, JpnAdminMainPerf, Useractions, CustomUser, GifuAreas, UserTracks, templocation, UserUpload, EventUser, GoalImages, CheckinImages, NewEvent, NewEvent2, Team, NewCategory, Category, Entry, Member, TempUser
 from django.contrib.auth.admin import UserAdmin
-from django.urls import path
+from django.urls import path,reverse
 from django.shortcuts import render
 from django import forms;
 import requests
 
+from django.http import HttpResponseRedirect
+from django.utils.html import format_html
+from .forms import CSVUploadForm
+from .views import process_csv_upload
+
 class RogAdmin(LeafletAdminListMixin, LeafletGeoAdminMixin, admin.ModelAdmin):
     list_display=['title', 'venue', 'at_date',]
 
@@ -196,10 +201,42 @@ class TempLocationAdmin(LeafletGeoAdmin):
     list_display = ('location_id','cp', 'location_name', 'category', 'event_name', 'event_active', 'auto_checkin', 'checkin_radius', 'checkin_point', 'buy_point',)
     actions = [tranfer_to_location,]
 
+ 
 @admin.register(NewEvent2)
 class NewEvent2Admin(admin.ModelAdmin):
-    list_display = ['event_name', 'start_datetime', 'end_datetime']
-    search_fields = ['event_name']
+    list_display = ['event_name', 'start_datetime', 'end_datetime', 'csv_upload_button']
+
+    def get_urls(self):
+        urls = super().get_urls()
+        my_urls = [
+                path('csv-upload/', self.admin_site.admin_view(self.csv_upload_view), name='newevent2_csv_upload'),
+        ]
+        return my_urls + urls
+
+    def csv_upload_view(self, request):
+        if request.method == 'POST':
+            form = CSVUploadForm(request.POST, request.FILES)
+            if form.is_valid():
+                csv_file = request.FILES['csv_file']
+                event = form.cleaned_data['event']
+                process_csv_upload(csv_file, event)
+                self.message_user(request, "CSV file has been processed successfully.")
+                return HttpResponseRedirect("../")
+        else:
+            form = CSVUploadForm()
+
+        return render(request, 'admin/csv_upload.html', {'form': form})
+
+    def csv_upload_button(self, obj):
+        url = reverse('admin:newevent2_csv_upload')
+        return format_html('<a class="button" href="{}">CSVアップロード</a>', url)
+    csv_upload_button.short_description = 'CSV Upload'
+
+    def changelist_view(self, request, extra_context=None):
+        extra_context = extra_context or {}
+        extra_context['csv_upload_url'] = reverse('admin:newevent2_csv_upload')
+        return super().changelist_view(request, extra_context=extra_context)
+
 
 @admin.register(Team)
 class TeamAdmin(admin.ModelAdmin):