From ddb041fe44800cc4a5de49cea18669e75c8169dc Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Mon, 9 Sep 2019 19:59:56 +1000
Subject: [PATCH] Filter by category too

---
 InvenTree/part/templates/part/category.html |  8 ++++++
 InvenTree/part/views.py                     | 32 ++++++++++++---------
 InvenTree/templates/stock_table.html        |  2 +-
 3 files changed, 28 insertions(+), 14 deletions(-)

diff --git a/InvenTree/part/templates/part/category.html b/InvenTree/part/templates/part/category.html
index 6a076ef8ca..26098be946 100644
--- a/InvenTree/part/templates/part/category.html
+++ b/InvenTree/part/templates/part/category.html
@@ -44,6 +44,7 @@
 
 <div id='button-toolbar'>
     <div class='button-toolbar container-fluid' style="float: right;">
+        <button class='btn btn-default' id='part-export' title='Export Part Data'>Export</button>
         <button class='btn btn-success' id='part-create'>New Part</button>
         <div class='dropdown' style='float: right;'>
             <button id='part-options' class='btn btn-primary dropdown-toggle' type='button' data-toggle="dropdown">Options<span class='caret'></span></button>
@@ -104,6 +105,13 @@
                         });
     })
 
+    $("#part-export").click(function() {
+        
+        var url = "{% url 'part-export' %}?category={{ category.id }}";
+
+        location.href = url;
+    });
+
     $("#part-create").click(function() {
         launchModalForm(
             "{% url 'part-create' %}",
diff --git a/InvenTree/part/views.py b/InvenTree/part/views.py
index 7959e2b6a6..26906cfad2 100644
--- a/InvenTree/part/views.py
+++ b/InvenTree/part/views.py
@@ -1166,12 +1166,29 @@ class PartExport(AjaxView):
         """ Extract part list from the POST parameters.
         Parts can be supplied as:
         
-        - List of part PK values
         - Part category
+        - List of part PK values
         """
 
-        part_list = Part.objects.all()
+        # Filter by part category
+        cat_id = request.GET.get('category', None)
 
+        print('cat_id:', cat_id)
+
+        part_list = None
+
+        if cat_id is not None:
+            try:
+                category = PartCategory.objects.get(pk=cat_id)
+                part_list = category.get_parts()
+            except (ValueError, PartCategory.DoesNotExist):
+                pass
+
+        # Backup - All parts
+        if part_list is None:
+            part_list = Part.objects.all()
+
+        # Also optionally filter by explicit list of part IDs
         part_ids = request.GET.get('parts', '')
         parts = []
 
@@ -1181,20 +1198,9 @@ class PartExport(AjaxView):
             except ValueError:
                 pass
 
-        # Filter by list of Part IDs
         if len(parts) > 0:
             part_list = part_list.filter(pk__in=parts)
 
-        # Filter by part category
-        cat_id = request.GET.get('category', None)
-
-        if cat_id is not None:
-            try:
-                category = PartCategory.objects.get(cat_id)
-                part_list = part_list.filter(category=category)
-            except (ValueError, PartCategory.DoesNotExist):
-                pass
-
         # Prefetch related fields to reduce DB hits
         part_list = part_list.prefetch_related(
             'category',
diff --git a/InvenTree/templates/stock_table.html b/InvenTree/templates/stock_table.html
index 5b707eab24..c5d16b8ed6 100644
--- a/InvenTree/templates/stock_table.html
+++ b/InvenTree/templates/stock_table.html
@@ -1,6 +1,6 @@
 <div id='button-toolbar'>
     <div class='button-toolbar container-fluid' style='float: right;'>
-        <button class='btn btn-success' id='stock-export' title='Export Stock Information'>Export</button>
+        <button class='btn btn-default' id='stock-export' title='Export Stock Information'>Export</button>
         {% if not part or part.is_template == False %}
         <button class="btn btn-success" id='item-create'>New Stock Item</button>
         {% endif %}