# HG changeset patch
# User Brian Neal <bgneal@gmail.com>
# Date 1343263236 18000
# Node ID 34b1dd3f84fa6292aae54338c535f5aa307e3dbc
# Parent  c8b4dfb2d1e526cd390a9b3cd6c6d6a1b1070170
Created a tool to generate ISO-3166 country data as a JSON file.

diff -r c8b4dfb2d1e5 -r 34b1dd3f84fa tools/flag_data.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tools/flag_data.py	Wed Jul 25 19:40:36 2012 -0500
@@ -0,0 +1,46 @@
+"""
+Application to read the ISO-3166 country data and output a JSON datastructure
+for use in the SG101 code. We also print a report of any missing country
+icons. If we don't have an icon for a country, we don't include it in our JSON
+data.
+
+"""
+import argparse
+import json
+import os.path
+from xml.etree.ElementTree import ElementTree
+
+
+def main():
+    parser = argparse.ArgumentParser(description=("Transform Debian's ISO-3166 "
+        "data into JSON for use on SG101"))
+    parser.add_argument('--xml', '-x', required=True, help='path to XML file')
+    parser.add_argument('--icon-dir', '-i', required=True,
+            help='path to icon directory')
+
+    args = parser.parse_args()
+
+    xml_file = os.path.expanduser(args.xml)
+    icon_dir = os.path.expanduser(args.icon_dir)
+
+    with open(xml_file, 'r') as fp:
+        et = ElementTree(file=fp)
+
+    country_data = {}
+    for node in et.iterfind('iso_3166_entry'):
+        code = node.get('alpha_2_code').lower()
+        name = node.get('common_name', node.get('name'))
+
+        # see if we have an icon for this country
+
+        if not os.path.exists(os.path.join(icon_dir, '%s.png' % code)):
+            print "Could not find icon for %s (%s)" % (name, code)
+        else:
+            country_data[code] = name
+
+    s = json.dumps(country_data, indent=4, sort_keys=True, ensure_ascii=False)
+    print s.encode('utf-8')
+
+
+if __name__ == '__main__':
+    main()