Python 类型提示高级应用指南

最新推荐文章于 2026-06-19 19:20:54 发布

原创最新推荐文章于 2026-06-19 19:20:54 发布 · 158 阅读

3 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#rust #github #python

Python 类型提示高级应用指南

1. 什么是类型提示？

类型提示是 Python 3.5+ 引入的特性，允许我们为变量、函数参数和返回值添加类型标注。它可以提高代码的可读性和可维护性，同时也可以帮助 IDE 提供更好的代码补全和类型检查。

# 基本类型提示
def greet(name: str) -> str:
    return f"Hello, {name}!"

# 变量类型提示
age: int = 25

2. 基本类型

2.1 内置类型

# 基本类型
x: int = 1
y: float = 1.0
z: bool = True
s: str = "hello"

# 容器类型
from typing import List, Tuple, Dict, Set

numbers: List[int] = [1, 2, 3]
point: Tuple[float, float] = (1.0, 2.0)
person: Dict[str, str] = {"name": "Alice", "age": "30"}
tags: Set[str] = {"python", "programming"}

2.2 泛型类型

from typing import Generic, TypeVar, List

T = TypeVar('T')

class Stack(Generic[T]):
    def __init__(self):
        self.items: List[T] = []
    
    def push(self, item: T) -> None:
        self.items.append(item)
    
    def pop(self) -> T:
        return self.items.pop()

# 使用泛型栈
int_stack = Stack[int]()
int_stack.push(1)
int_stack.push(2)

str_stack = Stack[str]()
str_stack.push("hello")
str_stack.push("world")

3. 高级类型

3.1 Optional 类型

from typing import Optional

def greet(name: Optional[str] = None) -> str:
    if name:
        return f"Hello, {name}!"
    return "Hello, World!"

3.2 Union 类型

from typing import Union

def process_value(value: Union[int, str, float]) -> str:
    if isinstance(value, int):
        return f"Integer: {value}"
    elif isinstance(value, str):
        return f"String: {value}"
    else:
        return f"Float: {value}"

3.3 Literal 类型

from typing import Literal

def get_status(code: Literal[200, 404, 500]) -> str:
    if code == 200:
        return "Success"
    elif code == 404:
        return "Not Found"
    else:
        return "Internal Server Error"

3.4 Type 类型

from typing import Type, TypeVar

T = TypeVar('T')

def create_instance(cls: Type[T]) -> T:
    return cls()

class Person:
    def __init__(self):
        self.name = ""

person = create_instance(Person)

3.5 Callable 类型

from typing import Callable

def apply_function(func: Callable[[int, int], int], a: int, b: int) -> int:
    return func(a, b)

def add(a: int, b: int) -> int:
    return a + b

result = apply_function(add, 1, 2)
print(result)  # 输出: 3

3.6 Any 类型

from typing import Any

def process_any(value: Any) -> None:
    print(value)

process_any(1)
process_any("hello")
process_any([1, 2, 3])

4. 类型别名

类型别名可以为复杂的类型定义一个更简洁的名称。

from typing import List, Dict, Tuple, TypeAlias

# 类型别名
UserId: TypeAlias = int
UserDict: TypeAlias = Dict[str, str]
Point: TypeAlias = Tuple[float, float]

# 使用类型别名
def get_user(user_id: UserId) -> UserDict:
    return {"id": str(user_id), "name": "Alice"}

def get_distance(p1: Point, p2: Point) -> float:
    return ((p1[0] - p2[0]) ** 2 + (p1[1] - p2[1]) ** 2) ** 0.5

5. 协议和抽象基类

5.1 协议

协议是一种定义接口的方式，它允许我们指定对象应该具有哪些方法和属性。

from typing import Protocol

class Drawable(Protocol):
    def draw(self) -> None:
        ...

class Circle:
    def draw(self) -> None:
        print("Drawing a circle")

class Square:
    def draw(self) -> None:
        print("Drawing a square")

def render(obj: Drawable) -> None:
    obj.draw()

# 使用协议
circle = Circle()
square = Square()

render(circle)  # 输出: Drawing a circle
render(square)  # 输出: Drawing a square

5.2 抽象基类

抽象基类是一种不能直接实例化的类，它定义了子类必须实现的方法。

from abc import ABC, abstractmethod

class Shape(ABC):
    @abstractmethod
    def area(self) -> float:
        pass

class Circle(Shape):
    def __init__(self, radius: float):
        self.radius = radius
    
    def area(self) -> float:
        return 3.14159 * self.radius ** 2

class Rectangle(Shape):
    def __init__(self, width: float, height: float):
        self.width = width
        self.height = height
    
    def area(self) -> float:
        return self.width * self.height

# 使用抽象基类
def calculate_area(shape: Shape) -> float:
    return shape.area()

circle = Circle(5.0)
rectangle = Rectangle(4.0, 6.0)

print(calculate_area(circle))  # 输出: 78.53975
print(calculate_area(rectangle))  # 输出: 24.0

6. 类型检查

6.1 使用 mypy 进行类型检查

# 安装 mypy
pip install mypy

# 运行类型检查
mypy your_script.py

6.2 类型检查示例

# example.py
def add(a: int, b: int) -> int:
    return a + b

# 正确使用
result = add(1, 2)
print(result)  # 输出: 3

# 错误使用 (mypy 会检测到)
result = add("1", "2")
print(result)  # 运行时会工作，但 mypy 会报错

7. 实际应用场景

7.1 函数参数和返回值类型提示

from typing import List, Dict, Optional

def process_users(users: List[Dict[str, str]]) -> Optional[Dict[str, str]]:
    """处理用户列表，返回第一个用户"""
    if users:
        return users[0]
    return None

# 使用函数
users = [
    {"id": "1", "name": "Alice"},
    {"id": "2", "name": "Bob"}
]

first_user = process_users(users)
print(first_user)  # 输出: {'id': '1', 'name': 'Alice'}

7.2 类属性和方法类型提示

from typing import List

class User:
    def __init__(self, name: str, age: int):
        self.name: str = name
        self.age: int = age
    
    def greet(self) -> str:
        return f"Hello, my name is {self.name}"

class UserManager:
    def __init__(self):
        self.users: List[User] = []
    
    def add_user(self, user: User) -> None:
        self.users.append(user)
    
    def get_user_by_name(self, name: str) -> User:
        for user in self.users:
            if user.name == name:
                return user
        raise ValueError(f"User {name} not found")

# 使用类
manager = UserManager()
user1 = User("Alice", 30)
user2 = User("Bob", 25)

manager.add_user(user1)
manager.add_user(user2)

user = manager.get_user_by_name("Alice")
print(user.greet())  # 输出: Hello, my name is Alice

7.3 类型提示与 IDE 集成

类型提示可以帮助 IDE 提供更好的代码补全和错误检查。

# IDE 会提示 name 是 str 类型，age 是 int 类型
class Person:
    def __init__(self, name: str, age: int):
        self.name = name
        self.age = age

# IDE 会提示 greet 方法返回 str 类型
def greet(person: Person) -> str:
    return f"Hello, {person.name}!"

# IDE 会提示 person 没有 email 属性
person = Person("Alice", 30)
print(person.email)  # IDE 会标记为错误