updated auth token handling using new users api.

mdrsclient/api/__init__.py

@@ -1,11 +1,11 @@
 from mdrsclient.api.file import FileApi
 from mdrsclient.api.folder import FolderApi
 from mdrsclient.api.laboratory import LaboratoryApi
-from mdrsclient.api.user import UserApi
+from mdrsclient.api.users import UsersApi
 
 __all__ = [
     "FileApi",
     "FolderApi",
     "LaboratoryApi",
-    "UserApi",
+    "UsersApi",
 ]

mdrsclient/api/users.py

@@ -16,27 +16,55 @@ class UserAuthResponse(Token):
     laboratories: list[Laboratory] = field(default_factory=list)
 
 
-class UserApi(BaseApi):
+@dataclass(frozen=True)
+class UsersCurrentResponseLaboratory:
+    id: int
+    name: str
+    role: int
+
+
+@dataclass(frozen=True)
+class UsersCurrentResponse:
+    id: int
+    username: str
+    full_name: str
+    email: str
+    laboratories: list[UsersCurrentResponseLaboratory]
+    is_staff: bool
+    is_active: bool
+    is_superuser: bool
+    is_reviewer: bool
+    last_login: str  # ISO8601
+    date_joined: str  # ISO8601
+
+
+class UsersApi(BaseApi):
     ENTRYPOINT: Final[str] = "v2/"
 
-    def auth(self, username: str, password: str) -> tuple[User, Token]:
+    def current(self) -> User:
         # print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
-        url = self.ENTRYPOINT + "auth/"
+        url = self.ENTRYPOINT + "users/current/"
+        response = self.connection.get(url)
+        self._raise_response_error(response)
+        obj = TypeAdapter(UsersCurrentResponse).validate_python(response.json())
+        laboratory_ids = list(map(lambda x: x.id, obj.laboratories))
+        user = User(id=obj.id, username=obj.username, laboratory_ids=laboratory_ids, is_reviewer=obj.is_reviewer)
+        return user
+
+    def token(self, username: str, password: str) -> Token:
+        # print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
+        url = self.ENTRYPOINT + "users/token/"
         data: dict[str, str | int] = {"username": username, "password": password}
         response = self.connection.post(url, data=data)
         if response.status_code == requests.codes.unauthorized:
             raise UnauthorizedException("Invalid username or password.")
         self._raise_response_error(response)
-        obj = TypeAdapter(UserAuthResponse).validate_python(response.json())
-        token = Token(access=obj.access, refresh=obj.refresh)
-        laboratory_ids = list(map(lambda x: x.id, obj.laboratories))
-        is_reviewer = obj.is_reviewer if obj.is_reviewer is not None else False
-        user = User(id=token.user_id, username=username, laboratory_ids=laboratory_ids, is_reviewer=is_reviewer)
-        return (user, token)
+        token = TypeAdapter(Token).validate_python(response.json())
+        return token
 
-    def refresh(self, token: Token) -> Token:
+    def tokenRefresh(self, token: Token) -> Token:
         # print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
-        url = self.ENTRYPOINT + "refresh/"
+        url = self.ENTRYPOINT + "users/token/refresh/"
         data: dict[str, str | int] = {"refresh": token.refresh}
         response = self.connection.post(url, data=data)
         if response.status_code == requests.codes.unauthorized:

mdrsclient/api/utils.py

@@ -1,4 +1,4 @@
-from mdrsclient.api.user import UserApi
+from mdrsclient.api.users import UsersApi
 from mdrsclient.connection import MDRSConnection
 from mdrsclient.exceptions import UnauthorizedException
 
@@ -8,9 +8,9 @@ def token_check(connection: MDRSConnection) -> None:
         connection.lock.acquire()
         if connection.token is not None:
             if connection.token.is_refresh_required:
-                user_api = UserApi(connection)
+                user_api = UsersApi(connection)
                 try:
-                    connection.token = user_api.refresh(connection.token)
+                    connection.token = user_api.tokenRefresh(connection.token)
                 except UnauthorizedException:
                     connection.logout()
             elif connection.token.is_expired: