fix missing root form due to API change (#44)

bellingcat · Oct 28, 2024 · 9db40ef · 9db40ef
1 parent 596b465
commit 9db40ef
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/edgar_tool/text_search.py b/edgar_tool/text_search.py
@@ -82,8 +82,10 @@ def _parse_row(row: Dict[str, Any]) -> Dict[str, Any]:
         ciks_trimmed: List[str] = [c.lstrip("0") for c in ciks]
 
         # Get form and human readable name
-        root_form = _source.get("root_form")
-        form_name = TEXT_SEARCH_FORM_MAPPING.get(root_form, {}).get("title", "")
+        root_forms = _source.get("root_forms")
+        form_name = [TEXT_SEARCH_FORM_MAPPING.get(form, {}).get("title", "") for form in root_forms]
+        root_forms = unpack_singleton_list(root_forms)
+        form_name = unpack_singleton_list(form_name)
 
         # Build adsh for url
         data_adsh = _source.get("adsh", "")
@@ -145,7 +147,7 @@ def _parse_row(row: Dict[str, Any]) -> Dict[str, Any]:
         ]
 
         parsed = {
-            "root_form": root_form,
+            "root_form": root_forms,
             "form_name": form_name,
             "filed_at": filed_at,
             "reporting_for": end_date,