sweatstack 0.19.1__py3-none-any.whl → 0.20.0__py3-none-any.whl

sweatstack/client.py

@@ -107,7 +107,14 @@ class OAuth2Mixin:
 
 
 class DelegationMixin:
-    def _get_delegated_token(self, user_id: str):
+    def _validate_user(self, user: str | UserSummary):
+        if isinstance(user, UserSummary):
+            return user.id
+        else:
+            return user
+
+    def _get_delegated_token(self, user: str | UserSummary):
+        user_id = self._validate_user(user)
         with self._http_client() as client:
             response = client.post(
                 "/api/v1/oauth/delegated-token",
@@ -117,8 +124,8 @@ class DelegationMixin:
 
         return response.json()
 
-    def switch_user(self, user_id: str):
-        token_response = self._get_delegated_token(user_id)
+    def switch_user(self, user: str | UserSummary):
+        token_response = self._get_delegated_token(user)
         self.api_key = token_response["access_token"]
         self.refresh_token = token_response["refresh_token"]
 
@@ -135,8 +142,8 @@ class DelegationMixin:
         self.api_key = token_response["access_token"]
         self.refresh_token = token_response["refresh_token"]
 
-    def delegated_client(self, user_id: str):
-        token_response = self._get_delegated_token(user_id)
+    def delegated_client(self, user: str | UserSummary):
+        token_response = self._get_delegated_token(user)
         return self.__class__(
             api_key=token_response["access_token"],
             refresh_token=token_response["refresh_token"],
@@ -154,7 +161,7 @@ class DelegationMixin:
         )
 
 
-class Client(OAuth2Mixin, DelegationMixin, JupyterInteractivityMixin):
+class Client(OAuth2Mixin, DelegationMixin):
     def __init__(
         self,
         api_key: str | None = None,
@@ -604,7 +611,7 @@ class Client(OAuth2Mixin, DelegationMixin, JupyterInteractivityMixin):
             response.raise_for_status()
             return response.json()
 
-    def list_users(self) -> list[UserSummary]:
+    def get_users(self) -> list[UserSummary]:
         with self._http_client() as client:
             response = client.get(
                 url="/api/v1/users/",
@@ -651,7 +658,7 @@ _generate_singleton_methods(
     [
         "login",
 
-        "list_users",
+        "get_users",
 
         "get_activities",
 

{sweatstack-0.19.1.dist-info → sweatstack-0.20.0.dist-info}/METADATA

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

{sweatstack-0.19.1.dist-info → sweatstack-0.20.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 sweatstack/__init__.py,sha256=tiVfgKlswRPaDMEy0gA7u8rveqEYZTA_kyB9lJ3J6Sc,21
 sweatstack/cli.py,sha256=N1NWOgEZR2yaJvIXxo9qvp_jFlypZYb0nujpbVNYQ6A,720
-sweatstack/client.py,sha256=iPSuaRZRQIJUygywZL2PWkCMIeDm6ErSdSDwCPM_PtQ,22191
+sweatstack/client.py,sha256=8oYnYisQHRJWdEUO_SqIf0xU48PvkInUO90LYwr98Qo,22396
 sweatstack/constants.py,sha256=fGO6ksOv5HeISv9lHRoYm4besW1GTveXS8YD3K0ljg0,41
 sweatstack/ipython_init.py,sha256=zBGUlMFkdpLvsNpOpwrNaKRUpUZhzaICvH8ODJgMPcI,229
 sweatstack/jupyterlab_oauth2_startup.py,sha256=eZ6xi0Sa4hO4vUanimq0SqjduHtiywCURSDNWk_I-7s,1200
@@ -11,7 +11,7 @@ sweatstack/streamlit.py,sha256=F5oQdWkcJ76Um5fRqFT5QrjpEz8v3OaiH9kMQOYktgo,4466
 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.19.1.dist-info/METADATA,sha256=zj_LBkQjFACMDH_c7lWV7rPttI_lG9muNqzY45QgXFA,2962
-sweatstack-0.19.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sweatstack-0.19.1.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
-sweatstack-0.19.1.dist-info/RECORD,,
+sweatstack-0.20.0.dist-info/METADATA,sha256=DbpDHyf-ffe-w3whUXOdksfENiOR1ikQyfnvzVcPMYM,2962
+sweatstack-0.20.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sweatstack-0.20.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
+sweatstack-0.20.0.dist-info/RECORD,,