From 0ec3dc267d6e69cee9bafc49f8886442900dd616 Mon Sep 17 00:00:00 2001
From: Ashish Acharya <ashish.acharya14@gmail.com>
Date: Wed, 29 Nov 2023 13:20:35 -0600
Subject: [PATCH] Add admin action to download a csv with all candidate URLs

---
 sde_collections/admin.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/sde_collections/admin.py b/sde_collections/admin.py
index aa21c2bc..077bf88a 100644
--- a/sde_collections/admin.py
+++ b/sde_collections/admin.py
@@ -9,6 +9,31 @@
 from .tasks import import_candidate_urls_from_api
 
 
+@admin.action(description="Download candidate URLs as csv")
+def download_candidate_urls_as_csv(modeladmin, request, queryset):
+    response = HttpResponse(content_type="text/csv")
+    response["Content-Disposition"] = 'attachment; filename="candidate_urls.csv"'
+
+    writer = csv.writer(response)
+
+    if len(queryset) > 1:
+        messages.add_message(
+            request, messages.ERROR, "You can only export one collection at a time."
+        )
+        return
+
+    urls = CandidateURL.objects.filter(collection=queryset.first()).values_list(
+        "url", flat=True
+    )
+
+    # Write your headers here
+    writer.writerow(["candidate_url"])
+    for url in urls:
+        writer.writerow([url])
+
+    return response
+
+
 @admin.action(description="Import metadata from Sinequa configs")
 def import_sinequa_metadata(modeladmin, request, queryset):
     for collection in queryset.all():
@@ -168,6 +193,7 @@ class CollectionAdmin(admin.ModelAdmin, ExportCsvMixin, UpdateConfigMixin):
     actions = [
         "export_as_csv",
         "update_config",
+        download_candidate_urls_as_csv,
         import_candidate_urls_test,
         import_candidate_urls_production,
         import_candidate_urls_secret_test,