User agent version (#161)

* try dynamic version

* add version to user agent string

* add types for setuptools

Author: clickingbuttons

poetry.lock

@@ -5,6 +5,9 @@
 from enum import Enum
 from typing import Optional, Any, Dict
 from datetime import datetime
+import pkg_resources  # part of setuptools
+
+version = pkg_resources.require("polygon-api-client")[0].version
 
 
 class BaseClient:
@@ -31,7 +34,7 @@ def __init__(
             num_pools=num_pools,
             headers={
                 "Authorization": "Bearer " + self.API_KEY,
-                "User-Agent": "Python client",
+                "User-Agent": "Python client " + version,
             },
             ca_certs=certifi.where(),
             cert_reqs="CERT_REQUIRED",

polygon/rest/base.py

@@ -77,8 +77,11 @@ async def connect(
         if self.verbose:
             print("connect:", self.url)
         # darwin needs some extra <3
-        ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
-        ssl_context.load_verify_locations(certifi.where())
+        ssl_context = None
+        if self.url.startswith("wss://"):
+            ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
+            ssl_context.load_verify_locations(certifi.where())
+
         async for s in connect(
             self.url, close_timeout=close_timeout, ssl=ssl_context, **kwargs
         ):

polygon/websocket/__init__.py

@@ -39,6 +39,7 @@ sphinx-rtd-theme = "^1.0.0"
 # keep this in sync with docs/requirements.txt for readthedocs.org
 sphinx-autodoc-typehints = "^1.18.1"
 types-certifi = "^2021.10.8"
+types-setuptools = "^57.4.14"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]