Wmts Probe

GeoHealthCheck/config_main.py

@@ -104,6 +104,7 @@
     # Probes
     'GeoHealthCheck.plugins.probe.owsgetcaps',
     'GeoHealthCheck.plugins.probe.wms',
+    'GeoHealthCheck.plugins.probe.wmts',
     'GeoHealthCheck.plugins.probe.wfs',
     'GeoHealthCheck.plugins.probe.tms',
     'GeoHealthCheck.plugins.probe.http',

GeoHealthCheck/plugins/probe/owsgetcaps.py

@@ -158,6 +158,29 @@ def __init__(self):
     })
     """Param defs"""
 
+    # This is to catch errors because url is only accessible through REST
+    # owslib will always do KVP request and this way the GetCap url will be:
+    # .../1.0.0/WMTSCapabilities.xml?service=WMTS&version=1.0.0&
+    # request=GetCapabilities
+    # This new request will return a valid WebMapTileService object.
+    def before_request(self):
+        self.original_url = self._resource.url
+
+        try:
+            response = Probe.perform_get_request(self, self._resource.url)
+        except Exception:
+            self._resource.url = self._resource.url + \
+                                 '/1.0.0/WMTSCapabilities.xml'
+            return
+
+        if (response.status_code != 200 and
+                '<ServiceException' in response.text):
+            self._resource.url = self._resource.url + \
+                                 '/1.0.0/WMTSCapabilities.xml'
+
+    def after_request(self):
+        self._resource.url = self.original_url
+
 
 class WpsGetCaps(OwsGetCaps):
     """WPS GetCapabilities Probe"""