1 files changed, 21 insertions, 1 deletions
diff --git a/procurement/views.py b/procurement/views.py
index 1516455..c1e66c9 100644
--- a/procurement/views.py
+++ b/procurement/views.py
@@ -1,6 +1,6 @@
 from django.views.generic import FormView, TemplateView
 
-from procurement.forms import ComponentSearchForm
+from procurement.forms import ComponentSearchForm, SupplierSearchForm
 from procurement.models import Supplier, Component, Representative
 
 
@@ -49,6 +49,26 @@ class ComponentSearchView(FormView):
 
         return super(ComponentSearchView, self).get(self.request)
 
+class SupplierSearchView(FormView):
+    template_name = "procurement/view_suppliers.html"
+    form_class = SupplierSearchForm
+
+    supplier = None
+    component_form = None
+
+    def form_valid(self, form):
+        self.supplier = form.cleaned_data['supplier']
+        if self.supplier:
+            self.component_form = ComponentSearchForm()
+            self.component_form.fields["component"].queryset = self.supplier.components
+        return super(SupplierSearchView, self).get(self.request)
+
+    def get_context_data(self):
+        context = super().get_context_data()
+        context["supplier"] = self.supplier
+        context["component_form"] = self.component_form
+        context["page_name"] = "View Supplier Details"
+        return context
 
 class DocumentationView(TemplateView):
     template_name = 'procurement/documentation.html'