diff --git a/ingestion/ingest.py b/ingestion/ingest.py index 81434094..1da69fa7 100644 --- 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)