diff --git a/pyproject.toml b/pyproject.toml index 2b5228ba..10200297 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ packages = [ {include = "**/*.py", from = "src"}, ] readme = "README.md" -version = "0.18.2" +version = "0.18.3" [tool.poetry.dependencies] # For certifi, use ">=" instead of "^" since it upgrades its "major version" every year, not really following semver diff --git a/src/groundlight/client.py b/src/groundlight/client.py index 55130782..8f520e73 100644 --- a/src/groundlight/client.py +++ b/src/groundlight/client.py @@ -205,7 +205,7 @@ def whoami(self) -> str: :return: str """ obj = self.user_api.who_am_i() - return obj["username"] + return obj["email"] def _user_is_privileged(self) -> bool: """ @@ -213,7 +213,7 @@ def _user_is_privileged(self) -> bool: Privleged users have elevated permissions, so care should be taken when using a privileged account. """ obj = self.user_api.who_am_i() - return obj["is_superuser"] or obj["is_staff"] + return obj["is_superuser"] def get_detector(self, id: Union[str, Detector]) -> Detector: # pylint: disable=redefined-builtin """ diff --git a/test/unit/test_user.py b/test/unit/test_user.py new file mode 100644 index 00000000..927da25b --- /dev/null +++ b/test/unit/test_user.py @@ -0,0 +1,7 @@ +from groundlight import Groundlight + + +def test_whoami(gl: Groundlight): + user = gl.whoami() + assert user is not None + assert isinstance(user, str)