parse_wikipedia_tables.py

#!/usr/bin/env python3
# -*- coding: UTF-8 -*-

# /**
# * Software Name : MCC_MNC
# * Version : 0.1
# *
# * Copyright 2020. Benoit Michau. P1 Security.
# *
# * This program is free software: you can redistribute it and/or modify
# * it under the terms of the GNU Affero General Public License as
# * published by the Free Software Foundation, either version 3 of the
# * License, or (at your option) any later version.
# *
# * This program is distributed in the hope that it will be useful,
# * but WITHOUT ANY WARRANTY; without even the implied warranty of
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# * GNU Affero General Public License for more details.
# *
# * You should have received a copy of the GNU Affero General Public License
# * along with this program.  If not, see <https://www.gnu.org/licenses/>.
# *
# *--------------------------------------------------------
# * File Name : parse_wikipedia_tables.py
# * Created : 2020-09-14
# * Authors : Benoit Michau 
# *--------------------------------------------------------
# */


import sys
import argparse
import urllib.request
import json
import re

from pprint import PrettyPrinter
from lxml   import etree


PATH_PRE = 'raw/'

RE_WIKI_REF = re.compile(r'.*(\[.*\]){1,}$')

def _strip_wiki_ref(s):
    m = RE_WIKI_REF.match(s)
    if m:
        return s[:s.find('[')].strip()
    else:
        return s.strip()


RE_WIKI_REFNOTE = re.compile(r'\[[ a-zA-Z0-9]*\]')

def _strip_wiki_refnote(s):
    m = RE_WIKI_REFNOTE.match(s)
    if m:
        return s[m.end():].strip()
    else:
        return s.strip()


#------------------------------------------------------------------------------#
# parsing Wikipedia ISO-3166 codes
#------------------------------------------------------------------------------#

URL_PREF          = "https://en.wikipedia.org"
URL_CODE_ALPHA_2  = "https://en.wikipedia.org/wiki/List_of_ISO_3166_country_codes"


REC_ISO3166 = {
    'country_name'          : '', # str, ISO 3166 country name
    'country_url'           : '', # url to the Wikipedia page of the country
    'state_name'            : '', # str
    'sovereignity'          : '', # str: UN member, UN observer, antarctic, disputed or another country_name
    'code_alpha_2'          : '', # 2-char
    'code_alpha_3'          : '', # 3-char
    'code_num'              : '', # 3-digit
    #'regions'               : {}, # dict of region code, region name
    'regions_url'           : '', # url to the Wikipedia page of the regional code
    'cc_tld'                : '', # str
    'cc_tld_url'            : '', # url to the Wikipedia page of the ccTLD
    }


RE_TEXT_INVAL   = re.compile(r'[!\:\{\}]')
is_text_valid   = lambda t: True if t.strip() and not RE_TEXT_INVAL.search(t) else False


SOVEREIGNITY_LUT = {
    'finland'           : 'FI',
    'united states'     : 'US',
    'united kingdom'    : 'GB',
    'netherlands'       : 'NL',
    'norway'            : 'NO',
    'australia'         : 'AU',
    'new zealand'       : 'NZ',
    'denmark'           : 'DK',
    'france'            : 'FR',
    'british crown'     : 'GB', # this is not exactly true, however...
    'china'             : 'CN',
    'un member state'   : 'UN member',
    'un observer'       : 'UN observer',
    'un observer state' : 'UN observer',
    'antarctic treaty'  : 'Antarctica',
    'disputed'          : 'disputed'
    }


CRAPPY_NAMES = {
    'Georgia (country)',
    'List of islands of the Netherlands Antilles',
    'List of French islands in the Indian and Pacific oceans',
    }


def import_html_doc(url):
    resp = urllib.request.urlopen(url)
    if resp.code == 200:
        R = etree.parse(resp, etree.HTMLParser()).getroot()
    else:
        raise(Exception('resource %s not available, HTTP code %i' % (url, resp.code)))
    #
    #return R.xpath('//table')
    return R


def explore_text(E):
    #print(E)
    if hasattr(E, 'text') and E.text is not None and is_text_valid(E.text):
        return E
    for e in E:
        t = explore_text(e)
        if t is not None and hasattr(t, 'text') and t.text is not None and is_text_valid(t.text):
            return t


def _get_country_url(e):
    e = explore_text(e)
    if e is not None:
        values = e.values()
        if len(values) == 2 and values[0][:6] == '/wiki/':
            url, name = values
            if name in CRAPPY_NAMES:
                name = e.text.strip()
            return name, URL_PREF + url.strip()
        elif len(values) == 3 and values[0][:6] == '/wiki/':
            url, _, name = values
            if name in CRAPPY_NAMES:
                name = e.text.strip()
            return name, URL_PREF + url.strip()
        #print(e.text)
    return None, None


def read_entry_iso3166(T, off):
    L   = T[off]
    rec = dict(REC_ISO3166)
    #
    rec['country_name'], rec['country_url'] = _get_country_url(L[0])
    rec['state_name']   = explore_text(L[1]).text.strip()
    rec['sovereignity'] = SOVEREIGNITY_LUT[explore_text(L[2]).text.strip().lower()]
    rec['code_alpha_2'] = explore_text(L[3]).text.strip().upper()
    rec['code_alpha_3'] = explore_text(L[4]).text.strip().upper()
    rec['code_num']     = explore_text(L[5]).text.strip()
    #rec['regions']      =
    rec['regions_url']  = URL_PREF + L[6][0].values()[0]
    rec['cc_tld']       = explore_text(L[7]).text.strip().lower()
    rec['cc_tld_url']   = URL_PREF + explore_text(L[7]).values()[0]
    if rec['cc_tld'] and rec['cc_tld'][0] != '.':
        rec['cc_tld']     = ''
        rec['cc_tld_url'] = ''
    return rec


def parse_table_iso3166():
    T    = import_html_doc(URL_CODE_ALPHA_2).xpath('//table')
    T_CC = T[0][0]
    D    = {}
    for i in range(2, len(T_CC)):
        try:
            rec = read_entry_iso3166(T_CC, i)
        except IndexError:
            #print('no entry at rank %i, after %s' % (i, rec['country_name']))
            pass
        else:
            if rec['code_alpha_2'] in D:
                raise(Exception('duplicate entries for %s' % rec['code_alpha_2']))
            else:
                D[rec['code_alpha_2']] = rec
    print('[+] parsed %i ISO-3166 country codes' % len(D))
    return D


#------------------------------------------------------------------------------#
# parsing Wikipedia MCC and MNC
#------------------------------------------------------------------------------#

# MCC and international operators
URL_MCC     = "https://en.wikipedia.org/wiki/Mobile_country_code"
# nationa operators
URL_MNC_EU  = "https://en.wikipedia.org/wiki/Mobile_Network_Codes_in_ITU_region_2xx_(Europe)"
URL_MNC_NA  = "https://en.wikipedia.org/wiki/Mobile_Network_Codes_in_ITU_region_3xx_(North_America)"
URL_MNC_AS  = "https://en.wikipedia.org/wiki/Mobile_Network_Codes_in_ITU_region_4xx_(Asia)"
URL_MNC_OC  = "https://en.wikipedia.org/wiki/Mobile_Network_Codes_in_ITU_region_5xx_(Oceania)"
URL_MNC_AF  = "https://en.wikipedia.org/wiki/Mobile_Network_Codes_in_ITU_region_6xx_(Africa)"
URL_MNC_SA  = "https://en.wikipedia.org/wiki/Mobile_Network_Codes_in_ITU_region_7xx_(South_America)"


REC_MCC = {
    'mcc'           : '', # 3-digit str
    'country_name'  : '', # str
    'country_url'   : '', # str
    'code_alpha_2'  : '', # 2-char
    'mcc_url'       : '', # str
    'authority'     : '', # str
    'notes'         : '', # str
    }


REC_MNC = {
    'country_name'  : '',   # str
    'country_url'   : '',   # str
    'country_sub'   : None, # 2-tuple of str (country_name, country_url)
    'codes_alpha_2' : [],   # list of 2-char
    'mcc'           : '',   # 3-digit str
    'mnc'           : '',   # 2 or 3-digit str
    'brand'         : '',   # str
    'operator'      : '',   # str
    'status'        : '',   # 'operational' or 'unknown'
    'bands'         : '',   # str
    'notes'         : '',   # str
    }


def read_entry_mcc(T, off):
    L   = T[off]
    rec = dict(REC_MCC)
    if len(L) < 4:
        full                = False
        rec['mcc']          = explore_text(L[0]).text.strip().upper()
        if len(L) > 2:
            rec['authority'] = _strip_wiki_ref(''.join(L[2].itertext()))
        if len(L) > 3:
            rec['notes']     = _strip_wiki_ref(''.join(L[3].itertext()))
    else:
        full                = True
        rec['mcc']          = explore_text(L[0]).text.strip().upper()
        rec['code_alpha_2'] = explore_text(L[2]).text.strip()
        rec['country_name'], rec['country_url'] = _get_country_url(L[1])
        if rec['country_name'] is None:
            # hidden HTML element with 1st letter of the country only, go to next
            rec['country_name'], rec['country_url'] = _get_country_url(L[1][1])
        mcc_url = explore_text(L[3]).values()
        if mcc_url:
            rec['mcc_url']      = URL_PREF + mcc_url[0]
        if len(L) > 4:
            rec['authority'] = _strip_wiki_ref(''.join(L[4].itertext()))
        if len(L) > 5:
            rec['notes']     = _strip_wiki_ref(''.join(L[5].itertext()))
    return rec, full


def parse_table_mcc():
    # warning: for some MCC, there are duplicated entries (mostly for islands...)
    T     = import_html_doc(URL_MCC).xpath('//table')
    T_MCC = T[1][1]
    L     = []
    cc2   = set()
    mcc   = {}
    for i in range(1, len(T_MCC)):
        rec, full = read_entry_mcc(T_MCC, i)
        if rec['mcc'] == 'YT':
            # MCC 647 entry is special, french... as usual !
            rec['mcc']          = '647'
            rec['mcc_url']      = L[-1]['mcc_url']
            rec['code_alpha_2'] = 'YT'
        #
        if not full:
            if not rec['country_name']:
                rec['country_name'] = L[-1]['country_name']
                rec['country_url']  = L[-1]['country_url']
            if not rec['code_alpha_2']:
                rec['code_alpha_2'] = L[-1]['code_alpha_2']
            if not rec['authority'] and L[-1]['authority']:
                rec['authority'] = L[-1]['authority']
            if not rec['notes'] and L[-1]['notes']:
                rec['notes']     = L[-1]['notes']
        #
        cc2.add(rec['code_alpha_2'])
        if rec['mcc'] in mcc:
            print('> duplicate entry for MCC %s: %s // %s'\
                  % (rec['mcc'], mcc[rec['mcc']]['country_name'], rec['country_name']))
        else:
            mcc[rec['mcc']] = rec
        L.append(rec)
    #
    L.sort(key=lambda r: (r['mcc'], r['code_alpha_2']))
    print('[+] parsed %i MCC entries (%i unique MCC) for %i ISO-3166 country codes'\
          % (len(L), len(mcc), len(cc2)))
    return L


def read_entry_mnc_title(e):
    name, url, sub, codes = '', '', None, []
    while e is not None:
        title = None
        for i in e:
            if i.values() and i.values()[0] == 'mw-headline' and '\n' not in ''.join(i.itertext()):
                title = i
                break
        if title is not None:
            break
        else:
            e = e.getprevious()
    #
    # country name
    if title is None:
        raise(Exception('unable to find headline title for MNC country name'))
    elif len(title) == 0:
        # raw title, without link
        name = title.values()[1].strip()
        return name, '', None, []
    else:
        name, url = _get_country_url(title[0])
        if len(title) > 1:
            # country sub-info, provided in parenthesis
            if len(e[1]) >= 2:
                sub = _get_country_url(e[1][1])
                if sub[0] == None:
                    sub = None            
            else:
                sub = None
    #
    # country alpha code
    # Warning, for EU MNC, separator is ' – ', whereas it is ' - ' for others
    title_txt = ''.join(title.itertext())
    title_sep = ' – ' if ' – ' in title_txt else ' - '
    codes = title_txt.split(title_sep, 1)[1].strip()
    if '/' in codes:
        codes = list(map(lambda s: s.strip().upper(), sorted(codes.split('/'))))
    elif '-' in codes:
        codes = list(map(lambda s: s.strip().upper(), sorted(codes.split('-'))))
    else:
        codes = [codes]
    for code in codes:
        if len(code) != 2 or not code.isalpha():
            raise(Exception('invalid title format'))
    #
    return name, url, sub, codes


def read_entry_mnc(T_MNC, off):
    L   = T_MNC[off]
    rec = dict(REC_MNC)
    rec['mcc']          = explore_text(L[0]).text.strip()
    rec['mnc']          = explore_text(L[1]).text.strip()
    rec['brand']        = ''.join(L[2].itertext()).strip()
    rec['operator']     = ''.join(L[3].itertext()).strip()
    rec['status']       = explore_text(L[4]).text.strip().lower()
    rec['bands']        = ''.join(L[5].itertext()).strip()
    rec['notes']        = _strip_wiki_refnote(_strip_wiki_ref(''.join(L[6].itertext())))
    #
    if len(rec['mcc']) > 3:
        # some HTML tab/ref in wikipedia may add the country name before the MCC
        rec['mcc'] = rec['mcc'][-3:]
    if rec['status'] != 'operational':
        rec['status'] = 'unknown'
    return rec


def parse_table_mnc(T_MNC):
    L       = []
    mcc     = set()
    mccmnc  = {}
    #
    # get table title with country name
    title = T_MNC[1].getparent().getparent().getprevious()
    country_infos = read_entry_mnc_title(title)
    #
    for i in range(1, len(T_MNC)):
        if len(T_MNC[i]) < 6:
            continue
        rec = read_entry_mnc(T_MNC, i)
        rec['country_name'], rec['country_url'], rec['country_sub'], rec['codes_alpha_2'] = \
            country_infos
        #
        if rec['mcc'].isdigit() and len(rec['mcc']) == 3 \
        and rec['mnc'].isdigit() and len(rec['mnc']) in (2, 3):
            mcc.add(rec['mcc'])
            L.append(rec)
        else:
            print('> invalid MCC MNC entry %s.%s, operator %s' % (rec['mcc'], rec['mnc'], rec['operator']))
    #
    print('[+] parsed %i MNC entries for MCC %s' % (len(L), ', '.join(sorted(mcc))))
    return L


def _insert_mnc(D, recs):
    mccmnc = set()
    for rec in recs:
        mcc0 = rec['mcc'][0]
        if mcc0 in D:
            D[mcc0].append(rec)
            mccmnc.add( rec['mcc'] + rec['mnc'] )
        else:
            raise(Exception('invalid MCC %s' % rec['mcc']))
    return mccmnc


def parse_table_mnc_all():
    mccmnc = set()
    D = {'0': [], '2': [], '3': [], '4': [], '5': [], '6': [], '7': [], '9': []}
    #
    # 1) MNC worldwide
    T_MCC = import_html_doc(URL_MCC).xpath('//table')
    # test networks
    mccmnc.update( _insert_mnc(D, parse_table_mnc(T_MCC[0][0])) )
    # intl networks
    mccmnc.update( _insert_mnc(D, parse_table_mnc(T_MCC[2][0])) )
    # UK ocean territory
    mccmnc.update( _insert_mnc(D, parse_table_mnc(T_MCC[3][0])) )
    #
    for url in (URL_MNC_EU, URL_MNC_NA, URL_MNC_AS, URL_MNC_OC, URL_MNC_AF, URL_MNC_SA):
        T_MNC = import_html_doc(url).xpath('//table')
        for i in range(0, len(T_MNC)):
            if not ''.join(T_MNC[i].itertext()).strip().startswith('MCC\nMNC'):
                continue
            try:
                mccmnc.update( _insert_mnc(D, parse_table_mnc(T_MNC[i][0])) )
            except Exception as err:
                print('> unable to extract MNC table from %s (index %i): %r' % (url, i, err))
                raise(err)
    #
    for L in D.values():
        L.sort(key=lambda r: (r['mcc'], r['mnc']))
    #
    print('[+] %i MCC MNC entries for %i unique MCC MNC'\
          % (sum(map(len, D.values())), len(mccmnc)))
    return D


#------------------------------------------------------------------------------#
# parsing Wikipedia international phone number prefixes
#------------------------------------------------------------------------------#

# International phone number prefixes
URL_MSISDN = "https://en.wikipedia.org/wiki/List_of_country_calling_codes"


def parse_table_msisdn_pref_over(T):
    # parse the table from the "overview" section, with {prefix: CC2 code, url prefix, url country}
    D = {}
    #
    # 2nd line: +1 prefix US + CA
    e        = T[2][1][0]
    pref     = e[0].text.strip()
    pref_url = URL_PREF + e[0].values()[0].strip()
    assert( pref == '1' and len(pref_url) > len(URL_PREF) )
    D['1'] = []
    for c in e[3:5]:
        cc2       = c.text.strip()
        _cucn     = c.values()
        cntr_url  = URL_PREF + _cucn[0].strip()
        cntr_name = _cucn[-1].strip()
        assert( len(cc2) == 2 and len(cntr_url) > len(URL_PREF) and cntr_name )
        D['1'].append( (cc2, cntr_name, cntr_url, pref_url) )
    D['1'].sort(key=lambda x:x[0])
    #
    # 2nd line: +1XYZ sub-prefixes North-America
    for e in T[3][2:]:
        if not len(e):
            continue
        e = e[0]
        for i in range(0, len(e)):
            if not e[i].text:
                if pref not in D:
                    # required for DO 1829 and 1849
                    D[pref]   = [(cc2, cntr_name, cntr_url, pref_url)]
            elif e[i].text.startswith('1'):
                # num prefix
                pref      = ''.join([c for c in e[i].text if c in '0123456789'])
                pref_url  = URL_PREF + e[i].values()[0].strip()
                assert( len(pref) >= 1 and len(pref_url) > len(URL_PREF) )
            elif e[i].text[0].isascii():
                # CC2
                cc2       = e[i].text.strip()
                _cucn     = e[i].values()
                cntr_url  = URL_PREF + _cucn[0].strip()
                cntr_name = _cucn[-1].strip()
                assert( len(cc2) == 2 and len(cntr_url) > len(URL_PREF) and cntr_name )
                assert( pref not in D )
                D[pref]   = [(cc2, cntr_name, cntr_url, pref_url)]
            else:
                assert()
    #
    # All the remaining lines
    # 7x Russian line should have a distinct layout
    for L in T[4:]:
        if not len(L):
            continue
        if (L[0].text is None or not L[0].text[0].isdigit()) and pref != '7':
            continue
        for e in L:
            if len(e) < 2 or not e[0].text:
                continue
            pref     = ''.join([c for c in e[0].text if c in '0123456789'])
            pref_url = URL_PREF + e[0].values()[0].strip()
            assert( len(pref) >= 1 and len(pref_url) > len(URL_PREF) )
            #
            vals = []
            for c in e[1:]:
                if c.text is None:
                    continue
                cc2 = c.text.strip()
                if len(cc2) != 2 or not cc2.isascii():
                    continue
                _cucn     = c.values()
                cntr_url  = URL_PREF + _cucn[0].strip()
                cntr_name = _cucn[-1].strip()
                assert( len(cntr_url) > len(URL_PREF) and cntr_name )
                vals.append( (cc2, cntr_name, cntr_url, pref_url) )
            assert( pref not in D )
            vals.sort(key=lambda x:x[0])
            D[pref] = vals
    #
    return D

# from simple 1 to 3 digits string to extended prefixes e.g., 374 (47, 97)
RE_WIKI_MSISDN_PREF = re.compile('([1-9]{1}[0-9]{0,2})(?: \(([0-9]{1,}(?:, {0,1}[0-9]{1,}){0,})\)){0,1}')

def parse_table_msisdn_pref_alphaord(T):
    # parse the table from the "Alphabetical order" section, with {country: prefix)
    D = {}
    #
    for L in T[1:]:
        name, prefs, utc, dst = tuple(map(str.strip, ''.join(L.itertext()).split('\n\n')))
        m = RE_WIKI_MSISDN_PREF.match(prefs)
        assert(m)
        pref, pref_exts = m.groups()
        if pref_exts:
            D[name] = tuple([pref + ext for ext in sorted(map(str.strip, pref_exts.split(',')))])
        else:
            D[name] = (pref, )
    #
    return D


def parse_table_msisdn_pref_locnocount(T):
    # parse the 2 tables from the "Locations with no country code" section, with {location name: prefix, location url, country, country url}
    D = {}
    #
    for L in T[1:]:
        e    = explore_text(L[0])
        name = e.text.strip()
        url  = URL_PREF + e.values()[0].strip()
        pref = L[1].text.strip()
        cntr = explore_text(L[2]).text.strip()
        assert( name not in D and len(pref) and pref.isdigit() and cntr )
        D[name] = (pref, cntr, url)
    #
    return D


def parse_table_msisdn_pref():
    T   = import_html_doc(URL_MSISDN).xpath('//table')
    #
    # extract the dict of {MSISDN prefix: country infos} from the 1st table
    # extract the dict of {country: MSISDN prefix} from the 2nd table
    # extract the dict of {location with no country code: MSISDN prefix} from the 3rd table
    return parse_table_msisdn_pref_over(T[0][0]), \
           parse_table_msisdn_pref_alphaord(T[1][0]), \
           parse_table_msisdn_pref_locnocount(T[2][0])


#------------------------------------------------------------------------------#
# parsing Wikipedia country borders
#------------------------------------------------------------------------------#

# this is used for both Wikipedia and World Factbook
CRAPPY_BORDERS = {
    'India , including Dahagram-Angarpota'      : 'India',
    'Bangladesh , including Dahagram-Angarpota' : 'Bangladesh',
    }


# Countries and borders
URL_BORDERS = "https://en.wikipedia.org/wiki/List_of_countries_and_territories_by_land_borders"


REC_BORDERS = {
    'country_name'  : '', # str
    'country_url'   : '', # str
    'country_sub'   : [], # list of 2-tuple of str, included countries / territories (country_name, country_url)
    'border_len_km' : 0,  # int
    'border_len_mi' : 0,  # int
    'neigh_num'     : 0,  # int
    'neigh'         : [], # list of 2-tuple or str, neighbour countries (country_name, country_url)
    }


# list of countries for which Wikipédia count of borders does not correspond
# to the list of border countries
BORDER_ISSUE = {'Afghanistan', 'China', 'Georgia', 'India', 'Israel', 'Russia', }


_stripbordref = lambda s: re.sub('\s{1,}', ' ', re.sub(r'\(.*?\)|\[.*?\]', ' ', s).strip())

def _get_bord(e):
    b = set()
    for s in map(str.strip, ''.join(e.itertext()).split('\xa0')):
        if s and s[0].isupper():
            name = _stripbordref(s)
            if ':' in name:
                name = name.split(':')[0].strip()
            if name in CRAPPY_BORDERS:
                name = CRAPPY_BORDERS[name]
            b.add(name)
    return list(sorted(b))


def _get_int(s):
    m = re.match('[0-9]{1,}', s)
    if not m:
        return 0
    else:
        return int(m.group())


def _get_subcntr(ec):
    subc = []
    for e in ec:
        if 'href' in e.attrib:
            subc.append( (e.attrib['title'].strip(),  URL_PREF + e.attrib['href']) )
    return subc


def read_entry_borders(T, off):
    L   = T[off]
    rec = dict(REC_BORDERS)
    #
    #rec['dbg']           = L
    rec['country_name'], rec['country_url'] = _get_country_url(L[0])
    rec['border_len_km'] = int(explore_text(L[1]).text.strip().replace(',', '').split('.')[0])
    rec['border_len_mi'] = int(explore_text(L[2]).text.strip().replace(',', '').split('.')[0])
    rec['neigh_num']     = _get_int(explore_text(L[4]).text.strip())
    #
    if len(L[0]) >= 3 and len(L[0][2]) >= 2 and len(L[0][2][1]) >= 2:
        # get list of sub countries
        # listed after '→includes:'
        subc = _get_subcntr(L[0][2][1])
        if subc:
            subc.sort(key=lambda t: t[0])
            rec['country_sub'] = subc
    #
    if len(L) >= 6:
        # get list of neighbours
        
        rec['neigh'] = _get_bord(L[5])
        #
        if rec['country_name'] not in BORDER_ISSUE and rec['neigh_num'] != len(rec['neigh']):
            assert()
    #
    return rec


def parse_table_borders():
    T   = import_html_doc(URL_BORDERS).xpath('//table')
    T_B = T[0][0]
    L   = []
    cns = set()
    for i in range(2, len(T_B)):
        rec = read_entry_borders(T_B, i)
        if rec['country_name'] in cns:
            print('> duplicate borders entry for %s' % rec['country_name'])
        else:
            cns.add(rec['country_name'])
        L.append(rec)
    L.sort(key=lambda r: r['country_name'])
    return L


#------------------------------------------------------------------------------#
# Main
#------------------------------------------------------------------------------#

def get_wiki_infos():
    try:
        D_iso  = parse_table_iso3166()
    except Exception as err:
        print('parse_table_iso3166: unable to download and / or parse Wikipedia HTML tables ; exception: %r' % err)
        return None, None, None, None, None, None, None
    try:
        L_mcc  = parse_table_mcc()
    except Exception as err:
        print('parse_table_mcc: unable to download and / or parse Wikipedia HTML tables ; exception: %r' % err)
        return None, None, None, None, None, None, None
    try:
        D_mnc  = parse_table_mnc_all()
    except Exception as err:
        print('parse_table_mnc_all: unable to download and / or parse Wikipedia HTML tables ; exception: %r' % err)
        return None, None, None, None, None, None, None
    try:
        D_pref, D_count, D_terr = parse_table_msisdn_pref()
    except Exception as err:
        print('parse_table_msisdn_pref: unable to download and / or parse Wikipedia HTML tables ; exception: %r' % err)
        return None, None, None, None, None, None, None
    try:
        L_bord = parse_table_borders()
    except Exception as err:
        print('parse_table_borders: unable to download and / or parse Wikipedia HTML tables ; exception: %r' % err)
        return None, None, None, None, None, None, None
    return D_iso, L_mcc, D_mnc, D_pref, D_count, D_terr, L_bord


def generate_json(d, destfile, src, license):
    meta = {'source': src, 'license': license}
    with open(destfile, 'w', encoding='utf-8') as fd:
        json.dump([meta, d], fp=fd, sort_keys=True, indent=2)
        fd.write('\n')
    print('[+] %s file generated' % destfile)


def generate_python(d, destfile, src, license):
    pp = PrettyPrinter(indent=2, width=120)
    varname = destfile[:-3].split('/')[-1].upper()
    with open(destfile, 'w', encoding='utf-8') as fd:
        fd.write('# -*- coding: UTF-8 -*-\n')
        fd.write('# source: %s\n' % ',\n#         '.join(src))
        fd.write('# license: %s\n\n' % license)
        fd.write('%s = \\\n%s\n' % (varname, pp.pformat(d)))
    print('[+] %s file generated' % destfile)


URL_LICENSE = "https://en.wikipedia.org/wiki/Wikipedia:Text_of_Creative_Commons_Attribution-ShareAlike_3.0_Unported_License"

def main():
    parser = argparse.ArgumentParser(description=
        'dump Wikipedia ISO-3166 country codes, MCC, MNC, numbering prefix and country borders tables into JSON or Python file')
    parser.add_argument('-j', action='store_true', help='produce JSON files (with suffix .json)')
    parser.add_argument('-p', action='store_true', help='produce Python files (with suffix .py)')
    args = parser.parse_args()
    D_iso, L_mcc, D_mnc, D_pref, D_count, D_terr, L_bord = get_wiki_infos()
    if D_iso is None:
        return 1
    #
    if args.j:
        generate_json(D_iso, PATH_PRE + 'wikip_iso3166.json', [URL_CODE_ALPHA_2], URL_LICENSE)
        generate_json(L_mcc, PATH_PRE + 'wikip_mcc.json', [URL_MCC], URL_LICENSE)
        generate_json(D_mnc, PATH_PRE + 'wikip_mnc.json',
                      [URL_MNC_EU, URL_MNC_NA, URL_MNC_AS, URL_MNC_OC, URL_MNC_AF, URL_MNC_SA], URL_LICENSE)
        generate_json(D_pref, PATH_PRE + 'wikip_msisdn.json', [URL_MSISDN], URL_LICENSE)
        generate_json(D_count, PATH_PRE + 'wikip_country.json', [URL_MSISDN], URL_LICENSE)
        generate_json(D_terr, PATH_PRE + 'wikip_territory.json', [URL_MSISDN], URL_LICENSE)
        generate_json(L_bord, PATH_PRE + 'wikip_borders.json', [URL_BORDERS], URL_LICENSE)
    if args.p:
        generate_python(D_iso, PATH_PRE + 'wikip_iso3166.py', [URL_CODE_ALPHA_2], URL_LICENSE)
        generate_python(L_mcc, PATH_PRE + 'wikip_mcc.py', [URL_MCC], URL_LICENSE)
        generate_python(D_mnc, PATH_PRE + 'wikip_mnc.py',
                        [URL_MNC_EU, URL_MNC_NA, URL_MNC_AS, URL_MNC_OC, URL_MNC_AF, URL_MNC_SA], URL_LICENSE)
        generate_python(D_pref, PATH_PRE + 'wikip_msisdn.py', [URL_MSISDN], URL_LICENSE)
        generate_python(D_count, PATH_PRE + 'wikip_country.py', [URL_MSISDN], URL_LICENSE)
        generate_python(D_terr, PATH_PRE + 'wikip_territory.py', [URL_MSISDN], URL_LICENSE)
        generate_python(L_bord, PATH_PRE + 'wikip_borders.py', [URL_BORDERS], URL_LICENSE)
    return 0


if __name__ == '__main__':
    sys.exit(main())