applying the mongo operator treatment recursively to arrays too in

shirouto · Sep 23, 2020 · f88a539 · f88a539
1 parent ab0cbf0
commit f88a539
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/mongoke/templates/resolvers_support.py b/mongoke/templates/resolvers_support.py
@@ -201,14 +201,19 @@ def make_edge(node, cursorField):
 
 def strip_nones(x: dict):
     result = {}
+
     for k, v in x.items():
         if not v == None and v != {}:
             if k in MONGODB_OPERATORS:
                 k = '$' + k
-            if isinstance(v, dict):
+
+            if isinstance(v, list):
+                result[k] = [strip_nones(x) for x in v if x != None and x != {}]
+            elif isinstance(v, dict):
                 result[k] = strip_nones(v)
             else:
                 result[k] = v
+
     return result
 
-'''
+'''