Merge pull request #24 from CanDIG/mshadbolt/null-error-handling

DIG-1527: Catch error when treatment_type is null
CanDIG · Mar 11, 2024 · 4bd3910 · 4bd3910
2 parents cd22b9b + d2d5a03
commit 4bd3910
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/query_server/query_operations.py b/query_server/query_operations.py
@@ -95,8 +95,12 @@ def get_summary_stats(donors, headers):
     treatment_type_count = {}
     for treatment in treatments:
         if treatment["submitter_donor_id"] in donors_by_id:
-            for treatment_type in treatment["treatment_type"]:
-                add_or_increment(treatment_type_count, treatment_type)
+            try:
+                for treatment_type in treatment["treatment_type"]:
+                    add_or_increment(treatment_type_count, treatment_type)
+            except TypeError as e:
+                print(e)
+                pass
 
     return {
         'age_at_diagnosis': age_at_diagnosis,