Fix for metadata

datakind · May 31, 2024 · 4978e57 · 4978e57
1 parent 911c396
commit 4978e57
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ingestion/ingest.py b/ingestion/ingest.py
@@ -313,8 +313,9 @@ def save_openapi_data(files_dir, conn, api_name):
                         r["api_description"] += f' : {meta["get"]["description"]}'
                     r["api_definition"] = str(meta)
                     r["file_name"] = f
-                    if 'location_name' in df.columns:
-                        r['countries'] = sorted(df['location_name'].unique())
+                    for field in ['location_name', 'origin_location_name']:
+                        if field in df.columns:
+                            r['countries'] = sorted(df[field].unique())
 
                     table_metadata.append(r)