Refactor the previous commit a bit

kiwitcms · Nov 28, 2023 · 8a5ff1a · 8a5ff1a
1 parent 31afc71
commit 8a5ff1a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tcms/testplans/views.py b/tcms/testplans/views.py
@@ -208,10 +208,8 @@ def get_context_data(self, **kwargs):
 
     def get_form(self, form_class=None):
         form = super().get_form(form_class)
-        if self.request.POST.get("product"):
-            form.populate(product_pk=self.request.POST["product"])
-        else:
-            form.populate(product_pk=self.object.product_id)
+        product_id = self.request.POST.get("product", self.object.product_id)
+        form.populate(product_pk=product_id)
         return form
 
     def get_form_kwargs(self):