From a5e41aae50587c330b7750c918ec1530ff36c197 Mon Sep 17 00:00:00 2001
From: Raphael Michel <mail@raphaelmichel.de>
Date: Wed, 9 Dec 2020 17:22:28 +0100
Subject: [PATCH] Add MapQuest as additional geocoding provider

---
 src/pretix/base/settings.py                   |  4 ++
 src/pretix/control/forms/global_settings.py   |  4 ++
 .../pretixcontrol/global_settings.html        |  2 +-
 src/pretix/control/views/geo.py               | 61 +++++++++++++++----
 src/pretix/locale/de/wordlist.txt             |  1 +
 src/pretix/locale/de_Informal/wordlist.txt    |  1 +
 src/pretix/locale/wordlist.txt                |  1 +
 7 files changed, 60 insertions(+), 14 deletions(-)

diff --git a/src/pretix/base/settings.py b/src/pretix/base/settings.py
index 54a79a547..fb5224923 100644
--- a/src/pretix/base/settings.py
+++ b/src/pretix/base/settings.py
@@ -1926,6 +1926,10 @@ Your {event} team"""))
         'default': None,
         'type': str
     },
+    'mapquest_apikey': {
+        'default': None,
+        'type': str
+    },
     'leaflet_tiles': {
         'default': None,
         'type': str
diff --git a/src/pretix/control/forms/global_settings.py b/src/pretix/control/forms/global_settings.py
index 142dce17c..4dfdc684d 100644
--- a/src/pretix/control/forms/global_settings.py
+++ b/src/pretix/control/forms/global_settings.py
@@ -41,6 +41,10 @@ class GlobalSettingsForm(SettingsForm):
                 required=False,
                 label=_("OpenCage API key for geocoding"),
             )),
+            ('mapquest_apikey', SecretKeySettingsField(
+                required=False,
+                label=_("MapQuest API key for geocoding"),
+            )),
             ('leaflet_tiles', forms.CharField(
                 required=False,
                 label=_("Leaflet tiles URL pattern"),
diff --git a/src/pretix/control/templates/pretixcontrol/global_settings.html b/src/pretix/control/templates/pretixcontrol/global_settings.html
index 59961d8bf..c3b988d10 100644
--- a/src/pretix/control/templates/pretixcontrol/global_settings.html
+++ b/src/pretix/control/templates/pretixcontrol/global_settings.html
@@ -6,7 +6,7 @@
     <form action="" method="post" class="form-horizontal" enctype="multipart/form-data">
         {% csrf_token %}
         {% bootstrap_form_errors form %}
-        {% bootstrap_form form layout='horizontal' %}
+        {% bootstrap_form form layout='control' %}
         <div class="form-group submit-group">
             <button type="submit" class="btn btn-primary btn-save">
                 {% trans "Save" %}
diff --git a/src/pretix/control/views/geo.py b/src/pretix/control/views/geo.py
index 419716af8..f3bb201a2 100644
--- a/src/pretix/control/views/geo.py
+++ b/src/pretix/control/views/geo.py
@@ -14,14 +14,7 @@ logger = logging.getLogger(__name__)
 
 class GeoCodeView(LoginRequiredMixin, View):
     def get(self, request, *args, **kwargs):
-        gs = GlobalSettingsObject()
-        if not gs.settings.opencagedata_apikey:
-            return JsonResponse({
-                'success': False,
-                'results': []
-            }, status=200)
-
-        q = request.GET.get('q')
+        q = self.request.GET.get('q')
         cd = cache.get('geocode:{}'.format(q))
         if cd:
             return JsonResponse({
@@ -29,6 +22,26 @@ class GeoCodeView(LoginRequiredMixin, View):
                 'results': cd
             }, status=200)
 
+        gs = GlobalSettingsObject()
+        if gs.settings.opencagedata_apikey:
+            res = self._use_opencage(q)
+        if gs.settings.mapquest_apikey:
+            res = self._use_mapquest(q)
+        else:
+            return JsonResponse({
+                'success': False,
+                'results': []
+            }, status=200)
+
+        cache.set('geocode:{}'.format(q), res, timeout=3600 * 6)
+        return JsonResponse({
+            'success': True,
+            'results': res
+        }, status=200)
+
+    def _use_opencage(self, q):
+        gs = GlobalSettingsObject()
+
         try:
             r = requests.get(
                 'https://api.opencagedata.com/geocode/v1/json?q={}&key={}'.format(
@@ -51,9 +64,31 @@ class GeoCodeView(LoginRequiredMixin, View):
                 'lon': r['geometry']['lng'],
             } for r in d['results']
         ]
-        cache.set('geocode:{}'.format(q), res, timeout=3600 * 6)
+        return res
 
-        return JsonResponse({
-            'success': True,
-            'results': res
-        }, status=200)
+    def _use_mapquest(self, q):
+        gs = GlobalSettingsObject()
+
+        try:
+            r = requests.get(
+                'http://www.mapquestapi.com/geocoding/v1/address?location={}&key={}'.format(
+                    quote(q), gs.settings.mapquest_apikey
+                )
+            )
+            r.raise_for_status()
+        except IOError:
+            logger.exception("Geocoding failed")
+            return JsonResponse({
+                'success': False,
+                'results': []
+            }, status=200)
+        else:
+            d = r.json()
+        res = [
+            {
+                'formatted': q,
+                'lat': r['locations'][0]['latLng']['lat'],
+                'lon': r['locations'][0]['latLng']['lng'],
+            } for r in d['results']
+        ]
+        return res
diff --git a/src/pretix/locale/de/wordlist.txt b/src/pretix/locale/de/wordlist.txt
index 3fb0ee4bd..2a34664e3 100644
--- a/src/pretix/locale/de/wordlist.txt
+++ b/src/pretix/locale/de/wordlist.txt
@@ -128,6 +128,7 @@ Leaflet
 loszulegen
 Ltd
 max
+MapQuest
 Merchandise
 Meta
 Metadaten
diff --git a/src/pretix/locale/de_Informal/wordlist.txt b/src/pretix/locale/de_Informal/wordlist.txt
index 844fda26e..762453129 100644
--- a/src/pretix/locale/de_Informal/wordlist.txt
+++ b/src/pretix/locale/de_Informal/wordlist.txt
@@ -128,6 +128,7 @@ Leaflet
 loszulegen
 Ltd
 max
+MapQuest
 Merchandise
 Meta
 Metadaten
diff --git a/src/pretix/locale/wordlist.txt b/src/pretix/locale/wordlist.txt
index 23503983f..064531117 100644
--- a/src/pretix/locale/wordlist.txt
+++ b/src/pretix/locale/wordlist.txt
@@ -48,6 +48,7 @@ Kosovo
 Lead
 Leaflet
 LLC
+MapQuest
 MOTO
 Multibanco
 multiline