Put a check for files.

The script was throwing an error. CRITICAL: NameError("name 'files' is not defined",)
EUDAT-B2SHARE · Apr 12, 2022 · c1facd3 · c1facd3
1 parent 0d83d90
commit c1facd3
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/check_b2share.py b/check_b2share.py
@@ -149,13 +149,14 @@ def get_dict_from_url(url, verify_tls_cert=False, verbosity=False):
 
             rec_with_files_url = None
             for hit in search_results['hits']['hits']:
-                # Check if there are files in the record
-                if len(hit['files']) > 0:
-                    # NTS: Could throw KeyError if there is something
-                    # seriously wrong or B2SHARE REST API responses have
-                    # changed.
-                    rec_with_files_url = hit['links']['self']
-                    break
+                if 'files' in hit:
+                    # Check if there are files in the record
+                    if len(hit['files']) > 0:
+                        # NTS: Could throw KeyError if there is something
+                        # seriously wrong or B2SHARE REST API responses have
+                        # changed.
+                        rec_with_files_url = hit['links']['self']
+                        break
 
             if rec_with_files_url: