Synchronise sensitive areas

GeotrekCE · Oct 18, 2017 · 16a6680 · 16a6680
1 parent d571d4c
commit 16a6680
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/geotrek/trekking/management/commands/sync_rando.py b/geotrek/trekking/management/commands/sync_rando.py
@@ -28,6 +28,8 @@
 from geotrek.flatpages.views import FlatPageViewSet
 from geotrek.tourism import models as tourism_models
 from geotrek.tourism import views as tourism_views
+if 'geotrek.sensitivity' in settings.INSTALLED_APPS:
+    from geotrek.sensitivity import views as sensitivity_views
 from geotrek.trekking import models as trekking_models
 from geotrek.trekking.views import (TrekViewSet, POIViewSet, TrekPOIViewSet,
                                     TrekGPXDetail, TrekKMLDetail, TrekServiceViewSet,
@@ -356,6 +358,9 @@ def sync_trek(self, lang, trek):
         if self.categories:
             self.sync_trek_touristiccontents(lang, trek, zipfile=self.zipfile)
 
+        if 'geotrek.sensitivity' in settings.INSTALLED_APPS:
+            self.sync_trek_sensitiveareas(lang, trek, zipfile=self.zipfile)
+
         if self.verbosity == 2:
             self.stdout.write(u"\x1b[36m{lang}\x1b[0m \x1b[1m{name}\x1b[0m ...".format(lang=lang, name=zipname),
                               ending="")
@@ -430,6 +435,9 @@ def sync_trekking(self, lang):
 
         self.sync_tourism(lang)
 
+        if 'geotrek.sensitivity' in settings.INSTALLED_APPS:
+            self.sync_sensitiveareas(lang)
+
         if self.verbosity == 2:
             self.stdout.write(u"\x1b[36m{lang}\x1b[0m \x1b[1m{name}\x1b[0m ...".format(lang=lang, name=zipname), ending="")
 
@@ -509,6 +517,16 @@ def sync_event(self, lang, event):
         for picture, resized in event.resized_pictures:
             self.sync_media_file(lang, resized)
 
+    def sync_sensitiveareas(self, lang):
+        self.sync_geojson(lang, sensitivity_views.SensitiveAreaViewSet, 'sensitiveareas.geojson')
+
+    def sync_trek_sensitiveareas(self, lang, trek, zipfile=None):
+        params = {'format': 'geojson'}
+
+        view = sensitivity_views.TrekSensitiveAreaViewSet.as_view({'get': 'list'})
+        name = os.path.join('api', lang, 'treks', str(trek.pk), 'sensitiveareas.geojson')
+        self.sync_view(lang, view, name, params=params, zipfile=zipfile, pk=trek.pk)
+
     def sync_tourism(self, lang):
         self.sync_geojson(lang, tourism_views.TouristicContentViewSet, 'touristiccontents.geojson')
         self.sync_geojson(lang, tourism_views.TouristicEventViewSet, 'touristicevents.geojson',