sweatstack 0.27.0__tar.gz → 0.28.0__tar.gz

{sweatstack-0.27.0 → sweatstack-0.28.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sweatstack
-Version: 0.27.0
+Version: 0.28.0
 Summary: The official Python client for SweatStack
 Author-email: Aart Goossens <aart@gssns.io>
 Requires-Python: >=3.9

{sweatstack-0.27.0 → sweatstack-0.28.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sweatstack"
-version = "0.27.0"
+version = "0.28.0"
 description = "The official Python client for SweatStack"
 readme = "README.md"
 authors = [

{sweatstack-0.27.0 → sweatstack-0.28.0}/src/sweatstack/streamlit.py

@@ -88,6 +88,11 @@ class StreamlitAuth:
 
         return authorization_url
 
+    def _set_api_key(self, api_key):
+        self.api_key = api_key
+        st.session_state["sweatstack_api_key"] = api_key
+        self.client = Client(self.api_key, streamlit_compatible=True)
+
     def _exchange_token(self, code):
         token_data = {
             "grant_type": "authorization_code",
@@ -107,10 +112,7 @@ class StreamlitAuth:
             raise Exception(f"SweatStack Python login failed. Please try again.") from e
         token_response = response.json()
 
-        self.api_key = token_response.get("access_token")
-        st.session_state["sweatstack_api_key"] = self.api_key
-
-        self.client = Client(self.api_key, streamlit_compatible=True)
+        self._set_api_key(token_response.get("access_token"))
 
         return
 
@@ -128,4 +130,21 @@ class StreamlitAuth:
             st.query_params.clear()
             st.rerun()
         else:
-            self._show_sweatstack_login()
+            self._show_sweatstack_login()
+
+    def switch_user(self):
+        self.switch_back()
+        other_users = self.client.get_users()
+        selected_user = st.selectbox(
+            "Select a user",
+            other_users,
+            format_func=lambda user: user.display_name,
+        )
+        self.client.switch_user(selected_user)
+        self._set_api_key(self.client.api_key)
+
+        return selected_user
+
+    def switch_back(self):
+        self.client.switch_back()
+        self._set_api_key(self.client.api_key)