view tools/flag_data.py @ 917:0365fdbb4d78

Fix app conflict with messages. Django's messages app label conflicts with our messages app. We can't easily rename our label as that will make us rename database tables. Since our app came first we'll just customize Django messages label. For Django 1.7.7 upgrade.
author Brian Neal <bgneal@gmail.com>
date Mon, 06 Apr 2015 20:02:25 -0500 (2015-04-07)
parents 678a1a2ef55a
children
line wrap: on
line source
"""
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
import sys
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)):
            sys.stderr.write("Could not find icon for %s (%s)\n" % (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()