Fix mypy

Author: brichet

jupyter_server/auth/identity.py

@@ -20,7 +20,7 @@
 from http.cookies import Morsel
 
 from tornado import escape, httputil, web
-from traitlets import Bool, Dict, List, TraitError, Type, Unicode, default, validate
+from traitlets import Bool, Dict, Enum, List, TraitError, Type, Unicode, default, validate
 from traitlets.config import LoggingConfigurable
 
 from jupyter_server.transutils import _i18n
@@ -194,7 +194,7 @@ class IdentityProvider(LoggingConfigurable):
 
     # Define the fields that can be updated
     updatable_fields = List(
-        trait=Unicode(),
+        trait=Enum(list(t.get_args(UpdatableField))),
         default_value=["color"],  # Default updatable field
         config=True,
         help=_i18n("List of fields in the User model that can be updated."),
@@ -296,7 +296,7 @@ def update_user(
         self, handler: web.RequestHandler, user_data: dict[UpdatableField, str]
     ) -> User:
         """Update user information."""
-        current_user = handler.current_user  # type:ignore[attr-defined]
+        current_user = t.cast(User, handler.current_user)
 
         for field in user_data:
             if field not in self.updatable_fields:

jupyter_server/services/api/handlers.py

@@ -4,14 +4,14 @@
 # Distributed under the terms of the Modified BSD License.
 import json
 import os
-from typing import Any
+from typing import Any, cast
 
 from jupyter_core.utils import ensure_async
 from tornado import web
 
 from jupyter_server._tz import isoformat, utcfromtimestamp
 from jupyter_server.auth.decorator import authorized
-from jupyter_server.auth.identity import IdentityProvider
+from jupyter_server.auth.identity import IdentityProvider, UpdatableField
 
 from ...base.handlers import APIHandler, JupyterHandler
 
@@ -118,7 +118,7 @@ async def get(self):
     @web.authenticated
     async def patch(self):
         """Update user information."""
-        user_data = self.get_json_body()
+        user_data = cast(dict[UpdatableField, str], self.get_json_body())
         if not user_data:
             raise web.HTTPError(400, "Invalid or missing JSON body")