Make SSLContext a singleton

samsungtvws/async_connection.py

@@ -37,6 +37,7 @@
     MS_CHANNEL_CONNECT_EVENT,
     MS_CHANNEL_UNAUTHORIZED,
 )
+from .helper import get_ssl_context
 
 _LOGGING = logging.getLogger(__name__)
 
@@ -66,9 +67,7 @@ async def open(self) -> WebSocketClientProtocol:
         _LOGGING.debug("WS url %s", url)
         connect_kwargs: Dict[str, Any] = {}
         if self._is_ssl_connection():
-            ssl_context = ssl.SSLContext()
-            ssl_context.verify_mode = ssl.CERT_NONE
-            connect_kwargs["ssl"] = ssl_context
+            connect_kwargs["ssl"] = get_ssl_context()
         connection = await connect(url, open_timeout=self.timeout, **connect_kwargs)
 
         event: Optional[str] = None

samsungtvws/helper.py

@@ -1,11 +1,13 @@
 import base64
 import json
 import logging
-from typing import Any, Dict, Union
+import ssl
+from typing import Any, Dict, Optional, Union
 
 from . import exceptions
 
 _LOGGING = logging.getLogger(__name__)
+_SSL_CONTEXT: Optional[ssl.SSLContext] = None
 
 
 def serialize_string(string: Union[str, bytes]) -> str:
@@ -23,3 +25,11 @@ def process_api_response(response: Union[str, bytes]) -> Dict[str, Any]:
         raise exceptions.ResponseError(
             "Failed to parse response from TV. Maybe feature not supported on this model"
         ) from err
+
+
+def get_ssl_context() -> ssl.SSLContext:
+    global _SSL_CONTEXT
+    if not _SSL_CONTEXT:
+        _SSL_CONTEXT = ssl.SSLContext()
+        _SSL_CONTEXT.verify_mode = ssl.CERT_NONE
+    return _SSL_CONTEXT