Add button to country to fetch geoshape from wikidata.

everypolitician · Jan 25, 2019 · ed1c655 · ed1c655
1 parent 46d3fdc
commit ed1c655
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/commons_api/wikidata/templates/wikidata/country_detail.html b/commons_api/wikidata/templates/wikidata/country_detail.html
@@ -10,6 +10,7 @@
       {% if country.iso_3166_1_code %}
         <input type="submit" name="update-boundaries" value="Update boundaries">
       {% endif %}
+      <input type="submit" name="get-geoshape" value="Get boundary from wikidata">
     </form>
     </div>
 {% endblock %}

diff --git a/commons_api/wikidata/views.py b/commons_api/wikidata/views.py
@@ -50,6 +50,14 @@ def post(self, request, *args, **kwargs):
             messages.info(request, "Boundaries for {} will "
                                    "be refreshed".format(self.object))
 
+        if 'get-geoshape' in request.POST:
+            from commons_api.proto_commons.tasks import import_geoshape
+            from django.contrib.contenttypes.models import ContentType
+            ct = ContentType.objects.get_for_model(self.model)
+            import_geoshape.delay(ct.app_label, ct.model, self.object.id)
+            messages.info(request, "Getting boundary from wikidata "
+                                   "for {}".format(self.object))
+
         return redirect(self.object.get_absolute_url())