為什麼你應該更多地使用 attrs

發佈於2024-11-07

Why should you use attrs more

介绍

Python 的 attrs 库对于希望简化类创建并减少样板代码的开发人员来说是一个游戏规则改变者。这个库甚至受到 NASA 的信任。
attrs 由 Hynek Schlawack 于 2015 年创建，因其能够自动生成特殊方法并提供干净、声明性的方式来定义类，而迅速成为 Python 开发人员最喜欢的工具。
dataclasses 是 attrs 的一种子集。

为什么 attrs 有用：

减少样板代码
提高代码可读性和可维护性
提供强大的数据验证和转换功能
通过优化实施提高性能

2. 属性入门

安装：
要开始使用 attrs，您可以使用 pip:
安装它

pip install attrs

基本用法：
下面是一个如何使用 attrs 定义类的简单示例：

import attr

@attr.s
class Person:
    name = attr.ib()
    age = attr.ib()

# Creating an instance
person = Person("Alice", 30)
print(person)  # Person(name='Alice', age=30)

3. attrs的核心特性

一个。自动方法生成：

attrs 会自动为您的类生成 init、repr 和 eq 方法：

@attr.s
class Book:
    title = attr.ib()
    author = attr.ib()
    year = attr.ib()

book1 = Book("1984", "George Orwell", 1949)
book2 = Book("1984", "George Orwell", 1949)

print(book1)  # Book(title='1984', author='George Orwell', year=1949)
print(book1 == book2)  # True

b.具有类型和默认值的属性定义：

import attr
from typing import List

@attr.s
class Library:
    name = attr.ib(type=str)
    books = attr.ib(type=List[str], default=attr.Factory(list))
    capacity = attr.ib(type=int, default=1000)

library = Library("City Library")
print(library)  # Library(name='City Library', books=[], capacity=1000)

c.验证器和转换器：

import attr

def must_be_positive(instance, attribute, value):
    if value 




  
  
  4. 高级使用



  
  
  一个。自定义属性行为：




import attr

@attr.s
class User:
    username = attr.ib()
    _password = attr.ib(repr=False)  # Exclude from repr

    @property
    def password(self):
        return self._password

    @password.setter
    def password(self, value):
        self._password = hash(value)  # Simple hashing for demonstration

user = User("alice", "secret123")
print(user)  # User(username='alice')





  
  
  b.冻结的实例和槽：




@attr.s(frozen=True) # slots=True is the default
class Point:
    x = attr.ib()
    y = attr.ib()

point = Point(1, 2)
try:
    point.x = 3  # This will raise an AttributeError
except AttributeError as e:
    print(e)  # can't set attribute





  
  
  c.工厂函数和初始化后处理：




import attr
import uuid

@attr.s
class Order:
    id = attr.ib(factory=uuid.uuid4)
    items = attr.ib(factory=list)
    total = attr.ib(init=False)

    def __attrs_post_init__(self):
        self.total = sum(item.price for item in self.items)

@attr.s
class Item:
    name = attr.ib()
    price = attr.ib(type=float)

order = Order(items=[Item("Book", 10.99), Item("Pen", 1.99)])
print(order)  # Order(id=UUID('...'), items=[Item(name='Book', price=10.99), Item(name='Pen', price=1.99)], total=12.98)





  
  
  5. 最佳实践和常见陷阱



  
  
  最佳实践：

使用类型注释以获得更好的代码可读性和 IDE 支持
利用验证器确保数据完整性
对不可变对象使用冻结类
利用自动方法生成来减少代码重复

常见陷阱：

忘记在类上使用 @attr.s 装饰器
过度使用可能是单独方法的复杂验证器
不考虑大量使用工厂函数对性能的影响

6. attrs 与其他库的比较

图书馆	特征	表现	社区
属性	自动方法生成、具有类型和默认值的属性定义、验证器和转换器	比手动代码更好的性能	活跃社区
pydantic	数据验证和设置管理、自动方法生成、具有类型和默认值的属性定义、验证器和转换器	表现良好	活跃社区
数据类	内置于 Python 3.7 中，使它们更易于访问	与 Python 版本相关	内置Python库

属性和数据类比 pydantic¹.

与数据类的比较：

attrs 功能更加丰富且灵活
数据类内置于 Python 3.7 中，使它们更易于访问
attrs 在大多数情况下具有更好的性能
数据类与 Python 版本相关，而 attrs 作为外部库可以与任何 Python 版本一起使用。

与pydantic的比较：

pydantic 专注于数据验证和设置管理
attrs 更通用，并且与现有代码库集成得更好
pydantic 具有内置 JSON 序列化，而 attrs 需要额外的库

何时选择属性：

对于具有自定义行为的复杂类层次结构
当您需要对属性定义进行细粒度控制时
对于需要 Python 2 兼容性的项目（尽管现在不太相关）

7. 性能和实际应用

表现：
由于其优化的实现，attrs 通常比手动编写的类或其他库提供更好的性能。

真实示例：

from attr import define, Factory
from typing import List, Optional

@define
class Customer:
    id: int
    name: str
    email: str
    orders: List['Order'] = Factory(list)

@define
class Order:
    id: int
    customer_id: int
    total: float
    items: List['OrderItem'] = Factory(list)

@define
class OrderItem:
    id: int
    order_id: int
    product_id: int
    quantity: int
    price: float

@define
class Product:
    id: int
    name: str
    price: float
    description: Optional[str] = None

# Usage
customer = Customer(1, "Alice", "[email protected]")
product = Product(1, "Book", 29.99, "A great book")
order_item = OrderItem(1, 1, 1, 2, product.price)
order = Order(1, customer.id, 59.98, [order_item])
customer.orders.append(order)

print(customer)