From 87ead6a3810bf99f0daa34a7a4f3da9211954daa Mon Sep 17 00:00:00 2001 From: brandon Date: Mon, 14 Oct 2024 10:05:50 -0700 Subject: [PATCH 1/3] adds fixes for moving to a new auth provider --- src/groundlight/client.py | 4 ++-- test/unit/test_user.py | 9 +++++++++ 2 files changed, 11 insertions(+), 2 deletions(-) create mode 100644 test/unit/test_user.py 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..61bc6d46 --- /dev/null +++ b/test/unit/test_user.py @@ -0,0 +1,9 @@ +from datetime import datetime + +import pytest +from groundlight import Groundlight + +def test_whoami(gl: Groundlight): + user = gl.whoami() + assert user is not None + assert isinstance(user, str) \ No newline at end of file From 961ac01916db58a8a856fe25ae0dc5c2b1811c19 Mon Sep 17 00:00:00 2001 From: Auto-format Bot Date: Mon, 14 Oct 2024 17:07:31 +0000 Subject: [PATCH 2/3] Automatically reformatting code --- test/unit/test_user.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/test/unit/test_user.py b/test/unit/test_user.py index 61bc6d46..927da25b 100644 --- a/test/unit/test_user.py +++ b/test/unit/test_user.py @@ -1,9 +1,7 @@ -from datetime import datetime - -import pytest from groundlight import Groundlight + def test_whoami(gl: Groundlight): user = gl.whoami() assert user is not None - assert isinstance(user, str) \ No newline at end of file + assert isinstance(user, str) From 78e11ef53e8af400ec28d355792ea69128eb04a7 Mon Sep 17 00:00:00 2001 From: brandon Date: Mon, 14 Oct 2024 10:25:07 -0700 Subject: [PATCH 3/3] update pyproject version in order to push to pypi --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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