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

sweatstack/streamlit.py

@@ -39,36 +39,42 @@ class StreamlitAuth:
             st.session_state.pop("sweatstack_api_key")
             st.rerun()
 
+    def _running_on_streamlit_cloud(self):
+        return os.environ.get("HOSTNAME") == "streamlit"
+
     def _show_sweatstack_login(self):
         authorization_url = self._get_authorization_url()
-        st.markdown(
-            f"""
-            <style>
-                .animated-button {{
-                }}
-                .animated-button:hover {{
-                    transform: scale(1.05);
-                }}
-                .animated-button:active {{
-                    transform: scale(1);
-                }}
-            </style>
-            <a href="{authorization_url}"
-                target="_top"
-                class="animated-button"
-                style="display: inline-block;
-                    padding: 10px 20px;
-                    background-color: #EF2B2D;
-                    color: white;
-                    text-decoration: none;
-                    border-radius: 6px;
-                    border: none;
-                    transition: all 0.3s ease;
-                    cursor: pointer;"
-                >Login with SweatStack</a>
-            """,
-            unsafe_allow_html=True,
-        )
+        if not self._running_on_streamlit_cloud():
+            st.markdown(
+                f"""
+                <style>
+                    .animated-button {{
+                    }}
+                    .animated-button:hover {{
+                        transform: scale(1.05);
+                    }}
+                    .animated-button:active {{
+                        transform: scale(1);
+                    }}
+                </style>
+                <a href="{authorization_url}"
+                    target="_top"
+                    class="animated-button"
+                    style="display: inline-block;
+                        padding: 10px 20px;
+                        background-color: #EF2B2D;
+                        color: white;
+                        text-decoration: none;
+                        border-radius: 6px;
+                        border: none;
+                        transition: all 0.3s ease;
+                        cursor: pointer;"
+                    >Login with SweatStack</a>
+                """,
+                unsafe_allow_html=True,
+            )
+        else:
+            st.link_button("Login with SweatStack", authorization_url)
 
     def _get_authorization_url(self):
         params = {
@@ -82,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",
@@ -101,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
 
@@ -122,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.26.0.dist-info → sweatstack-0.28.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sweatstack
-Version: 0.26.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.26.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=Z6nS0R4TwdPHrpGDgXmyCoqhDOQzN-3jp1RtKQIxiSU,4619
+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.26.0.dist-info/METADATA,sha256=6IcxRIyFIVQ27l67nUOJwW-SzVb17vmQ1g4B81zdvDU,775
-sweatstack-0.26.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sweatstack-0.26.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
-sweatstack-0.26.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,,