fix: Allow deleting subfields in review plan pico

datakind · Mar 27, 2024 · e21c290 · e21c290
1 parent 66c17e3
commit e21c290
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/colandr/apis/resources/review_plans.py b/colandr/apis/resources/review_plans.py
@@ -120,8 +120,16 @@ def delete(self, id, fields):
             for field in fields:
                 if field == "objective":
                     review_plan.objective = ""
-                elif field == "pico":
-                    review_plan.pico = {}
+                elif field.startswith("pico"):
+                    if field == "pico":
+                        review_plan.pico = {}
+                    else:
+                        _, subfield = field.split(".")
+                        review_plan.pico = {
+                            key: val
+                            for key, val in review_plan.pico.items()
+                            if key != subfield
+                        }
                 else:
                     setattr(review_plan, field, [])
         else: