From 5753d348ea25d87ca3e192c2e7c3429b42d91acb Mon Sep 17 00:00:00 2001
From: Andreas Zweili <andreas@2li.ch>
Date: Sat, 29 Feb 2020 11:46:44 +0100
Subject: [PATCH] Set initial value for customer dropdown

---
 computers/views.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/computers/views.py b/computers/views.py
index 883f0c6..b09a9f8 100644
--- a/computers/views.py
+++ b/computers/views.py
@@ -97,6 +97,15 @@ class ComputerCreateFromCustomerView(LoginRequiredMixin, CreateView):
         kwargs.update({'user': self.request.user})
         return kwargs
 
+    def get_initial(self):
+        """
+        Set the customer dropdown to the customer from the previews view.
+        """
+        customer = get_object_or_404(Customer, id=self.kwargs.get('pk'))
+        return {
+            'customer': customer,
+        }
+
 
 class ComputerUpdateView(LoginRequiredMixin, UpdateView):
     model = Computer