Use pydash for faster deep dict checking

nationalarchives · Jan 18, 2024 · d6cb3ab · d6cb3ab
1 parent 0864dbe
commit d6cb3ab
Show file tree

Hide file tree

Showing 5 changed files with 108 additions and 201 deletions.
diff --git a/app/records/schemas/details.py b/app/records/schemas/details.py
@@ -30,6 +30,7 @@ class Record(Details):
     closure_status: str | None = None
     access_condition: str | None = None
     languages: list[str] = []
+    related_materials: list[dict] = []
     hierarchy: list[dict] = []
 
     def __init__(self, id: str):
@@ -56,6 +57,7 @@ class Aggregation(Details):
     closure_status: str | None = None
     access_condition: str | None = None
     languages: list[str] = []
+    related_materials: list[dict] = []
     hierarchy: list[dict] = []
 
     def __init__(self, id: str):

diff --git a/app/sources/rosetta/api.py b/app/sources/rosetta/api.py
@@ -101,6 +101,7 @@ def parse_results(self, raw_results, source_url):
             record.closure_status = parsed_data.closure_status()
             record.access_condition = parsed_data.access_condition()
             record.languages = parsed_data.languages()
+            record.related_materials = parsed_data.related_materials()
             record.hierarchy = (
                 parsed_data.hierarchies()[0]
                 if len(parsed_data.hierarchies())
@@ -132,6 +133,7 @@ def parse_results(self, raw_results, source_url):
             record.closure_status = parsed_data.closure_status()
             record.access_condition = parsed_data.access_condition()
             record.languages = parsed_data.languages()
+            record.related_materials = parsed_data.related_materials()
             record.hierarchy = (
                 parsed_data.hierarchies()[0]
                 if len(parsed_data.hierarchies())