Fix type hints for json encoder/decoder

Author: samuelhwilliams

eel/__init__.py

@@ -1,7 +1,7 @@
 from builtins import range
 import traceback
 from io import open
-from typing import Union, Any, Dict, List, Set, Tuple, Optional, Callable, TYPE_CHECKING
+from typing import Union, Any, Dict, List, Set, Tuple, Optional, Callable, TYPE_CHECKING, cast, Type
 
 if TYPE_CHECKING:
     from eel.types import OptionsDictT, WebSocketT
@@ -28,7 +28,7 @@
 mimetypes.add_type('application/javascript', '.js')
 _eel_js_file: str = pkg.resource_filename('eel', 'eel.js')
 _eel_js: str = open(_eel_js_file, encoding='utf-8').read()
-_eel_json_dumps_default_function: Callable = lambda o: None
+_eel_json_dumps_default_function: Callable[[Any], Any] = lambda o: None
 _websockets: List[Tuple[Any, WebSocketT]] = []
 _call_return_values: Dict[Any, Any] = {}
 _call_return_callbacks: Dict[float, Tuple[Callable[..., Any], Optional[Callable[..., Any]]]] = {}
@@ -302,10 +302,10 @@ def register_eel_routes(app: btl.Bottle) -> None:
 # Private functions
 
 def _safe_json_loads(obj: str) -> Any:
-    return jsn.loads(obj, cls=_start_args['json_decoder'])
+    return jsn.loads(obj, cls=cast(Optional[Type[jsn.JSONDecoder]], _start_args['json_decoder']))
 
 def _safe_json_dumps(obj: Any) -> str:
-    return jsn.dumps(obj, cls=_start_args['json_encoder'],
+    return jsn.dumps(obj, cls=cast(Optional[Type[jsn.JSONEncoder]], _start_args['json_encoder']),
                      default=_eel_json_dumps_default_function if not _start_args['json_encoder'] else None)
 
 
@@ -416,7 +416,7 @@ def _detect_shutdown() -> None:
 def _websocket_close(page: str) -> None:
     global _shutdown
 
-    close_callback = _start_args.get('close_callback')
+    close_callback = cast(Callable[..., Any], _start_args.get('close_callback'))
 
     if close_callback is not None:
         if not callable(close_callback):

eel/types.py

@@ -1,4 +1,4 @@
-from typing import Union, Dict, List, Tuple, Callable, Optional, Any, TYPE_CHECKING
+from typing import Union, Dict, List, Tuple, Callable, Optional, Any, TYPE_CHECKING, Type
 
 # This business is slightly awkward, but needed for backward compatibility,
 # because Python < 3.7 doesn't have __future__/annotations, and <3.10 doesn't
@@ -12,17 +12,20 @@
         JinjaEnvironmentT = Environment # type: ignore
     from geventwebsocket.websocket import WebSocket
     WebSocketT = WebSocket
+    from json import JSONDecoder, JSONEncoder
 else:
     JinjaEnvironmentT = None
     WebSocketT = Any
+    JSONEncoder = Any
+    JSONDecoder = Any
 
 OptionsDictT = Dict[
         str,
         Optional[
             Union[
                 str, bool, int, float,
                 List[str], Tuple[int, int], Dict[str, Tuple[int, int]],
-                Callable[..., Any], JinjaEnvironmentT
+                Callable[..., Any], JinjaEnvironmentT, Type[JSONEncoder], Type[JSONDecoder]
             ]
         ]
     ]