Skip to content

Commit

Permalink
Update source of continentmap to original CSV file.
Browse files Browse the repository at this point in the history
Now fetching from https://github.com/datasets/country-codes repository.
  • Loading branch information
fancycode committed Mar 11, 2024
1 parent 1fa731f commit b0f2e6e
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 6 deletions.
2 changes: 1 addition & 1 deletion continentmap.go
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
package signaling

// This file has been automatically generated, do not modify.
// Source: https://datahub.io/core/country-codes/r/country-codes.json
// Source: https://github.com/datasets/country-codes/raw/master/data/country-codes.csv

var (
ContinentMap = map[string][]string{
Expand Down
11 changes: 6 additions & 5 deletions scripts/get_continent_map.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,11 +26,11 @@
from cStringIO import StringIO
except ImportError:
from io import StringIO
import json
import csv
import subprocess
import sys

URL = 'https://datahub.io/core/country-codes/r/country-codes.json'
URL = 'https://github.com/datasets/country-codes/raw/master/data/country-codes.csv'

def tostr(s):
if isinstance(s, bytes) and not isinstance(s, str):
Expand All @@ -55,12 +55,13 @@ def generate_map(filename):
'-L',
URL,
])
data = json.loads(tostr(data))

reader = csv.DictReader(StringIO(tostr(data)), delimiter=',')
continents = {}
for entry in data:
for entry in reader:
country = entry['ISO3166-1-Alpha-2']
continent = entry['Continent']
if country is None and continent is None:
if not country and not continent:
continue

continents.setdefault(country, []).append(continent)
Expand Down

0 comments on commit b0f2e6e

Please sign in to comment.