auto-fallback on db if s3 not avail

materialsproject · Dec 7, 2024 · 8c1f6d1 · 8c1f6d1
1 parent 6de5199
commit 8c1f6d1
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/mp_api/client/core/client.py b/mp_api/client/core/client.py
@@ -468,8 +468,6 @@ def _query_resource(
                 }
             )
             and num_chunks is None
-            and "substrates" not in self.suffix
-            and "phonon" not in self.suffix
         )
 
         if fields:
@@ -506,11 +504,22 @@ def _query_resource(
 
                 keys = []
                 for page in pages:
-                    for obj in page["Contents"]:
+                    for obj in page.get("Contents", []):
                         key = obj.get("Key")
                         if key:
                             keys.append(key)
 
+                if len(keys) < 1:
+                    self._submit_requests(
+                        url=url,
+                        criteria=criteria,
+                        use_document_model=use_document_model,
+                        parallel_param=parallel_param,
+                        num_chunks=num_chunks,
+                        chunk_size=chunk_size,
+                        timeout=timeout,
+                    )
+
                 decoder = (
                     MontyDecoder().decode if self.monty_decode else json_util.loads
                 )