WIP: MyPy all the things

freezegun/__init__.py

@@ -5,8 +5,8 @@
 :copyright: (c) 2012 by Steve Pulec.
 
 """
-from .api import freeze_time
-from .config import configure
+from .api import freeze_time as freeze_time
+from .config import configure as configure
 
 __title__ = 'freezegun'
 __version__ = '1.3.1'

freezegun/_async.py

@@ -1,18 +1,22 @@
 import functools
-from typing import Any, Callable, TypeVar, cast
+from typing import Awaitable, Any, Callable, TypeVar, Coroutine
+from typing import TYPE_CHECKING
 
+if TYPE_CHECKING:
+    from typing_extensions import ParamSpec
 
-_CallableT = TypeVar("_CallableT", bound=Callable[..., Any])
+    P = ParamSpec("P")
+T = TypeVar("T")
 
 
-def wrap_coroutine(api: Any, coroutine: _CallableT) -> _CallableT:
+def wrap_coroutine(api: Any, coroutine: "Callable[P, Awaitable[T]]") -> "Callable[P, Coroutine[Any, Any, T]]":
     @functools.wraps(coroutine)
-    async def wrapper(*args: Any, **kwargs: Any) -> Any:
+    async def wrapper(*args: "P.args", **kwargs: "P.kwargs") -> T:
         with api as time_factory:
             if api.as_arg:
                 result = await coroutine(time_factory, *args, **kwargs)
             else:
                 result = await coroutine(*args, **kwargs)
         return result
 
-    return cast(_CallableT, wrapper)
+    return wrapper