Merge pull request #3 from ilteriskeskin/error-management

Added error management

Author: ilteriskeskin

main.py

@@ -1,9 +1,13 @@
 import datetime
+import logging
 import json
 
 from urllib.request import ssl, socket
 
 
+logger = logging.getLogger(__name__)
+
+
 def check_ssl(hostname: str, port: str) -> dict:
     """
     This function is check ssl certificate
@@ -13,17 +17,27 @@ def check_ssl(hostname: str, port: str) -> dict:
     """
     context = ssl.create_default_context()
 
-    with socket.create_connection((hostname, port)) as sock:
-        with context.wrap_socket(sock, server_hostname=hostname) as ssock:
-            certificate = ssock.getpeercert()
-
-    certExpires = datetime.datetime.strptime(
-        certificate['notAfter'], '%b %d %H:%M:%S %Y %Z')
-
-    return {
-        'Expiration Day': f"{certExpires.year}-{certExpires.month}-{certExpires.day}",
-        'Remaining Day': (certExpires - datetime.datetime.now()).days
-    }
+    try:
+        with socket.create_connection(address=(hostname, port), timeout=3) as sock:
+            with context.wrap_socket(sock, server_hostname=hostname) as ssock:
+                certificate = ssock.getpeercert()
+
+        certExpires = datetime.datetime.strptime(
+            certificate['notAfter'], '%b %d %H:%M:%S %Y %Z')
+
+        data = {
+            'status': 200,
+            'expiration_day': f"{certExpires.year}-{certExpires.month}-{certExpires.day}",
+            'remaining_day': (certExpires - datetime.datetime.now()).days
+        }
+    except Exception as e:
+        logger.error(e)
+        data = {
+            'status': 500,
+            'message': 'Server Error! Try again or contact admin.'
+        }
+
+    return data
 
 
 def run(event, context) -> list: