From 67d9d2bb04e4eb46d221be0f576bac679de46457 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Mon, 13 Apr 2020 12:47:06 +1000
Subject: [PATCH] Separate views for customer / supplier / manufacturer

(cherry picked from commit 9e9e29679dcb5466c62b0a436c7a4234dc78c35a)
---
 .../static/script/inventree/company.js        |  2 +-
 .../company/templates/company/index.html      |  7 +++--
 InvenTree/company/urls.py                     |  6 ++--
 InvenTree/company/views.py                    | 31 +++++++++++++++++++
 InvenTree/templates/navbar.html               | 10 ++++--
 5 files changed, 49 insertions(+), 7 deletions(-)

diff --git a/InvenTree/InvenTree/static/script/inventree/company.js b/InvenTree/InvenTree/static/script/inventree/company.js
index f007ed8c14..3bb9da60b8 100644
--- a/InvenTree/InvenTree/static/script/inventree/company.js
+++ b/InvenTree/InvenTree/static/script/inventree/company.js
@@ -40,7 +40,7 @@ function loadCompanyTable(table, url, options={}) {
                     var html = imageHoverIcon(row.image) + renderLink(value, row.url);
 
                     if (row.is_customer) {
-                        html += `<span title='Customer' class='fas fa-user label-right'></span>`;
+                        html += `<span title='Customer' class='fas fa-user-tie label-right'></span>`;
                     }
                     
                     if (row.is_manufacturer) {
diff --git a/InvenTree/company/templates/company/index.html b/InvenTree/company/templates/company/index.html
index 245e357489..e264c376c9 100644
--- a/InvenTree/company/templates/company/index.html
+++ b/InvenTree/company/templates/company/index.html
@@ -9,12 +9,12 @@ InvenTree | {% trans "Supplier List" %}
 
 {% block content %}
 
-<h3>{% trans "Supplier List" %}</h3>
+<h3>{{ title }}</h3>
 <hr>
 
 <div id='button-toolbar'>
     <div class='btn-group'>
-        <button type='button' class="btn btn-success" id='new-company' title='Add new supplier'>{% trans "New Supplier" %}</button>
+        <button type='button' class="btn btn-success" id='new-company'>{{ new_button_text }}</button>
     </div>
 </div>
 
@@ -35,6 +35,9 @@ InvenTree | {% trans "Supplier List" %}
 
     loadCompanyTable("#company-table", "{% url 'api-company-list' %}",
         {
+            params: {
+                {% for key,value in filters.items %}{{ key }}: "{{ value }}",{% endfor %}
+            }
         }
     );
 
diff --git a/InvenTree/company/urls.py b/InvenTree/company/urls.py
index c00d1e7a4b..848463610c 100644
--- a/InvenTree/company/urls.py
+++ b/InvenTree/company/urls.py
@@ -33,10 +33,12 @@ company_urls = [
 
     url(r'^(?P<pk>\d+)/', include(company_detail_urls)),
 
-    url(r'', views.CompanyIndex.as_view(), name='company-index'),
+    url(r'suppliers/', views.CompanyIndex.as_view(), name='supplier-index'),
+    url(r'manufacturers/', views.CompanyIndex.as_view(), name='manufacturer-index'),
+    url(r'customers/', views.CompanyIndex.as_view(), name='customer-index'),
 
     # Redirect any other patterns to the 'company' index which displays all companies
-    url(r'^.*$', RedirectView.as_view(url='', permanent=False), name='company-index'),
+    url(r'^.*$', views.CompanyIndex.as_view(), name='company-index'),
 ]
 
 price_break_urls = [
diff --git a/InvenTree/company/views.py b/InvenTree/company/views.py
index 243a982872..36e1e7bd91 100644
--- a/InvenTree/company/views.py
+++ b/InvenTree/company/views.py
@@ -39,6 +39,37 @@ class CompanyIndex(ListView):
     context_object_name = 'companies'
     paginate_by = 50
 
+    def get_context_data(self, **kwargs):
+
+        ctx = super().get_context_data(**kwargs)
+
+        url = self.request.path
+
+        print(url)
+        print(reverse('supplier-index'))
+        print(reverse('manufacturer-index'))
+        print(reverse('customer-index'))
+
+        if url == reverse('supplier-index'):
+            ctx["title"] = _("Suppliers")
+            ctx['new_button_text'] = _("New Supplier")
+            ctx["filters"] = {"is_supplier": "true"}
+        elif url == reverse('manufacturer-index'):
+            ctx["title"] = _("Manufacturers")
+            ctx['new_button_text'] = _("New Manufacturer")
+            ctx["filters"] = {"is_manufacturer": "true"}
+        elif url == reverse('customer-index'):
+            ctx["title"] = _("Customers")
+            ctx['new_button_text'] = _("New Customer")
+            ctx["filters"] = {"is_customer": "true"}
+        else:
+            ctx["title"] = _("Companies")
+            ctx["new_button_text"] = _("New Company")
+            ctx["filters"] = {}
+
+        return ctx
+
+
     def get_queryset(self):
         """ Retrieve the Company queryset based on HTTP request parameters.
 
diff --git a/InvenTree/templates/navbar.html b/InvenTree/templates/navbar.html
index 202fb2a0ce..c41ef3718f 100644
--- a/InvenTree/templates/navbar.html
+++ b/InvenTree/templates/navbar.html
@@ -13,11 +13,17 @@
       <li class='nav navbar-nav'>
         <a class='dropdown-toggle' data-toggle='dropdown' href='#'><span class='fas fa-shopping-cart icon-header'></span>{% trans "Buy" %}</a>
         <ul class='dropdown-menu'>
-          <li><a href="{% url 'company-index' %}"><span class='fas fa-building icon-header'></span>{% trans "Suppliers" %}</a></li>
-          <li><a href="{% url 'company-index' %}"><span class='fas fa-industry icon-header'></span>{% trans "Manufacturers" %}</a></li>
+          <li><a href="{% url 'supplier-index' %}"><span class='fas fa-building icon-header'></span>{% trans "Suppliers" %}</a></li>
+          <li><a href="{% url 'manufacturer-index' %}"><span class='fas fa-industry icon-header'></span>{% trans "Manufacturers" %}</a></li>
           <li><a href="{% url 'po-index' %}"><span class='fas fa-list icon-header'></span>{% trans "Purchase Orders" %}</a></li>
         </ul>
       </li>
+      <li class='nav navbar-nav'>
+        <a class='dropdown-toggle' data-toggle='dropdown' href='#'><span class='fas fa-truck icon-header'></span>{% trans "Sell" %}</a>
+        <ul class='dropdown-menu'>
+          <li><a href="{% url 'customer-index' %}"><span class='fas fa-user-tie icon-header'></span>{% trans "Customers" %}</a>
+        </ul>
+      </li>
     </ul>
     <ul class="nav navbar-nav navbar-right">
         {% include "search_form.html" %}