default to env vars for session init

Author: Graeme22

tastytrade/session.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import json
+import os
 import time
 from contextlib import asynccontextmanager
 from datetime import date, datetime
@@ -251,8 +252,10 @@ class Session(AsyncContextManagerMixin):
 
     Prefer using with its async context manager to ensure proper cleanup.
 
-    :param provider_secret: OAuth secret for your provider
-    :param refresh_token: refresh token for the user
+    :param provider_secret:
+        OAuth secret for your provider; defaults to the $TT_SECRET environment variable
+    :param refresh_token:
+        refresh token for the user; defaults to the $TT_REFRESH environment variable
     :param is_test:
         whether to use the test API endpoints, default False
     :param proxy:
@@ -265,8 +268,8 @@ class Session(AsyncContextManagerMixin):
 
     def __init__(
         self,
-        provider_secret: str,
-        refresh_token: str,
+        provider_secret: str | None = None,
+        refresh_token: str | None = None,
         is_test: bool = False,
         proxy: str | None = None,
         **client_kwargs: Any,
@@ -276,9 +279,9 @@ def __init__(
         #: Proxy URL to use for requests and web sockets
         self.proxy = proxy
         #: OAuth secret for your provider
-        self.provider_secret = provider_secret
+        self.provider_secret = provider_secret or os.environ["TT_SECRET"]
         #: Refresh token for the user
-        self.refresh_token = refresh_token
+        self.refresh_token = refresh_token or os.environ["TT_REFRESH"]
         # The headers to use for API requests
         headers = {"Accept": "application/json", "Content-Type": "application/json"}
         if not is_test:  # not accepted in sandbox

tests/test_session.py

@@ -1,5 +1,4 @@
 import json
-import os
 
 import pytest
 from proxy import TestCase
@@ -9,16 +8,11 @@
 pytestmark = pytest.mark.anyio
 
 
-@pytest.fixture(scope="module")
-def session_with_kwargs(anyio_backend: str) -> Session:
-    return Session(os.environ["TT_SECRET"], os.environ["TT_REFRESH"], timeout=10.0)
-
-
 async def test_get_customer(session: Session):
     await session.get_customer()
 
 
-def test_serialize_deserialize(session: Session, session_with_kwargs: Session):
+def test_serialize_deserialize(session: Session):
     # No client_kwargs
     data = session.serialize()
     obj = Session.deserialize(data)
@@ -30,6 +24,7 @@ def test_serialize_deserialize(session: Session, session_with_kwargs: Session):
     legacy_obj = Session.deserialize(json.dumps(legacy_payload))
     assert legacy_obj.client_kwargs == {}
     # With timeout client_kwargs
+    session_with_kwargs = Session(timeout=10.0)
     data_with_kwargs = session_with_kwargs.serialize()
     obj_with_kwargs = Session.deserialize(data_with_kwargs)
     assert set(obj_with_kwargs.__dict__.keys()) == set(