sweatstack 0.27.0__py3-none-any.whl → 0.28.0__py3-none-any.whl

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)

{sweatstack-0.27.0.dist-info → sweatstack-0.28.0.dist-info}/METADATA

@@ -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.dist-info → sweatstack-0.28.0.dist-info}/RECORD

@@ -7,11 +7,11 @@ sweatstack/jupyterlab_oauth2_startup.py,sha256=eZ6xi0Sa4hO4vUanimq0SqjduHtiywCUR
 sweatstack/openapi_schemas.py,sha256=XlgiL7qkfcfoDHcQrIm9e5hvhY98onC0QmZWG69bl-s,13441
 sweatstack/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sweatstack/schemas.py,sha256=CArM9jgrJDBUT-ZIIiGN_insqaEGV43MX134Ezf4jyA,103
-sweatstack/streamlit.py,sha256=zBPIC29OvgqvuKNlezgLOTwgvS-Cbg885QFwuW4NMRk,4968
+sweatstack/streamlit.py,sha256=z4xggT1igrTLQCPMQeCBbXUnC2w35dFSH3NSQRVrZgc,5527
 sweatstack/sweatshell.py,sha256=MYLNcWbOdceqKJ3S0Pe8dwHXEeYsGJNjQoYUXpMTftA,333
 sweatstack/utils.py,sha256=HtR1NNCKus59vfgfaCSFS-tHA11mtdcuUx5lS6ZX58g,1773
 sweatstack/Sweat Stack examples/Getting started.ipynb,sha256=k2hiSffWecoQ0VxjdpDcgFzBXDQiYEebhnAYlu8cgX8,6335204
-sweatstack-0.27.0.dist-info/METADATA,sha256=YKWKjaX0UYaqA3ODVWZKlOqba4z-xKF0PI5TS8z3P7E,775
-sweatstack-0.27.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sweatstack-0.27.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
-sweatstack-0.27.0.dist-info/RECORD,,
+sweatstack-0.28.0.dist-info/METADATA,sha256=o6fVjDMx337xaw9ujyRtRtfWIaNTdIU4FjrAMe9zMjw,775
+sweatstack-0.28.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sweatstack-0.28.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
+sweatstack-0.28.0.dist-info/RECORD,,