Merge branch 'dev' into syft_error_improvements

OpenMined · May 22, 2024 · 15ccf55 · 15ccf55
2 parents 41f3835 + d185823
commit 15ccf55
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/packages/syft/src/syft/service/code/user_code.py b/packages/syft/src/syft/service/code/user_code.py
@@ -353,13 +353,13 @@ def _coll_repr_(self) -> dict[str, Any]:
     def user(self) -> UserView | SyftError:
         api = APIRegistry.api_for(
             node_uid=self.syft_node_location,
-            user_verify_key=self.user_verify_key,
+            user_verify_key=self.syft_client_verify_key,
         )
         if api is None:
             return SyftError(
                 message=f"Can't access Syft API. You must login to {self.syft_node_location}"
             )
-        return api.services.user.get_current_user()
+        return api.services.user.get_by_verify_key(self.user_verify_key)
 
     @property
     def status(self) -> UserCodeStatusCollection | SyftError:

diff --git a/packages/syft/src/syft/service/user/user_service.py b/packages/syft/src/syft/service/user/user_service.py
@@ -37,6 +37,7 @@
 from .user import UserViewPage
 from .user import check_pwd
 from .user import salt_and_hash_password
+from .user_roles import ADMIN_ROLE_LEVEL
 from .user_roles import DATA_OWNER_ROLE_LEVEL
 from .user_roles import DATA_SCIENTIST_ROLE_LEVEL
 from .user_roles import GUEST_ROLE_LEVEL
@@ -220,7 +221,23 @@ def get_current_user(self, context: AuthedServiceContext) -> UserView | SyftErro
             credentials=context.credentials, verify_key=context.credentials
         )
         if result.is_ok():
-            # this seems weird that we get back None as Ok(None)
+            user = result.ok()
+            if user:
+                return user.to(UserView)
+            else:
+                SyftError(message="User not found!")
+        return SyftError(message=str(result.err()))
+
+    @service_method(
+        path="user.get_by_verify_key", name="get_by_verify_key", roles=ADMIN_ROLE_LEVEL
+    )
+    def get_by_verify_key_endpoint(
+        self, context: AuthedServiceContext, verify_key: SyftVerifyKey
+    ) -> UserView | SyftError:
+        result = self.stash.get_by_verify_key(
+            credentials=context.credentials, verify_key=verify_key
+        )
+        if result.is_ok():
             user = result.ok()
             if user:
                 return user.to(UserView)