From ea7bdcaa03666486f6752a5622873a1ae37c48d8 Mon Sep 17 00:00:00 2001
From: Florian Mounier <florian.mounier@kozea.fr>
Date: Tue, 5 May 2015 11:32:23 +0200
Subject: [PATCH] Add continents

---
 demo/moulinrouge/tests.py | 14 +++++----
 pygal/graph/worldmap.py   |  1 +
 pygal/i18n.py             | 60 ++++++++++++++++++++++++++++++++++++---
 3 files changed, 65 insertions(+), 10 deletions(-)

diff --git a/demo/moulinrouge/tests.py b/demo/moulinrouge/tests.py
index 88ced63..46b33eb 100644
--- a/demo/moulinrouge/tests.py
+++ b/demo/moulinrouge/tests.py
@@ -430,12 +430,14 @@ def get_test_routes(app):
     def test_supranational():
         wmap = SupranationalWorldmap(style=choice(list(styles.values())))
 
-        # wmap.add('1st', [('nafta', 100), ('oecd', 10)])
-        wmap.add('2nd', [{
-            'value': ('europe', 10),
-            'label': 'EUROP3',
-            'xlink': 'http://google.com?q=tw'
-        }])
+        wmap.add('Asia', [('asia', 1)])
+        wmap.add('Europe', [('europe', 1)])
+        wmap.add('Africa', [('africa', 1)])
+        wmap.add('North america', [('north_america', 1)])
+        wmap.add('South america', [('south_america', 1)])
+        wmap.add('Oceania', [('oceania', 1)])
+        wmap.add('Antartica', [('antartica', 1)])
+
         wmap.title = 'Supra World Map !!'
         return wmap.render_response()
 
diff --git a/pygal/graph/worldmap.py b/pygal/graph/worldmap.py
index 5e5314b..a98a2f7 100644
--- a/pygal/graph/worldmap.py
+++ b/pygal/graph/worldmap.py
@@ -60,6 +60,7 @@ class Worldmap(BaseMap):
 
 class SupranationalWorldmap(Worldmap):
     """SupranationalWorldmap graph"""
+    x_labels = list(SUPRANATIONAL.keys())
 
     def enumerate_values(self, serie):
         """Replaces the values if it contains a supranational code."""
diff --git a/pygal/i18n.py b/pygal/i18n.py
index 181ab48..4618767 100644
--- a/pygal/i18n.py
+++ b/pygal/i18n.py
@@ -185,9 +185,6 @@ COUNTRIES = {
     'zw': 'Zimbabwe'
 }
 
-EUROPE = ['at', 'be', 'bg', 'hr', 'cy', 'cz', 'dk', 'ee', 'fi', 'fr', 'de',
-          'gr', 'hu', 'ie', 'it', 'lv', 'lt', 'lu', 'mt', 'nl', 'pl', 'pt',
-          'ro', 'sk', 'si', 'es', 'se', 'gb']
 
 EUR = ['be', 'de', 'ie', 'gr', 'es', 'fr', 'it', 'cy', 'lu', 'mt', 'nl', 'at',
        'pt', 'si', 'sk', 'fi', 'ee']
@@ -198,8 +195,63 @@ OECD = ['au', 'at', 'be', 'ca', 'cl', 'cz', 'dk', 'ee', 'fi', 'fr', 'de', 'gr',
 
 NAFTA = ['ca', 'mx', 'us']
 
+# Continents
 
-SUPRANATIONAL = {'europe': EUROPE, 'oecd': OECD, 'nafta': NAFTA, 'eur': EUR}
+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):