LibCST Codemod to transform relative typing import (from typing import...) or generic typing import (import typing) and their refs to import typing as t + t. prefixed refs respectively.
pip install typing-as-t- Help:
typing-as-t --help
Before:
from typing import Callable, Optional, Generator, cast, Any
a : Callable[..., Any] = lambda : "test"
def func(c: Optional[int] = None) -> Generator:
return cast(Generator, "blabla")After:
import typing as t
a : t.Callable[..., t.Any] = lambda : "test"
def func(c: t.Optional[int] = None) -> t.Generator:
return t.cast(t.Generator, "blabla")Before:
from typing import cast, List, Dict, Any, Sequence, TypeAlias, Sequence, Generator
float_list : TypeAlias = list[float]
def func(z: float_list, c: Sequence[tuple[tuple[str, int], List[Dict[str, Any]]]] = None) -> Generator:
return cast(Generator, "blabla")After:
import typing as t
float_list : t.TypeAlias = list[float]
def func(z: float_list, c: t.Sequence[tuple[tuple[str, int], t.List[t.Dict[str, t.Any]]]] = None) -> t.Generator:
return t.cast(t.Generator, "blabla")Before:
import typing
a : typing.Callable[..., typing.Any] = lambda : "test"
def func(c: typing.Optional[int] = None) -> typing.Generator:
return typing.cast(typing.Generator, "blabla")After:
import typing as t
a : t.Callable[..., t.Any] = lambda : "test"
def func(c: t.Optional[int] = None) -> t.Generator:
return t.cast(t.Generator, "blabla")Before:
from typing import Any, TYPE_CHECKING as TC
from typing import TypeAlias as TA
if TC:
foo: dict[str, Any]
Vector: TA = list[float]After:
import typing as t
if t.TYPE_CHECKING:
a: dict[str, t.Any]
Vector: t.TypeAlias = list[float]
