PR #353 for issue #352 - adding WWW-LINK Resource error and title mis…

…sing * #352 fixes adding WWW-LINK Resource error and title missing * #352 fixes adding WWW-LINK Resource error and title missing - flake8
geopython · Apr 26, 2021 · be792d8 · be792d8
1 parent b229528
commit be792d8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/.dockerignore b/.dockerignore
@@ -1,2 +1 @@
 .git
-
diff --git a/GeoHealthCheck/healthcheck.py b/GeoHealthCheck/healthcheck.py
@@ -183,15 +183,16 @@ def sniff_test_resource(config, resource_type, url):
             raise ValueError(message)
 
         if resource_type == 'WWW:LINK':
-            content_type = ows.info().getheader('Content-Type')
+            content_type = ows.info().get('Content-Type')
 
-            # Check content if the response is not an image
+            # When response is not an image try parse out Title and
+            # any Exceptions
             if 'image/' not in content_type:
                 content = ows.read()
                 import re
                 try:
-                    title_re = re.compile("<title>(.+?)</title>")
-                    title = title_re.search(content).group(1)
+                    title_re = re.compile('<title>(.+?)</title>'.encode())
+                    title = title_re.search(content).group(1).decode()
                 except Exception:
                     title = url
 
@@ -200,8 +201,9 @@ def sniff_test_resource(config, resource_type, url):
                     exception_text = None
                     try:
                         except_re = re.compile(
-                            "ServiceException>|ExceptionReport>")
-                        exception_text = except_re.search(content).group(0)
+                            'ServiceException>|ExceptionReport>'.encode())
+                        exception_text = except_re.search(content).\
+                            group(0).decode()
                     except Exception:
                         # No Exception in Response text
                         pass