improved null checking and handling to prevent iterating over none

UIUC-Chatbot · Feb 27, 2024 · ee5509b · ee5509b
1 parent 5d1168a
commit ee5509b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ai_ta_backend/main.py b/ai_ta_backend/main.py
@@ -739,7 +739,7 @@ def appendDocumentToGroup():
     doc_group = data['doc_group']
 
     document_data = data.get('document', {})
-    document_data['doc_groups'] = [group for group in document_data.get('doc_groups', []) if group is not None]
+    document_data['doc_groups'] = [group for group in (document_data.get('doc_groups') or []) if group is not None]
 
     try:
         document = MaterialDocument(**document_data)
@@ -766,7 +766,7 @@ def removeDocumentFromGroup():
     doc_group = data['doc_group']
 
     document_data = data.get('document', {})
-    document_data['doc_groups'] = [group for group in document_data.get('doc_groups', []) if group is not None]
+    document_data['doc_groups'] = [group for group in (document_data.get('doc_groups') or []) if group is not None]
 
     try:
         document = MaterialDocument(**document_data)