python的dataclasses

在写代码过程中，我们经常需要创建一些简单的类来存储数据。传统方式需要手动编写__init__、__repr__、__eq__等方法，不仅繁琐还容易出错。直到引入的dataclasses模块。

安装

Python 3.6，可以通过pip安装：

pip install dataclasses

【3.7 +的内置模块，无需安装即可使用】

 

基本用法

from dataclasses import dataclass@dataclass
class Book:title: strauthor: strprice: floatpages: int = 0book = Book("python", "Lowen", 99)
print(book)  #Book(title='python', author='Lowen', price=99, pages=0)
print(book.title)  #python
print(book.author) #Lowen
print(book.price) #99
print(book.pages) #0

　　

只需要添加@dataclass装饰器，就自动获得了：

• 初始化方法
• 字符串表示
• 比较方法
• 可选的默认值支持

 

高级用法

from dataclasses import dataclass, field, asdict, replace@dataclass(frozen=True) # 创建不可变类
class Point:x: floaty: float# 使用field自定义字段属性coordinates: tuple = field(default_factory=tuple, repr=False)# 不可变对象
p1 = Point(3, 4)
p11 = Point(3, 4, (4,))
print(p1)  # Point(x=3, y=4)
print(p11.coordinates[0])  # 4
# p11.x = 5  # 这会引发错误# 转换为字典
point_dict = asdict(p1)
print(point_dict)  # {'x': 3, 'y': 4, 'coordinates': ()}# 创建对象的修改副本
p2 = replace(p1, x=5)
print(p2)  # Point(x=5, y=4)

　　

实战用法

from dataclasses import dataclass, field
from typing import List
from decimal import Decimal@dataclass
class Product:name: strprice: Decimalquantity: int = 0@dataclass
class ShoppingCart:items: List[Product] = field(default_factory=list)def add_item(self, product: Product):self.items.append(product)@propertydef total(self) -> Decimal:return sum(item.price * item.quantity for item in self.items)# 使用示例
cart = ShoppingCart()
cart.add_item(Product("Python书籍", Decimal("49.9"), 2))
cart.add_item(Product("键盘", Decimal("299"), 1))
print(f"购物车总价: ￥{cart.total}")