Login Schema now in use.

usnistgov · Sep 11, 2023 · c610542 · c610542
1 parent 4ec1dc5
commit c610542
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/src/dioptra/restapi/auth/controller.py b/src/dioptra/restapi/auth/controller.py
@@ -56,8 +56,7 @@ def post(self) -> Response:
         log: BoundLogger = LOGGER.new(
             request_id=str(uuid.uuid4()), resource="auth", request_type="POST"
         )
-        return self._auth_service.login(data=request.parsed_obj, log=log)
-        # return self._auth_service.login(login_data=request.parsed_obj, log=log)
+        return self._auth_service.login(login_data=request.parsed_obj, log=log)
 
 @api.route("/logout")
 class LogoutResource(Resource):

diff --git a/src/dioptra/restapi/auth/service.py b/src/dioptra/restapi/auth/service.py
@@ -44,16 +44,13 @@ def __init__(
 
     def login(
         self, 
-        data: dict[str, Any],
-        # login_data: LoginData,
+        login_data: LoginData,
         **kwargs,
     ) -> Response:
     # ) -> Response | tuple[Response, int]:
         log: BoundLogger = kwargs.get("log", LOGGER.new())
-        username = data.get("username", "guest")
-        password = data.get("password", "")
-        # username = login_data.get("username", "guest")
-        # password = login_data.get("password", "")
+        username = login_data.get("username", "guest")
+        password = login_data.get("password", "")
         user = self._user_service.authenticate_user(name=username, password=password)
 
         if not user:

diff --git a/src/dioptra/restapi/errors.py b/src/dioptra/restapi/errors.py
@@ -29,6 +29,7 @@ def register_error_handlers(api: Api) -> None:
     Args:
         api: The main REST |Api| object.
     """
+    from .auth import register_error_handlers as attach_auth_error_handlers
     from .experiment import register_error_handlers as attach_experiment_error_handlers
     from .job import register_error_handlers as attach_job_error_handlers
     from .queue import register_error_handlers as attach_job_queue_error_handlers
@@ -38,6 +39,7 @@ def register_error_handlers(api: Api) -> None:
     from .user import register_error_handlers as attach_user_error_handlers
 
     # Add error handlers
+    attach_auth_error_handlers(api)
     attach_experiment_error_handlers(api)
     attach_job_error_handlers(api)
     attach_job_queue_error_handlers(api)