Add exclude param (#38)

* quick fix, use exclude * quick fix, use exclude
nulib · Oct 12, 2024 · b133bb0 · b133bb0
1 parent c0c1976
commit b133bb0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/nuldc/dump.py b/nuldc/dump.py
@@ -65,7 +65,8 @@ def dump_collection(col_id):
     params = {
         "query": f"collection.id:{col_id}",
         "size": "25",
-        "sort": "id:asc"}
+        "sort": "id:asc",
+        "_source_excludes":"embedding"}
     try:
         data = helpers.get_search_results(API,
                                           "works",
@@ -76,7 +77,7 @@ def dump_collection(col_id):
         filename = f"{slugify(col_title)}-{col_id}"
         save_files(filename, data)
     except Exception as e:
-        sys.exit(f"Error with collection {col_id}: {e} \n\n CONTEXT: {data}")
+        sys.exit(f"Error with collection {col_id}: {e} \n\n CONTEXT:")
 
 
 def dump_collections(query_string):

diff --git a/tests/test_nuldc.py b/tests/test_nuldc.py
@@ -24,7 +24,7 @@ def _mock_dcapi(next_url):
                  "parent": {"child": "child value1",
                             "label": "parent1 label"},
                  "list": ["1", "2", "3"],
-                 "embedding": [.9, .8, .7, .6]}, 
+                 "embedding": [.9, .8, .7, .6]},
                 {"id": "2",
                  "title": "2 title",
                  "parent": {"child": "child value2",
@@ -70,12 +70,12 @@ def _mock_dcapi_iiif():
                         "id": "https://example.org/iiif/result-2.json",
                         "type": "Manifest",
                         "label": {"none": ["Paged Collection - Result 2"]}
-                        },
+                    },
                 {
                         "id": "https://example.org/iiif/paged-2.json",
                         "type": "Collection",
                         "label": {"none": ["Paged Collection - Page 2"]}
-                        }
+                    }
             ]
         }
         return data
@@ -167,6 +167,6 @@ def test_sort_fields_and_values(mock_dcapi):
                 "parent" in all_fields,
                 "parent" not in some_fields,
                 # verify sort
-                ['id','list', 'parent', 'title'] == all_fields,
+                ['id', 'list', 'parent', 'title'] == all_fields,
                 ['1', '1|2|3', 'parent1 label', '1 title'] == all_values[0]]
                )