mirror of https://github.com/Kozea/pygal.git
Florian Mounier
10 years ago
10 changed files with 12 additions and 2835 deletions
|
Before Width: | Height: | Size: 760 KiB |
@ -1,69 +0,0 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2014 Kozea |
||||
# |
||||
# This library is free software: you can redistribute it and/or modify it under |
||||
# the terms of the GNU Lesser General Public License as published by the Free |
||||
# Software Foundation, either version 3 of the License, or (at your option) any |
||||
# later version. |
||||
# |
||||
# This library 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 Lesser General Public License for more |
||||
# details. |
||||
# |
||||
# You should have received a copy of the GNU Lesser General Public License |
||||
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||
""" |
||||
Worldmap chart |
||||
|
||||
""" |
||||
|
||||
from __future__ import division |
||||
from pygal.util import cached_property |
||||
from pygal.graph.map import BaseMap |
||||
from pygal.i18n import COUNTRIES, SUPRANATIONAL |
||||
import os |
||||
|
||||
|
||||
with open(os.path.join( |
||||
os.path.dirname(__file__), 'maps', |
||||
'worldmap.svg')) as file: |
||||
WORLD_MAP = file.read() |
||||
|
||||
|
||||
class Worldmap(BaseMap): |
||||
"""Worldmap graph""" |
||||
x_labels = list(COUNTRIES.keys()) |
||||
area_names = COUNTRIES |
||||
area_prefix = '' |
||||
svg_map = WORLD_MAP |
||||
kind = 'country' |
||||
|
||||
@cached_property |
||||
def countries(self): |
||||
return [val[0] |
||||
for serie in self.all_series |
||||
for val in serie.values |
||||
if val[0] is not None] |
||||
|
||||
@cached_property |
||||
def _values(self): |
||||
"""Getter for series values (flattened)""" |
||||
return [val[1] |
||||
for serie in self.series |
||||
for val in serie.values |
||||
if val[1] is not None] |
||||
|
||||
|
||||
class SupranationalWorldmap(Worldmap): |
||||
"""SupranationalWorldmap graph""" |
||||
x_labels = list(SUPRANATIONAL.keys()) |
||||
|
||||
def enumerate_values(self, serie): |
||||
"""Replaces the values if it contains a supranational code.""" |
||||
for i, (code, value) in enumerate(serie.values): |
||||
for subcode in SUPRANATIONAL.get(code, []): |
||||
yield i, (subcode, value) |
||||
@ -1,260 +0,0 @@
|
||||
COUNTRIES = { |
||||
'ad': 'Andorra', |
||||
'ae': 'United Arab Emirates', |
||||
'af': 'Afghanistan', |
||||
'al': 'Albania', |
||||
'am': 'Armenia', |
||||
'ao': 'Angola', |
||||
'aq': 'Antarctica', |
||||
'ar': 'Argentina', |
||||
'at': 'Austria', |
||||
'au': 'Australia', |
||||
'az': 'Azerbaijan', |
||||
'ba': 'Bosnia and Herzegovina', |
||||
'bd': 'Bangladesh', |
||||
'be': 'Belgium', |
||||
'bf': 'Burkina Faso', |
||||
'bg': 'Bulgaria', |
||||
'bh': 'Bahrain', |
||||
'bi': 'Burundi', |
||||
'bj': 'Benin', |
||||
'bn': 'Brunei Darussalam', |
||||
'bo': 'Bolivia, Plurinational State of', |
||||
'br': 'Brazil', |
||||
'bt': 'Bhutan', |
||||
'bw': 'Botswana', |
||||
'by': 'Belarus', |
||||
'bz': 'Belize', |
||||
'ca': 'Canada', |
||||
'cd': 'Congo, the Democratic Republic of the', |
||||
'cf': 'Central African Republic', |
||||
'cg': 'Congo', |
||||
'ch': 'Switzerland', |
||||
'ci': "Cote d'Ivoire", |
||||
'cl': 'Chile', |
||||
'cm': 'Cameroon', |
||||
'cn': 'China', |
||||
'co': 'Colombia', |
||||
'cr': 'Costa Rica', |
||||
'cu': 'Cuba', |
||||
'cv': 'Cape Verde', |
||||
'cy': 'Cyprus', |
||||
'cz': 'Czech Republic', |
||||
'de': 'Germany', |
||||
'dj': 'Djibouti', |
||||
'dk': 'Denmark', |
||||
'do': 'Dominican Republic', |
||||
'dz': 'Algeria', |
||||
'ec': 'Ecuador', |
||||
'ee': 'Estonia', |
||||
'eg': 'Egypt', |
||||
'eh': 'Western Sahara', |
||||
'er': 'Eritrea', |
||||
'es': 'Spain', |
||||
'et': 'Ethiopia', |
||||
'fi': 'Finland', |
||||
'fr': 'France', |
||||
'ga': 'Gabon', |
||||
'gb': 'United Kingdom', |
||||
'ge': 'Georgia', |
||||
'gf': 'French Guiana', |
||||
'gh': 'Ghana', |
||||
'gl': 'Greenland', |
||||
'gm': 'Gambia', |
||||
'gn': 'Guinea', |
||||
'gq': 'Equatorial Guinea', |
||||
'gr': 'Greece', |
||||
'gt': 'Guatemala', |
||||
'gu': 'Guam', |
||||
'gw': 'Guinea-Bissau', |
||||
'gy': 'Guyana', |
||||
'hk': 'Hong Kong', |
||||
'hn': 'Honduras', |
||||
'hr': 'Croatia', |
||||
'ht': 'Haiti', |
||||
'hu': 'Hungary', |
||||
'id': 'Indonesia', |
||||
'ie': 'Ireland', |
||||
'il': 'Israel', |
||||
'in': 'India', |
||||
'iq': 'Iraq', |
||||
'ir': 'Iran, Islamic Republic of', |
||||
'is': 'Iceland', |
||||
'it': 'Italy', |
||||
'jm': 'Jamaica', |
||||
'jo': 'Jordan', |
||||
'jp': 'Japan', |
||||
'ke': 'Kenya', |
||||
'kg': 'Kyrgyzstan', |
||||
'kh': 'Cambodia', |
||||
'kp': "Korea, Democratic People's Republic of", |
||||
'kr': 'Korea, Republic of', |
||||
'kw': 'Kuwait', |
||||
'kz': 'Kazakhstan', |
||||
'la': "Lao People's Democratic Republic", |
||||
'lb': 'Lebanon', |
||||
'li': 'Liechtenstein', |
||||
'lk': 'Sri Lanka', |
||||
'lr': 'Liberia', |
||||
'ls': 'Lesotho', |
||||
'lt': 'Lithuania', |
||||
'lu': 'Luxembourg', |
||||
'lv': 'Latvia', |
||||
'ly': 'Libyan Arab Jamahiriya', |
||||
'ma': 'Morocco', |
||||
'mc': 'Monaco', |
||||
'md': 'Moldova, Republic of', |
||||
'me': 'Montenegro', |
||||
'mg': 'Madagascar', |
||||
'mk': 'Macedonia, the former Yugoslav Republic of', |
||||
'ml': 'Mali', |
||||
'mm': 'Myanmar', |
||||
'mn': 'Mongolia', |
||||
'mo': 'Macao', |
||||
'mr': 'Mauritania', |
||||
'mt': 'Malta', |
||||
'mu': 'Mauritius', |
||||
'mv': 'Maldives', |
||||
'mw': 'Malawi', |
||||
'mx': 'Mexico', |
||||
'my': 'Malaysia', |
||||
'mz': 'Mozambique', |
||||
'na': 'Namibia', |
||||
'ne': 'Niger', |
||||
'ng': 'Nigeria', |
||||
'ni': 'Nicaragua', |
||||
'nl': 'Netherlands', |
||||
'no': 'Norway', |
||||
'np': 'Nepal', |
||||
'nz': 'New Zealand', |
||||
'om': 'Oman', |
||||
'pa': 'Panama', |
||||
'pe': 'Peru', |
||||
'pg': 'Papua New Guinea', |
||||
'ph': 'Philippines', |
||||
'pk': 'Pakistan', |
||||
'pl': 'Poland', |
||||
'pr': 'Puerto Rico', |
||||
'ps': 'Palestine, State of', |
||||
'pt': 'Portugal', |
||||
'py': 'Paraguay', |
||||
're': 'Reunion', |
||||
'ro': 'Romania', |
||||
'rs': 'Serbia', |
||||
'ru': 'Russian Federation', |
||||
'rw': 'Rwanda', |
||||
'sa': 'Saudi Arabia', |
||||
'sc': 'Seychelles', |
||||
'sd': 'Sudan', |
||||
'se': 'Sweden', |
||||
'sg': 'Singapore', |
||||
'sh': 'Saint Helena, Ascension and Tristan da Cunha', |
||||
'si': 'Slovenia', |
||||
'sk': 'Slovakia', |
||||
'sl': 'Sierra Leone', |
||||
'sm': 'San Marino', |
||||
'sn': 'Senegal', |
||||
'so': 'Somalia', |
||||
'sr': 'Suriname', |
||||
'st': 'Sao Tome and Principe', |
||||
'sv': 'El Salvador', |
||||
'sy': 'Syrian Arab Republic', |
||||
'sz': 'Swaziland', |
||||
'td': 'Chad', |
||||
'tg': 'Togo', |
||||
'th': 'Thailand', |
||||
'tj': 'Tajikistan', |
||||
'tl': 'Timor-Leste', |
||||
'tm': 'Turkmenistan', |
||||
'tn': 'Tunisia', |
||||
'tr': 'Turkey', |
||||
'tw': 'Taiwan, Province of China', |
||||
'tz': 'Tanzania, United Republic of', |
||||
'ua': 'Ukraine', |
||||
'ug': 'Uganda', |
||||
'us': 'United States', |
||||
'uy': 'Uruguay', |
||||
'uz': 'Uzbekistan', |
||||
'va': 'Holy See (Vatican City State)', |
||||
've': 'Venezuela, Bolivarian Republic of', |
||||
'vn': 'Viet Nam', |
||||
'ye': 'Yemen', |
||||
'yt': 'Mayotte', |
||||
'za': 'South Africa', |
||||
'zm': 'Zambia', |
||||
'zw': 'Zimbabwe' |
||||
} |
||||
|
||||
|
||||
EUR = ['be', 'de', 'ie', 'gr', 'es', 'fr', 'it', 'cy', 'lu', 'mt', 'nl', 'at', |
||||
'pt', 'si', 'sk', 'fi', 'ee'] |
||||
|
||||
OECD = ['au', 'at', 'be', 'ca', 'cl', 'cz', 'dk', 'ee', 'fi', 'fr', 'de', 'gr', |
||||
'hu', 'is', 'ie', 'il', 'it', 'jp', 'kr', 'lu', 'mx', 'nl', 'nz', 'no', |
||||
'pl', 'pt', 'sk', 'si', 'es', 'se', 'ch', 'tr', 'gb', 'us'] |
||||
|
||||
NAFTA = ['ca', 'mx', 'us'] |
||||
|
||||
# Continents |
||||
|
||||
ASIA = [ |
||||
'af', 'am', 'az', 'bh', 'bd', 'bt', 'bn', 'mn', 'mm', 'kh', 'cn', 'cy', |
||||
'tl', 'ge', 'in', 'id', 'ir', 'iq', 'il', 'jp', 'jo', 'kz', 'kw', 'kg', |
||||
'la', 'lb', 'my', 'mv', 'mn', 'np', 'kp', 'om', 'pk', 'ps', 'ph', 'ru', |
||||
'sa', 'sg', 'lk', 'kr', 'sy', 'tw', 'tj', 'th', 'tr', 'tm', 'ae', 'uz', |
||||
'vn', 'ye', 'mo', 'hk' |
||||
] |
||||
|
||||
EUROPE = [ |
||||
'al', 'am', 'at', 'az', 'be', 'by', 'ba', 'bg', 'hr', 'cy', 'cz', 'dk', |
||||
'ee', 'fi', 'fr', 'ge', 'de', 'gr', 'hu', 'is', 'ie', 'it', 'kz', 'lv', |
||||
'li', 'lt', 'lu', 'mk', 'mt', 'md', 'mc', 'me', 'nl', 'no', 'pl', 'pt', |
||||
'ro', 'ru', 'sm', 'rs', 'sk', 'si', 'es', 'se', 'ch', 'tr', 'ua', 'gb', |
||||
'va' |
||||
] |
||||
|
||||
AFRICA = [ |
||||
'dz', 'eg', 'ly', 'ma', 'sd', 'tn', 'eh', 'dj', 'er', 'et', 'so', 'bi', |
||||
'ke', 'mg', 'mw', 'mu', 'yt', 'mz', 're', 'rw', 'sc', 'tz', 'ug', 'zm', |
||||
'zw', 'ao', 'cm', 'cf', 'td', 'cd', 'cg', 'gq', 'ga', 'st', 'bw', 'ls', |
||||
'na', 'za', 'sz', 'bj', 'bf', 'cv', 'gm', 'gh', 'gn', 'gw', 'ci', 'lr', |
||||
'ml', 'mr', 'ne', 'ng', 'sh', 'sn', 'sl', 'tg' |
||||
] |
||||
|
||||
NORTH_AMERICA = [ |
||||
'ca', 'mx', 'us', 'gl', 'cu', 'do', 'ht', 'jm', 'pr', 'bz', 'cr', 'sv', |
||||
'gt', 'hn', 'ni', 'pa' |
||||
] |
||||
|
||||
SOUTH_AMERICA = [ |
||||
'ar', 'bo', 'br', 'cl', 'co', 'ec', 'gf', 'gy', 'py', 'pe', 'sr', 'uy', |
||||
've' |
||||
] |
||||
|
||||
OCEANIA = [ |
||||
'au', 'nz', 'pg', 'gu', |
||||
] |
||||
|
||||
ANTARTICA = [ |
||||
'aq' |
||||
] |
||||
|
||||
SUPRANATIONAL = { |
||||
'oecd': OECD, |
||||
'nafta': NAFTA, |
||||
'eur': EUR, |
||||
|
||||
'asia': ASIA, |
||||
'europe': EUROPE, |
||||
'africa': AFRICA, |
||||
'north_america': NORTH_AMERICA, |
||||
'south_america': SOUTH_AMERICA, |
||||
'oceania': OCEANIA, |
||||
'antartica': ANTARTICA |
||||
} |
||||
|
||||
|
||||
def set_countries(countries, clear=False): |
||||
if clear: |
||||
COUNTRIES.clear() |
||||
COUNTRIES.update(countries) |
||||
@ -1,87 +0,0 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2014 Kozea |
||||
# |
||||
# This library is free software: you can redistribute it and/or modify it under |
||||
# the terms of the GNU Lesser General Public License as published by the Free |
||||
# Software Foundation, either version 3 of the License, or (at your option) any |
||||
# later version. |
||||
# |
||||
# This library 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 Lesser General Public License for more |
||||
# details. |
||||
# |
||||
# You should have received a copy of the GNU Lesser General Public License |
||||
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||
|
||||
from pygal import ( |
||||
Worldmap, SupranationalWorldmap) |
||||
from pygal.i18n import COUNTRIES, SUPRANATIONAL, set_countries |
||||
import operator |
||||
try: |
||||
from functools import reduce |
||||
except ImportError: |
||||
pass |
||||
|
||||
_COUNTRIES = dict(COUNTRIES) |
||||
|
||||
|
||||
def test_worldmap(): |
||||
set_countries(_COUNTRIES, True) |
||||
datas = {} |
||||
for i, ctry in enumerate(COUNTRIES): |
||||
datas[ctry] = i |
||||
|
||||
wmap = Worldmap() |
||||
wmap.add('countries', datas) |
||||
q = wmap.render_pyquery() |
||||
assert len( |
||||
q('.country.color-0') |
||||
) == len(COUNTRIES) |
||||
assert 'France' in q('.country.fr').text() |
||||
|
||||
|
||||
def test_worldmap_i18n(): |
||||
set_countries(_COUNTRIES, True) |
||||
datas = {} |
||||
for i, ctry in enumerate(COUNTRIES): |
||||
datas[ctry] = i |
||||
|
||||
set_countries({'fr': 'Francia'}) |
||||
wmap = Worldmap() |
||||
wmap.add('countries', datas) |
||||
q = wmap.render_pyquery() |
||||
assert len( |
||||
q('.country.color-0') |
||||
) == len(COUNTRIES) |
||||
assert 'Francia' in q('.country.fr').text() |
||||
|
||||
|
||||
def test_worldmap_i18n_clear(): |
||||
set_countries(_COUNTRIES, True) |
||||
wmap = Worldmap() |
||||
wmap.add('countries', dict(fr=12)) |
||||
set_countries({'fr': 'Frankreich'}, clear=True) |
||||
q = wmap.render_pyquery() |
||||
assert len( |
||||
q('.country.color-0') |
||||
) == 1 |
||||
assert 'Frankreich' in q('.country.fr').text() |
||||
|
||||
|
||||
def test_supranationalworldmap(): |
||||
set_countries(_COUNTRIES, True) |
||||
datas = {} |
||||
for i, supra in enumerate(SUPRANATIONAL): |
||||
datas[supra] = i + 1 |
||||
|
||||
wmap = SupranationalWorldmap() |
||||
wmap.add('supra', datas) |
||||
q = wmap.render_pyquery() |
||||
assert len( |
||||
q('.country.color-0') |
||||
) == len( |
||||
reduce(operator.or_, map(set, SUPRANATIONAL.values()))) |
||||
Loading…
Reference in new issue