From 878154c305f6ad99e0810824f5aa0706ebb9610c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20R=C3=B8dvand?= <martin@rodvand.net>
Date: Tue, 30 Mar 2021 23:10:08 +0200
Subject: [PATCH] Closes #5830: Add as_attachment field to API serializers and
 admin view.

---
 netbox/extras/admin.py           | 4 ++--
 netbox/extras/api/serializers.py | 2 +-
 netbox/extras/models/models.py   | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/netbox/extras/admin.py b/netbox/extras/admin.py
index 4e11307fc..6e421e679 100644
--- a/netbox/extras/admin.py
+++ b/netbox/extras/admin.py
@@ -193,7 +193,7 @@ class ExportTemplateForm(forms.ModelForm):
 class ExportTemplateAdmin(admin.ModelAdmin):
     fieldsets = (
         ('Export Template', {
-            'fields': ('content_type', 'name', 'description', 'mime_type', 'file_extension')
+            'fields': ('content_type', 'name', 'description', 'mime_type', 'file_extension', 'as_attachment')
         }),
         ('Content', {
             'fields': ('template_code',),
@@ -201,7 +201,7 @@ class ExportTemplateAdmin(admin.ModelAdmin):
         })
     )
     list_display = [
-        'name', 'content_type', 'description', 'mime_type', 'file_extension',
+        'name', 'content_type', 'description', 'mime_type', 'file_extension', 'as_attachment',
     ]
     list_filter = [
         'content_type',
diff --git a/netbox/extras/api/serializers.py b/netbox/extras/api/serializers.py
index 4a1b154d3..66627bfbc 100644
--- a/netbox/extras/api/serializers.py
+++ b/netbox/extras/api/serializers.py
@@ -116,7 +116,7 @@ class ExportTemplateSerializer(ValidatedModelSerializer):
         model = ExportTemplate
         fields = [
             'id', 'url', 'display', 'content_type', 'name', 'description', 'template_code', 'mime_type',
-            'file_extension',
+            'file_extension', 'as_attachment',
         ]
 
 
diff --git a/netbox/extras/models/models.py b/netbox/extras/models/models.py
index 8b9c4cc60..226aa135a 100644
--- a/netbox/extras/models/models.py
+++ b/netbox/extras/models/models.py
@@ -253,7 +253,7 @@ class ExportTemplate(BigIDModel):
     )
     as_attachment = models.BooleanField(
         default=True,
-        help_text="Present file as attachment"
+        help_text="Download file as attachment"
     )
 
     objects = RestrictedQuerySet.as_manager()