sweatstack 0.13.1__tar.gz → 0.14.0__tar.gz

{sweatstack-0.13.1 → sweatstack-0.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sweatstack
-Version: 0.13.1
+Version: 0.14.0
 Summary: The official Python client for SweatStack
 Author-email: Aart Goossens <aart@gssns.io>
 Requires-Python: >=3.12
@@ -13,7 +13,6 @@ Requires-Dist: ipython>=8.31.0; extra == 'jupyterlab'
 Requires-Dist: jupyterlab>=4.3.4; extra == 'jupyterlab'
 Requires-Dist: matplotlib; extra == 'jupyterlab'
 Provides-Extra: streamlit
-Requires-Dist: streamlit-cookies-controller>=0.0.4; extra == 'streamlit'
 Requires-Dist: streamlit>=1.42.0; extra == 'streamlit'
 Description-Content-Type: text/markdown
 

{sweatstack-0.13.1 → sweatstack-0.14.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sweatstack"
-version = "0.13.1"
+version = "0.14.0"
 description = "The official Python client for SweatStack"
 readme = "README.md"
 authors = [
@@ -22,7 +22,6 @@ jupyterlab = [
 ]
 streamlit = [
     "streamlit>=1.42.0",
-    "streamlit-cookies-controller>=0.0.4",
 ]
 
 [build-system]

{sweatstack-0.13.1 → sweatstack-0.14.0}/src/sweatstack/streamlit.py

@@ -1,7 +1,4 @@
-import hashlib
-import base64
 import os
-import secrets
 import urllib.parse
 
 try:
@@ -13,58 +10,42 @@ except ImportError:
         "pip install 'sweatstack[streamlit]'\n\n"
     )
 import httpx
-from streamlit_cookies_controller import CookieController
 from sweatstack import Client
 
 from .constants import DEFAULT_URL
 
 
-cookie_controller = CookieController()
-
-
 class StreamlitAuth:
-    def __init__(self, set_env_var=False, client_id=None, client_secret=None, scope=None, redirect_uri=None):
+    def __init__(self, client_id=None, client_secret=None, scope=None, redirect_uri=None):
         """
         Args:
-            set_env_var: Whether to set the SWEATSTACK_API_KEY environment variable. Default is False.
             client_id: The client ID to use. If not provided, the SWEATSTACK_CLIENT_ID environment variable will be used.
             client_secret: The client secret to use. If not provided, the SWEATSTACK_CLIENT_SECRET environment variable will be used.
             scope: The scope to use. If not provided, the SWEATSTACK_SCOPE environment variable will be used.
             redirect_uri: The redirect URI to use. If not provided, the SWEATSTACK_REDIRECT_URI environment variable will be used.
         """
-        self.set_env_var = set_env_var
         self.client_id = client_id or os.environ.get("SWEATSTACK_CLIENT_ID")
         self.client_secret = client_secret or os.environ.get("SWEATSTACK_CLIENT_SECRET")
         self.scope = scope or os.environ.get("SWEATSTACK_SCOPE")
         self.redirect_uri = redirect_uri or os.environ.get("SWEATSTACK_REDIRECT_URI")
 
-        self.api_key = cookie_controller.get("sweatstack_api_key")
+        self.api_key = st.session_state.get("sweatstack_api_key")
         self.client = Client(self.api_key)
 
-        if self.api_key and self.set_env_var:
-            os.environ["SWEATSTACK_API_KEY"] = self.api_key
-
     def _show_sweatstack_logout(self):
         if st.button("Logout"):
             self.api_key = None
             self.client = Client()
-            cookie_controller.remove("sweatstack_api_key")
-            if self.set_env_var:
-                os.environ.pop("SWEATSTACK_API_KEY")
+            st.session_state.pop("sweatstack_api_key")
+            st.rerun()
 
     def _show_sweatstack_login(self):
         st.link_button("Login", self._get_authorization_url_implicit())
 
     def _get_authorization_url_implicit(self):
-        code_verifier = secrets.token_urlsafe(32)
-        cookie_controller.set("code_verifier", code_verifier)
-        code_challenge = hashlib.sha256(code_verifier.encode("ascii")).digest()
-        code_challenge = base64.urlsafe_b64encode(code_challenge).rstrip(b"=").decode("ascii")
-
         params = {
             "client_id": self.client_id,
             "redirect_uri": self.redirect_uri,
-            "code_challenge": code_challenge,
             "scope": "data:read",
         }
         path = "/oauth/authorize"
@@ -72,14 +53,12 @@ class StreamlitAuth:
 
         return authorization_url
 
-
-    def _exchange_token_implicit(self, code):
-        code_verifier = cookie_controller.get("code_verifier")
+    def _exchange_token(self, code):
         token_data = {
             "grant_type": "authorization_code",
             "client_id": self.client_id,
+            "client_secret": self.client_secret,
             "code": code,
-            "code_verifier": code_verifier
         }
         response = httpx.post(
             f"{DEFAULT_URL}/oauth/token",
@@ -92,14 +71,10 @@ class StreamlitAuth:
         token_response = response.json()
 
         self.api_key = token_response.get("access_token")
-        cookie_controller.set("sweatstack_api_key", self.api_key)
-        if self.set_env_var:
-            os.environ["SWEATSTACK_API_KEY"] = self.api_key
+        st.session_state["sweatstack_api_key"] = self.api_key
 
         self.client = Client(self.api_key)
 
-        cookie_controller.remove("code_verifier")
-
         return
 
     def is_authenticated(self):
@@ -109,7 +84,7 @@ class StreamlitAuth:
         if self.is_authenticated():
             self._show_sweatstack_logout()
         elif code := st.query_params.get("code"):
-            self._exchange_token_implicit(code)
+            self._exchange_token(code)
             st.query_params.clear()
             st.rerun()
         else:

{sweatstack-0.13.1 → sweatstack-0.14.0}/uv.lock

@@ -1855,21 +1855,9 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/ad/dc/69068179e09488d0833a970d06e8bf40e35669a7bddb8a3caadc13b7dff4/streamlit-1.42.0-py2.py3-none-any.whl", hash = "sha256:edf333fd3525b7c64b19e1156b483a1a93cbdb09a3a06f26478388d68f971090", size = 9560180 },
 ]
 
-[[package]]
-name = "streamlit-cookies-controller"
-version = "0.0.4"
-source = { registry = "https://pypi.org/simple" }
-dependencies = [
-    { name = "streamlit" },
-]
-sdist = { url = "https://files.pythonhosted.org/packages/b0/c8/7caf505d62faeb8c0f6e6b74b554e1c2b0c3ca46c0a39250445c574a5a18/streamlit-cookies-controller-0.0.4.tar.gz", hash = "sha256:f5df2543de858f1585dcb6e90f58f38aafbf313ee8e237c64663b2d9761583cc", size = 406305 }
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/02/c6/3bbdfecf5009943bcb62d37e3fd4b3fe45e87137e449a50e3025170bc9f7/streamlit_cookies_controller-0.0.4-py3-none-any.whl", hash = "sha256:f97fec6acdeee9cb9e16da25c3fc91d404b5b0ddced87c1d9fa9c62f65ca3251", size = 409303 },
-]
-
 [[package]]
 name = "sweatstack"
-version = "0.12.0"
+version = "0.13.2"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },
@@ -1886,7 +1874,6 @@ jupyterlab = [
 ]
 streamlit = [
     { name = "streamlit" },
-    { name = "streamlit-cookies-controller" },
 ]
 
 [package.dev-dependencies]
@@ -1904,7 +1891,6 @@ requires-dist = [
     { name = "pyarrow", specifier = ">=19.0.0" },
     { name = "pydantic", specifier = ">=2.10.5" },
     { name = "streamlit", marker = "extra == 'streamlit'", specifier = ">=1.42.0" },
-    { name = "streamlit-cookies-controller", marker = "extra == 'streamlit'", specifier = ">=0.0.4" },
 ]
 
 [package.metadata.requires-dev]