Script to fetch country codes #2559

Would work, if not for the mongo perf issues

verification/scripts/README.md

@@ -21,4 +21,11 @@ mongo "mongodb://user:pass@some-connection-string/somedb" \
 ```
 
 NOTE: For localhost (during dev) the DB connection string is "mongodb://localhost:27017/covid19" but for the atlas DB you have to use a proper connection string as the data service or curator services are doing on AWS.
-You can go to the web atlas console and generate creds for you, they'll also show you the connection string to use those creds you just created.
+You can go to the web atlas console and generate creds for you, they'll also show you the connection string to use those creds you just created.
+
+## Generating list of country codes
+
+The script `country_codes_map` uses the curator service to generate a list of country codes found in the case data, and any of the names that we would display or parse from ingestion scripts for each country. You need to set two environment variables:
+
+ - `GDOTH_API_BASEURL` is the base URL for the Global.health instance API: e.g. `https://data.covid-19.global.health/`
+ - `GDOTH_API_KEY` is the API key for a user (any user, no specific roles needed) on that instance

verification/scripts/country_codes_map/country_codes_map.py

@@ -0,0 +1,16 @@
+import json
+from os import environ
+import requests
+
+def retrieve_map(api_key: str, base_url: str) -> str:
+    api_endpoint = f"{base_url}/api/geocode/countryNames"
+    response = requests.get(api_endpoint, headers={ 'X-API-Key': api_key })
+    if response.ok:
+        return response.json()
+    else:
+        raise ValueError(f"Got {response.status_code} response from server: {response.text}")
+
+if __name__ == '__main__':
+    api_key = environ.get('GDOTH_API_KEY')
+    base_url = environ.get('GDOTH_API_BASEURL')
+    print(json.dumps(retrieve_map(api_key, base_url), indent=4))

verification/scripts/country_codes_map/pyproject.toml

@@ -0,0 +1,16 @@
+[tool.poetry]
+name = "country_codes_map"
+version = "0.1.0"
+description = "Generate a map of all the used country codes and their names"
+authors = ["Global.health <info@global.health>"]
+
+[tool.poetry.dependencies]
+python = "^3.9"
+requests = "^2.27.1"
+
+[tool.poetry.dev-dependencies]
+pytest = "^5.2"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"