Python Type Hints: Um Guia Prático para Anotações de Tipo

# Sem type hints - o que essa função espera?
def process_data(data, threshold, output):
    ...
 
# Com type hints - instantaneamente claro
def process_data(data: list[float], threshold: float, output: str) -> dict[str, float]:
    ...

# Anotações de variáveis básicas
name: str = "Alice"
age: int = 30
height: float = 5.9
is_active: bool = True
raw_data: bytes = b"hello"
 
# Anotações sem atribuição (apenas declaração)
username: str
count: int

def greet(name: str) -> str:
    return f"Hello, {name}!"
 
def calculate_average(numbers: list[float]) -> float:
    return sum(numbers) / len(numbers)
 
def save_record(record: dict[str, str], overwrite: bool = False) -> None:
    """Funções que não retornam nada usam -> None."""
    ...

def parse_config(path: str, encoding: str = "utf-8") -> dict[str, str]:
    config: dict[str, str] = {}
    with open(path, encoding=encoding) as f:
        for line in f:
            key, _, value = line.partition("=")
            config[key.strip()] = value.strip()
    return config

# Lists
scores: list[int] = [95, 87, 92]
names: list[str] = ["Alice", "Bob"]
 
# Dictionaries
user_ages: dict[str, int] = {"Alice": 30, "Bob": 25}
config: dict[str, list[str]] = {"servers": ["a.com", "b.com"]}
 
# Tuples - comprimento fixo com tipos específicos
point: tuple[float, float] = (3.14, 2.72)
record: tuple[str, int, bool] = ("Alice", 30, True)
 
# Tuplas de comprimento variável (todo mesmo tipo)
values: tuple[int, ...] = (1, 2, 3, 4, 5)
 
# Sets e frozensets
tags: set[str] = {"python", "typing"}
constants: frozenset[int] = frozenset({1, 2, 3})

from typing import List, Dict, Tuple, Set, FrozenSet
 
scores: List[int] = [95, 87, 92]
user_ages: Dict[str, int] = {"Alice": 30}
point: Tuple[float, float] = (3.14, 2.72)
tags: Set[str] = {"python", "typing"}

# Matriz: lista de listas de floats
matrix: list[list[float]] = [[1.0, 2.0], [3.0, 4.0]]
 
# Mapeando usuários para suas listas de scores
gradebook: dict[str, list[int]] = {
    "Alice": [95, 87, 92],
    "Bob": [78, 85, 90],
}
 
# Configuração: dicionários aninhados
app_config: dict[str, dict[str, str | int]] = {
    "database": {"host": "localhost", "port": 5432},
    "cache": {"host": "redis.local", "port": 6379},
}

from typing import Optional
 
# Sintaxe pré-3.10
def find_user(user_id: int) -> Optional[str]:
    """Retorna username ou None se não encontrado."""
    users = {1: "Alice", 2: "Bob"}
    return users.get(user_id)
 
# Sintaxe Python 3.10+ (preferida)
def find_user(user_id: int) -> str | None:
    """Retorna username ou None se não encontrado."""
    users = {1: "Alice", 2: "Bob"}
    return users.get(user_id)

from typing import Union
 
# Sintaxe pré-3.10
def format_value(value: Union[int, float, str]) -> str:
    return str(value)
 
# Sintaxe Python 3.10+ (preferida)
def format_value(value: int | float | str) -> str:
    return str(value)
 
# Padrão comum: aceitando múltiplos formatos de entrada
def load_data(source: str | Path) -> list[dict[str, str]]:
    path = Path(source) if isinstance(source, str) else source
    with open(path) as f:
        return json.load(f)

from typing import Any
 
def log_event(event: str, payload: Any) -> None:
    """Aceita qualquer tipo de payload - útil para logging genérico."""
    print(f"[{event}] {payload}")
 
# Any é compatível com todo tipo
log_event("click", {"x": 100, "y": 200})
log_event("error", 404)
log_event("message", "hello")

from typing import TypeVar
 
T = TypeVar("T")
 
def first_element(items: list[T]) -> T:
    """Retorna o primeiro elemento, preservando seu tipo."""
    return items[0]
 
# Verificadores de tipo inferem o tipo de retorno correto
name = first_element(["Alice", "Bob"])     # type: str
score = first_element([95, 87, 92])        # type: int
 
# TypeVar bounded - restrito a tipos específicos
Numeric = TypeVar("Numeric", int, float)
 
def add(a: Numeric, b: Numeric) -> Numeric:
    return a + b

from typing import TypeVar, Generic
 
T = TypeVar("T")
 
class Stack(Generic[T]):
    def __init__(self) -> None:
        self._items: list[T] = []
 
    def push(self, item: T) -> None:
        self._items.append(item)
 
    def pop(self) -> T:
        return self._items.pop()
 
    def peek(self) -> T:
        return self._items[-1]
 
    def is_empty(self) -> bool:
        return len(self._items) == 0
 
# Uso com tipos específicos
int_stack: Stack[int] = Stack()
int_stack.push(42)
value: int = int_stack.pop()
 
str_stack: Stack[str] = Stack()
str_stack.push("hello")

from typing import Callable
 
# Função que recebe um callback
def apply_operation(
    values: list[float],
    operation: Callable[[float], float]
) -> list[float]:
    return [operation(v) for v in values]
 
# Uso
import math
results = apply_operation([1.0, 4.0, 9.0], math.sqrt)
# results: [1.0, 2.0, 3.0]
 
# Assinaturas de callable mais complexas
Comparator = Callable[[str, str], int]
EventHandler = Callable[[str, dict[str, str]], None]
 
def sort_with_comparator(
    items: list[str],
    compare: Comparator
) -> list[str]:
    import functools
    return sorted(items, key=functools.cmp_to_key(compare))

from typing import Literal
 
def set_log_level(level: Literal["DEBUG", "INFO", "WARNING", "ERROR"]) -> None:
    print(f"Log level set to {level}")
 
set_log_level("INFO")      # OK
set_log_level("VERBOSE")   # Erro de tipo: não é um literal válido
 
# Útil para parâmetros de modo
def read_file(
    path: str,
    mode: Literal["text", "binary"] = "text"
) -> str | bytes:
    if mode == "text":
        return open(path).read()
    return open(path, "rb").read()

from typing import TypedDict, NotRequired
 
class UserProfile(TypedDict):
    name: str
    email: str
    age: int
    bio: NotRequired[str]  # Chave opcional (Python 3.11+)
 
def display_user(user: UserProfile) -> str:
    return f"{user['name']} ({user['email']}), age {user['age']}"
 
# Verificador de tipo valida a estrutura
user: UserProfile = {
    "name": "Alice",
    "email": "alice@example.com",
    "age": 30,
}
 
display_user(user)  # OK

from typing import Protocol, runtime_checkable
 
@runtime_checkable
class Drawable(Protocol):
    def draw(self, x: int, y: int) -> None: ...
 
class Circle:
    def draw(self, x: int, y: int) -> None:
        print(f"Drawing circle at ({x}, {y})")
 
class Square:
    def draw(self, x: int, y: int) -> None:
        print(f"Drawing square at ({x}, {y})")
 
def render(shape: Drawable, x: int, y: int) -> None:
    shape.draw(x, y)
 
# Ambos funcionam sem herdar de Drawable
render(Circle(), 10, 20)
render(Square(), 30, 40)
 
# runtime_checkable habilita checagens isinstance
print(isinstance(Circle(), Drawable))  # True

from typing import TypeAlias
 
# Aliases simples
UserId: TypeAlias = int
JsonDict: TypeAlias = dict[str, "JsonValue"]
JsonValue: TypeAlias = str | int | float | bool | None | list["JsonValue"] | JsonDict
 
# Aliases complexos simplificam assinaturas
Matrix: TypeAlias = list[list[float]]
Callback: TypeAlias = Callable[[str, int], bool]
Config: TypeAlias = dict[str, str | int | list[str]]
 
def transform_matrix(m: Matrix, factor: float) -> Matrix:
    return [[cell * factor for cell in row] for row in m]
 
# Python 3.12+ usa a instrução type
# type Vector = list[float]

# Instale o mypy
# pip install mypy
 
# Verifique um único arquivo
# mypy script.py
 
# Verifique um projeto inteiro
# mypy src/
 
# Verifique com versão específica do Python
# mypy --python-version 3.10 src/

# Configuração pyproject.toml
# [tool.mypy]
# python_version = "3.10"
# warn_return_any = true
# warn_unused_configs = true
# disallow_untyped_defs = true
# check_untyped_defs = true
# no_implicit_optional = true
# strict_equality = true
 
# Substituições por módulo
# [[tool.mypy.overrides]]
# module = "third_party_lib.*"
# ignore_missing_imports = true

# Erro: Tipo de valor de retorno incompatível (recebeu "Optional[str]", esperava "str")
# Correção: Lide com o caso None
def get_name(user_id: int) -> str:
    result = lookup(user_id)  # retorna str | None
    if result is None:
        raise ValueError(f"User {user_id} not found")
    return result  # mypy sabe que result é str aqui
 
# Erro: Item "None" de "Optional[str]" não tem atributo "upper"
# Correção: Restrinja o tipo primeiro
def format_name(name: str | None) -> str:
    if name is not None:
        return name.upper()
    return "UNKNOWN"
 
# Erro: Necessária anotação de tipo para variável
# Correção: Adicione anotação explícita
items: list[str] = []  # Não apenas: items = []
 
# Erro: Tipos incompatíveis em atribuição
# Correção: Use Union ou corrija o tipo
value: int | str = 42
value = "hello"  # OK com tipo union
 
# Suprimir erros específicos quando necessário
x = some_untyped_function()  # type: ignore[no-untyped-call]

from fastapi import FastAPI
from pydantic import BaseModel
 
app = FastAPI()
 
class UserCreate(BaseModel):
    name: str
    email: str
    age: int
    tags: list[str] = []
 
class UserResponse(BaseModel):
    id: int
    name: str
    email: str
 
@app.post("/users", response_model=UserResponse)
async def create_user(user: UserCreate) -> UserResponse:
    # FastAPI valida o corpo da requisição contra UserCreate
    # e serializa a resposta para corresponder a UserResponse
    return UserResponse(id=1, name=user.name, email=user.email)

import pandas as pd
from typing import Any
 
# Typing básico de DataFrame
def clean_dataframe(df: pd.DataFrame) -> pd.DataFrame:
    return df.dropna().reset_index(drop=True)
 
# Usando pandera para validação de schema
# pip install pandera
import pandera as pa
from pandera.typing import DataFrame, Series
 
class SalesSchema(pa.DataFrameModel):
    product: Series[str]
    quantity: Series[int] = pa.Field(ge=0)
    price: Series[float] = pa.Field(gt=0)
    date: Series[pd.Timestamp]
 
@pa.check_types
def process_sales(df: DataFrame[SalesSchema]) -> DataFrame[SalesSchema]:
    """Processamento de DataFrame com verificação de tipos."""
    return df[df["quantity"] > 0]

# Erro: default mutável com type hint
def bad_append(item: str, items: list[str] = []) -> list[str]:
    items.append(item)  # Default mutável compartilhado!
    return items
 
# Correção: use None como default
def good_append(item: str, items: list[str] | None = None) -> list[str]:
    if items is None:
        items = []
    items.append(item)
    return items

from typing import TypedDict, Callable, TypeAlias
from pathlib import Path
import csv
 
# Defina formas de dados com TypedDict
class RawRecord(TypedDict):
    name: str
    value: str
    category: str
 
class ProcessedRecord(TypedDict):
    name: str
    value: float
    category: str
    normalized: float
 
# Type alias para funções de transformação
Transform: TypeAlias = Callable[[list[ProcessedRecord]], list[ProcessedRecord]]
 
def load_csv(path: Path) -> list[RawRecord]:
    """Carrega dados CSV com saída tipada."""
    records: list[RawRecord] = []
    with open(path) as f:
        reader = csv.DictReader(f)
        for row in reader:
            records.append(RawRecord(
                name=row["name"],
                value=row["value"],
                category=row["category"],
            ))
    return records
 
def parse_records(raw: list[RawRecord]) -> list[ProcessedRecord]:
    """Converte registros raw de strings para registros tipados."""
    max_value = max(float(r["value"]) for r in raw) or 1.0
    return [
        ProcessedRecord(
            name=r["name"],
            value=float(r["value"]),
            category=r["category"],
            normalized=float(r["value"]) / max_value,
        )
        for r in raw
    ]
 
def filter_by_category(
    records: list[ProcessedRecord],
    category: str
) -> list[ProcessedRecord]:
    """Filtra registros, entrada e saída totalmente tipadas."""
    return [r for r in records if r["category"] == category]
 
def apply_transforms(
    records: list[ProcessedRecord],
    transforms: list[Transform]
) -> list[ProcessedRecord]:
    """Aplica uma cadeia de funções de transformação tipadas."""
    result = records
    for transform in transforms:
        result = transform(result)
    return result
 
# Uso
raw = load_csv(Path("data.csv"))
processed = parse_records(raw)
filtered = filter_by_category(processed, "electronics")

import pandas as pd
import pygwalker as pyg
 
# Seu pipeline tipado produz dados limpos e estruturados
data: list[ProcessedRecord] = parse_records(raw)
df = pd.DataFrame(data)
 
# PyGWalker renderiza como uma visualização interativa
walker = pyg.walk(df)