better key naming

johnseekins · johnseekins · commit 4d68bf8bd749 · 2025-09-19T15:03:26.000-06:00
Signed-off-by: John Seekins &lt;john.seekins@spoileralert.com&gt;
diff --git a/enrichers/openstreetmap.py b/enrichers/openstreetmap.py
@@ -15,7 +15,7 @@ def search(facility_name: str, address: dict) -> dict:
     search_name = clean_facility_name(facility_name)
     search_url = "https://nominatim.openstreetmap.org/search"
     resp_info = copy.deepcopy(enrich_resp_schema)
-    resp_info["query_type"] = "openstreetmap"
+    resp_info["enrichment_type"] = "openstreetmap"
     data = []
     if not address:
         logger.debug("No address for %s, simply searching for name", facility_name)
diff --git a/enrichers/wikidata.py b/enrichers/wikidata.py
@@ -25,7 +25,7 @@ def search(facility_name: str) -> dict:
         "limit": 3,
     }
     resp_info = copy.deepcopy(enrich_resp_schema)
-    resp_info["query_type"] = "wikidata"
+    resp_info["enrichment_type"] = "wikidata"
     data = {}
     try:
         response = session.get(search_url, params=params, timeout=10)  # type: ignore [arg-type]
diff --git a/enrichers/wikipedia.py b/enrichers/wikipedia.py
@@ -30,7 +30,7 @@ def search(facility_name: str) -> dict:
         "processing",
     ]
     resp_info = copy.deepcopy(enrich_resp_schema)
-    resp_info["query_type"] = "wikipedia"
+    resp_info["enrichment_type"] = "wikipedia"
     # Clean facility name for search
     search_name: str = clean_facility_name(facility_name)
     logger.debug("Searching Wikipedia for %s", facility_name)
diff --git a/schemas.py b/schemas.py
@@ -121,7 +121,7 @@
     "url": "",
     "details": {},
     "method": "none",
-    "query_type": "",
+    "enrichment_type": "",
 }
 
 enrich_allowed_types = [

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@ def search(facility_name: str) -> dict:`
`25`	`25`	`"limit": 3,`
`26`	`26`	`}`
`27`	`27`	`resp_info = copy.deepcopy(enrich_resp_schema)`
`28`		`- resp_info["query_type"] = "wikidata"`
	`28`	`+ resp_info["enrichment_type"] = "wikidata"`
`29`	`29`	`data = {}`
`30`	`30`	`try:`
`31`	`31`	`response = session.get(search_url, params=params, timeout=10) # type: ignore [arg-type]`
Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ def search(facility_name: str) -> dict:`
`30`	`30`	`"processing",`
`31`	`31`	`]`
`32`	`32`	`resp_info = copy.deepcopy(enrich_resp_schema)`
`33`		`- resp_info["query_type"] = "wikipedia"`
	`33`	`+ resp_info["enrichment_type"] = "wikipedia"`
`34`	`34`	`# Clean facility name for search`
`35`	`35`	`search_name: str = clean_facility_name(facility_name)`
`36`	`36`	`logger.debug("Searching Wikipedia for %s", facility_name)`
Original file line number	Diff line number	Diff line change
`@@ -121,7 +121,7 @@`
`121`	`121`	`"url": "",`
`122`	`122`	`"details": {},`
`123`	`123`	`"method": "none",`
`124`		`- "query_type": "",`
	`124`	`+ "enrichment_type": "",`
`125`	`125`	`}`
`126`	`126`
`127`	`127`	`enrich_allowed_types = [`