Download Chat Histories

django_app/redbox_app/redbox_core/admin.py

@@ -1,4 +1,7 @@
+import csv
+
 from django.contrib import admin
+from django.http import HttpResponse
 
 from . import models
 
@@ -18,9 +21,29 @@ class ChatMessageInline(admin.StackedInline):
 
 
 class ChatHistoryAdmin(admin.ModelAdmin):
+    def export_as_csv(self, request, queryset):  # noqa:ARG002
+        history_field_names = [field.name for field in models.ChatHistory._meta.fields]  # noqa:SLF001
+        message_field_names = [field.name for field in models.ChatMessage._meta.fields]  # noqa:SLF001
+
+        response = HttpResponse(content_type="text/csv")
+        response["Content-Disposition"] = "attachment; filename=chathistory.csv"
+        writer = csv.writer(response)
+
+        writer.writerow(history_field_names + message_field_names)
+        for chat_history in queryset:
+            for chat_message in chat_history.chatmessage_set.all():
+                writer.writerow(
+                    [getattr(chat_history, field) for field in history_field_names]
+                    + [getattr(chat_message, field) for field in message_field_names]
+                )
+
+        return response
+
+    export_as_csv.short_description = "Export Selected"
     inlines = [ChatMessageInline]
     list_display = ["name", "users"]
     list_filter = ["users"]
+    actions = ["export_as_csv"]
 
 
 admin.site.register(models.User, UserResource)