import json
import pygal_maps_world.maps
from pygal.style import LightColorizedStyle ,RotateStyle
from country_codes import get_country_code

#将数据加载到一个列表中
filename = 'population_data.json'
with open(filename) as f:
    pop_data = json.load(f)

#创建一个包含人口数量的字典
cc_populations ={}
for pop_dict in pop_data:
    if pop_dict['Year'] == '2010':
        country = pop_dict['Country Name']
        population = int(float(pop_dict['Value']))
        code = get_country_code(country)
        if code:
            cc_populations[code] = population

#根据人口数量 将所有的国家分成三组
cc_pops_1,cc_pops_2,cc_pops_3 = {},{},{}
for cc ,pop in cc_populations.items():
    if pop < 10000000:
        cc_pops_1[cc] = pop
    elif pop < 1000000000:
        cc_pops_2[cc] = pop
    else:
        cc_pops_3[cc] = pop

#看看每组分别包含多少国家
print(len(cc_pops_1),len(cc_pops_2),len(cc_pops_3))
#//light.. 使用时不能控制颜色 只能使用基色
#wm_style = LightColorizedStyle
#wm_style = RotateStyle('#336699')
#设置颜色 可使用Ran。。将light 作为基本样式
wm_style = RotateStyle('#336699',base_style=LightColorizedStyle)
wm = pygal_maps_world.maps.World(style=wm_style)

wm.title = 'World Population in 2010 , by Country'
wm.add('0-10m',cc_pops_1)
wm.add('10m-1bn',cc_pops_2)
wm.add('>1bn',cc_pops_3)

wm.render_to_file('F:/world_population3.svg')

"""
#打印每个国家2010年的人口数量
for pop_dict in pop_data:
    if pop_dict['Year'] == '2010':
        country_name = pop_dict['Country Name']
        population = int(float(pop_dict['Value']))
        code = get_country_code(country_name)
        if code :
            print(code+": "+str(population))
        else:
            print('ERROR-'+country_name)
"""
