モダンPython上級 #6 typing 上級 — Variance、ParamSpec、Self、overload

def add_dog(animals: list[Animal]) -> None:
    animals.append(Dog())

cats: list[Cat] = [Cat(), Cat()]
add_dog(cats)    # ✗ もし許されると、cats の中に Dog が入ってしまう

def feed_animal(a: Animal): ...
def feed_cat(c: Cat): ...

# 'Animal を受け取る関数' の位置に 'Cat だけ受け取る関数' を入れられるか?
fn: Callable[[Animal], None] = feed_cat   # ✗ — Dog が来ると壊れる

# 逆は?
fn2: Callable[[Cat], None] = feed_animal  # ✅ — Cat も Animal の一部なので OK

from typing import TypeVar

T_co = TypeVar("T_co", covariant=True)
T_contra = TypeVar("T_contra", contravariant=True)

class Producer(Generic[T_co]):
    def get(self) -> T_co: ...

class Consumer(Generic[T_contra]):
    def put(self, item: T_contra) -> None: ...

def log[**P, R](fn: Callable[P, R]) -> Callable[P, R]:
    def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
        ...
    return wrapper

from typing import Concatenate, Callable

def with_logger[**P, R](
    fn: Callable[Concatenate[Logger, P], R]
) -> Callable[P, R]:
    logger = make_logger()
    def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
        return fn(logger, *args, **kwargs)
    return wrapper

@with_logger
def do_work(logger: Logger, x: int, y: int) -> int:
    logger.info(f"{x} + {y}")
    return x + y

do_work(2, 3)   # logger は自動で注入される

from typing import Self

class Builder:
    def add(self, item: str) -> Self:
        ...
        return self

class SubBuilder(Builder):
    def special(self) -> Self:
        ...
        return self

b = SubBuilder().add("x").special()
# b の型は正確に SubBuilder と推論される

class Item:
    @classmethod
    def from_dict(cls, data: dict) -> Self:
        return cls(**data)

from typing import overload

@overload
def parse(value: str) -> str: ...
@overload
def parse(value: int) -> int: ...
@overload
def parse(value: list[str]) -> list[str]: ...

def parse(value):
    # 実際の実装 — overload シグネチャの下
    return value

a = parse("hello")        # str
b = parse(42)              # int
c = parse(["a", "b"])      # list[str]

@overload
def get(key: Literal["count"]) -> int: ...
@overload
def get(key: Literal["name"]) -> str: ...
@overload
def get(key: str) -> object: ...

def get(key): ...

@overload
def get(self, key: K) -> V | None: ...
@overload
def get(self, key: K, default: V) -> V: ...
@overload
def get(self, key: K, default: T) -> V | T: ...

from typing import TypeGuard

def is_str_list(items: list[object]) -> TypeGuard[list[str]]:
    return all(isinstance(x, str) for x in items)

def process(data: list[object]) -> None:
    if is_str_list(data):
        # ここで data は list[str] に絞り込まれる
        print(", ".join(data))

from typing import TypeIs

def is_str(x: object) -> TypeIs[str]:
    return isinstance(x, str)

def handle(x: int | str) -> None:
    if is_str(x):
        print(len(x))     # str
    else:
        print(x + 1)      # int  ← ここが違う!

from typing import Annotated

UserId = Annotated[int, "UserID — DB の users.id"]
Email = Annotated[str, "メール形式"]

def find(id: UserId, email: Email): ...

from typing import Annotated
from fastapi import Depends, Query

def get_db(): ...

def search(
    q: Annotated[str, Query(min_length=3)],
    db: Annotated[Database, Depends(get_db)],
): ...

from typing import LiteralString

def execute(query: LiteralString) -> list[Row]:
    ...

execute("SELECT * FROM users WHERE id = 1")    # OK
execute(f"SELECT * FROM users WHERE id = {user_input}")  # ✗ — f-string の結果は LiteralString ではない

from typing import NewType

UserId = NewType("UserId", int)
ProductId = NewType("ProductId", int)

def get_user(id: UserId): ...

uid = UserId(123)
pid = ProductId(456)

get_user(uid)    # OK
get_user(pid)    # ✗ — 別の型として扱われる
get_user(123)    # ✗ — int のままでは入らない

class Cache[K, V]:
    def __init__(self):
        self._data: dict[K, V] = {}

    def get(self, key: K) -> V | None: ...
    def put(self, key: K, value: V) -> None: ...

cache: Cache[str, int] = Cache()

class SortedList[T: (int, str)]:    # int または str だけ
    ...

class Container[T: Comparable]:     # Comparable のサブタイプだけ
    ...

def stack[*Ts](*args: *Ts) -> tuple[*Ts]:
    return args

t = stack(1, "hello", 3.14)
# t の型: tuple[int, str, float]

from typing import cast

raw: object = get_data()
data = cast(dict[str, int], raw)
# ランタイムにはただの raw、型チェッカーだけが dict[str, int] として見る

from typing import assert_type

x = some_function()
assert_type(x, int)    # チェッカーが x を int と推論しなければエラー

from typing import assert_never

def handle(event: Literal["click", "key"]):
    if event == "click": ...
    elif event == "key": ...
    else:
        assert_never(event)    # 新しいケースが増えたらここが壊れる