Bugfix when there's no partial submission found

Firesphere · Jun 19, 2018 · b8a99ac · b8a99ac
1 parent 002cf17
commit b8a99ac
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/extensions/SubmittedFormExtension.php b/src/extensions/SubmittedFormExtension.php
@@ -30,12 +30,14 @@ public function updateAfterProcess()
         $partialID = Controller::curr()->getRequest()->getSession()->get(PartialUserFormController::SESSION_KEY);
         /** @var PartialFormSubmission $partialForm */
         $partialForm = PartialFormSubmission::get()->byID($partialID);
-        foreach ($partialForm->PartialFields() as $field) {
-            $field->delete();
-            $field->destroy();
+        if ($partialForm) {
+            foreach ($partialForm->PartialFields() as $field) {
+                $field->delete();
+                $field->destroy();
+            }
+            $partialForm->delete();
+            $partialForm->destroy();
         }
-        $partialForm->delete();
-        $partialForm->destroy();
         Controller::curr()->getRequest()->getSession()->clear(PartialUserFormController::SESSION_KEY);
     }