Form / view for downloading stocktake info

InvenTree/stock/forms.py

@@ -96,6 +96,37 @@ class SerializeStockForm(forms.ModelForm):
         ]
 
 
+class ExportOptionsForm(HelperForm):
+    """ Form for selecting stock export options """
+
+    file_format = forms.ChoiceField(label=_('File Format'), help_text=_('Select output file format'))
+
+    include_sublocations = forms.BooleanField(required=False, initial=True, help_text=_("Include stock items in sub locations"))
+
+    class Meta:
+        model = StockLocation
+        fields = [
+            'file_format',
+            'include_sublocations',
+        ]
+
+    def get_format_choices(self):
+        """ File format choices """
+
+        choices = [
+            ('csv', 'CSV'),
+            ('xls', 'XLS'),
+            ('xlsx', 'XLSX'),
+        ]
+
+        return choices
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        self.fields['file_format'].choices = self.get_format_choices()
+
+
 class AdjustStockForm(forms.ModelForm):
     """ Form for performing simple stock adjustments.
 

InvenTree/stock/templates/stock/location.html

@@ -70,6 +70,25 @@
         sessionStorage.removeItem('inventree-show-part-locations');
     });
 
+    $("#location-export").click(function() {
+        launchModalForm("{% url 'stock-export-options' %}",
+            {
+                success: function(response) {
+                    var url = "{% url 'stock-export' %}";
+                    
+                    url += "?format=" + response.format;
+                    url += "&cascade=" + response.cascade;
+                    
+                    {% if location %}
+                    url += "&location={{ location.id }}";
+                    {% endif %}
+
+                    location.href = url;
+                }
+            }
+        );
+    });
+
     $('#location-create').click(function () {
         launchModalForm("{% url 'stock-location-create' %}",
                         {

InvenTree/stock/urls.py

@@ -51,6 +51,7 @@ stock_urls = [
 
     url(r'^adjust/?', views.StockAdjust.as_view(), name='stock-adjust'),
 
+    url(r'^export-options/?', views.StockExportOptions.as_view(), name='stock-export-options'),
     url(r'^export/?', views.StockExport.as_view(), name='stock-export'),
 
     # Individual stock items

InvenTree/stock/views.py

@@ -31,6 +31,7 @@ from .forms import EditStockItemForm
 from .forms import AdjustStockForm
 from .forms import TrackingEntryForm
 from .forms import SerializeStockForm
+from .forms import ExportOptionsForm
 
 
 class StockIndex(ListView):
@@ -119,6 +120,38 @@ class StockLocationQRCode(QRCodeView):
             return None
 
 
+class StockExportOptions(AjaxView):
+    """ Form for selecting StockExport options """
+
+    model = StockLocation
+    ajax_form_title = 'Stock Export Options'
+    form_class = ExportOptionsForm
+
+    def post(self, request, *args, **kwargs):
+
+        self.request = request
+
+        fmt = request.POST.get('file_format', 'csv').lower()
+        cascade = str2bool(request.POST.get('include_sublocations', True))
+
+        # Format a URL to redirect to
+        url = reverse('stock-export')
+
+        url += '?format=' + fmt
+        url += '&cascade=' + str(cascade)
+
+        data = {
+            'form_valid': True,
+            'format': fmt,
+            'cascade': cascade
+        }
+
+        return self.renderJsonResponse(self.request, self.form_class(), data=data)
+
+    def get(self, request, *args, **kwargs):
+        return self.renderJsonResponse(request, self.form_class())
+
+
 class StockExport(AjaxView):
     """ Export stock data from a particular location.
     Returns a file containing stock information for that location.