← 返回首页
PEP 557 – Data Classes | peps.python.org Following system colour scheme Selected dark colour scheme Selected light colour scheme

Python Enhancement Proposals

Toggle light / dark / auto colour theme PEP 557 – Data Classes

PEP 557 – Data Classes

Author: Eric V. Smith <eric at trueblade.com> Status: Final Type: Standards Track Created: 02-Jun-2017 Python-Version: 3.7 Post-History: 08-Sep-2017, 25-Nov-2017, 30-Nov-2017, 01-Dec-2017, 02-Dec-2017, 06-Jan-2018, 04-Mar-2018 Resolution: Python-Dev message Table of Contents

Notice for Reviewers

This PEP and the initial implementation were drafted in a separate repo: https://github.com/ericvsmith/dataclasses. Before commenting in a public forum please at least read the discussion listed at the end of this PEP.

Abstract

This PEP describes an addition to the standard library called Data Classes. Although they use a very different mechanism, Data Classes can be thought of as “mutable namedtuples with defaults”. Because Data Classes use normal class definition syntax, you are free to use inheritance, metaclasses, docstrings, user-defined methods, class factories, and other Python class features.

A class decorator is provided which inspects a class definition for variables with type annotations as defined in PEP 526, “Syntax for Variable Annotations”. In this document, such variables are called fields. Using these fields, the decorator adds generated method definitions to the class to support instance initialization, a repr, comparison methods, and optionally other methods as described in the Specification section. Such a class is called a Data Class, but there’s really nothing special about the class: the decorator adds generated methods to the class and returns the same class it was given.

As an example:

@dataclass class InventoryItem: '''Class for keeping track of an item in inventory.''' name: str unit_price: float quantity_on_hand: int = 0 def total_cost(self) -> float: return self.unit_price * self.quantity_on_hand

The @dataclass decorator will add the equivalent of these methods to the InventoryItem class:

def __init__(self, name: str, unit_price: float, quantity_on_hand: int = 0) -> None: self.name = name self.unit_price = unit_price self.quantity_on_hand = quantity_on_hand def __repr__(self): return f'InventoryItem(name={self.name!r}, unit_price={self.unit_price!r}, quantity_on_hand={self.quantity_on_hand!r})' def __eq__(self, other): if other.__class__ is self.__class__: return (self.name, self.unit_price, self.quantity_on_hand) == (other.name, other.unit_price, other.quantity_on_hand) return NotImplemented def __ne__(self, other): if other.__class__ is self.__class__: return (self.name, self.unit_price, self.quantity_on_hand) != (other.name, other.unit_price, other.quantity_on_hand) return NotImplemented def __lt__(self, other): if other.__class__ is self.__class__: return (self.name, self.unit_price, self.quantity_on_hand) < (other.name, other.unit_price, other.quantity_on_hand) return NotImplemented def __le__(self, other): if other.__class__ is self.__class__: return (self.name, self.unit_price, self.quantity_on_hand) <= (other.name, other.unit_price, other.quantity_on_hand) return NotImplemented def __gt__(self, other): if other.__class__ is self.__class__: return (self.name, self.unit_price, self.quantity_on_hand) > (other.name, other.unit_price, other.quantity_on_hand) return NotImplemented def __ge__(self, other): if other.__class__ is self.__class__: return (self.name, self.unit_price, self.quantity_on_hand) >= (other.name, other.unit_price, other.quantity_on_hand) return NotImplemented

Data Classes save you from writing and maintaining these methods.

Rationale

There have been numerous attempts to define classes which exist primarily to store values which are accessible by attribute lookup. Some examples include:

So, why is this PEP needed?

With the addition of PEP 526, Python has a concise way to specify the type of class members. This PEP leverages that syntax to provide a simple, unobtrusive way to describe Data Classes. With two exceptions, the specified attribute type annotation is completely ignored by Data Classes.

No base classes or metaclasses are used by Data Classes. Users of these classes are free to use inheritance and metaclasses without any interference from Data Classes. The decorated classes are truly “normal” Python classes. The Data Class decorator should not interfere with any usage of the class.

One main design goal of Data Classes is to support static type checkers. The use of PEP 526 syntax is one example of this, but so is the design of the fields() function and the @dataclass decorator. Due to their very dynamic nature, some of the libraries mentioned above are difficult to use with static type checkers.

Data Classes are not, and are not intended to be, a replacement mechanism for all of the above libraries. But being in the standard library will allow many of the simpler use cases to instead leverage Data Classes. Many of the libraries listed have different feature sets, and will of course continue to exist and prosper.

Where is it not appropriate to use Data Classes?

Specification

All of the functions described in this PEP will live in a module named dataclasses.

A function dataclass which is typically used as a class decorator is provided to post-process classes and add generated methods, described below.

The dataclass decorator examines the class to find fields. A field is defined as any variable identified in __annotations__. That is, a variable that has a type annotation. With two exceptions described below, none of the Data Class machinery examines the type specified in the annotation.

Note that __annotations__ is guaranteed to be an ordered mapping, in class declaration order. The order of the fields in all of the generated methods is the order in which they appear in the class.

The dataclass decorator will add various “dunder” methods to the class, described below. If any of the added methods already exist on the class, a TypeError will be raised. The decorator returns the same class that is called on: no new class is created.

The dataclass decorator is typically used with no parameters and no parentheses. However, it also supports the following logical signature:

def dataclass(*, init=True, repr=True, eq=True, order=False, unsafe_hash=False, frozen=False)

If dataclass is used just as a simple decorator with no parameters, it acts as if it has the default values documented in this signature. That is, these three uses of @dataclass are equivalent:

@dataclass class C: ... @dataclass() class C: ... @dataclass(init=True, repr=True, eq=True, order=False, unsafe_hash=False, frozen=False) class C: ...

The parameters to dataclass are:

fields may optionally specify a default value, using normal Python syntax:

@dataclass class C: a: int # 'a' has no default value b: int = 0 # assign a default value for 'b'

In this example, both a and b will be included in the added __init__ method, which will be defined as:

def __init__(self, a: int, b: int = 0):

TypeError will be raised if a field without a default value follows a field with a default value. This is true either when this occurs in a single class, or as a result of class inheritance.

For common and simple use cases, no other functionality is required. There are, however, some Data Class features that require additional per-field information. To satisfy this need for additional information, you can replace the default field value with a call to the provided field() function. The signature of field() is:

def field(*, default=MISSING, default_factory=MISSING, repr=True, hash=None, init=True, compare=True, metadata=None)

The MISSING value is a sentinel object used to detect if the default and default_factory parameters are provided. This sentinel is used because None is a valid value for default.

The parameters to field() are:

If the default value of a field is specified by a call to field(), then the class attribute for this field will be replaced by the specified default value. If no default is provided, then the class attribute will be deleted. The intent is that after the dataclass decorator runs, the class attributes will all contain the default values for the fields, just as if the default value itself were specified. For example, after:

@dataclass class C: x: int y: int = field(repr=False) z: int = field(repr=False, default=10) t: int = 20

The class attribute C.z will be 10, the class attribute C.t will be 20, and the class attributes C.x and C.y will not be set.

Field objects

Field objects describe each defined field. These objects are created internally, and are returned by the fields() module-level method (see below). Users should never instantiate a Field object directly. Its documented attributes are:

Other attributes may exist, but they are private and must not be inspected or relied on.

post-init processing

The generated __init__ code will call a method named __post_init__, if it is defined on the class. It will be called as self.__post_init__(). If no __init__ method is generated, then __post_init__ will not automatically be called.

Among other uses, this allows for initializing field values that depend on one or more other fields. For example:

@dataclass class C: a: float b: float c: float = field(init=False) def __post_init__(self): self.c = self.a + self.b

See the section below on init-only variables for ways to pass parameters to __post_init__(). Also see the warning about how replace() handles init=False fields.

Class variables

One place where dataclass actually inspects the type of a field is to determine if a field is a class variable as defined in PEP 526. It does this by checking if the type of the field is typing.ClassVar. If a field is a ClassVar, it is excluded from consideration as a field and is ignored by the Data Class mechanisms. For more discussion, see [8]. Such ClassVar pseudo-fields are not returned by the module-level fields() function.

Init-only variables

The other place where dataclass inspects a type annotation is to determine if a field is an init-only variable. It does this by seeing if the type of a field is of type dataclasses.InitVar. If a field is an InitVar, it is considered a pseudo-field called an init-only field. As it is not a true field, it is not returned by the module-level fields() function. Init-only fields are added as parameters to the generated __init__ method, and are passed to the optional __post_init__ method. They are not otherwise used by Data Classes.

For example, suppose a field will be initialized from a database, if a value is not provided when creating the class:

@dataclass class C: i: int j: int = None database: InitVar[DatabaseType] = None def __post_init__(self, database): if self.j is None and database is not None: self.j = database.lookup('j') c = C(10, database=my_database)

In this case, fields() will return Field objects for i and j, but not for database.

Frozen instances

It is not possible to create truly immutable Python objects. However, by passing frozen=True to the @dataclass decorator you can emulate immutability. In that case, Data Classes will add __setattr__ and __delattr__ methods to the class. These methods will raise a FrozenInstanceError when invoked.

There is a tiny performance penalty when using frozen=True: __init__ cannot use simple assignment to initialize fields, and must use object.__setattr__.

Inheritance

When the Data Class is being created by the @dataclass decorator, it looks through all of the class’s base classes in reverse MRO (that is, starting at object) and, for each Data Class that it finds, adds the fields from that base class to an ordered mapping of fields. After all of the base class fields are added, it adds its own fields to the ordered mapping. All of the generated methods will use this combined, calculated ordered mapping of fields. Because the fields are in insertion order, derived classes override base classes. An example:

@dataclass class Base: x: Any = 15.0 y: int = 0 @dataclass class C(Base): z: int = 10 x: int = 15

The final list of fields is, in order, x, y, z. The final type of x is int, as specified in class C.

The generated __init__ method for C will look like:

def __init__(self, x: int = 15, y: int = 0, z: int = 10):

Default factory functions

If a field specifies a default_factory, it is called with zero arguments when a default value for the field is needed. For example, to create a new instance of a list, use:

l: list = field(default_factory=list)

If a field is excluded from __init__ (using init=False) and the field also specifies default_factory, then the default factory function will always be called from the generated __init__ function. This happens because there is no other way to give the field an initial value.

Mutable default values

Python stores default member variable values in class attributes. Consider this example, not using Data Classes:

class C: x = [] def add(self, element): self.x += element o1 = C() o2 = C() o1.add(1) o2.add(2) assert o1.x == [1, 2] assert o1.x is o2.x

Note that the two instances of class C share the same class variable x, as expected.

Using Data Classes, if this code was valid:

@dataclass class D: x: List = [] def add(self, element): self.x += element

it would generate code similar to:

class D: x = [] def __init__(self, x=x): self.x = x def add(self, element): self.x += element assert D().x is D().x

This has the same issue as the original example using class C. That is, two instances of class D that do not specify a value for x when creating a class instance will share the same copy of x. Because Data Classes just use normal Python class creation they also share this problem. There is no general way for Data Classes to detect this condition. Instead, Data Classes will raise a TypeError if it detects a default parameter of type list, dict, or set. This is a partial solution, but it does protect against many common errors. See Automatically support mutable default values in the Rejected Ideas section for more details.

Using default factory functions is a way to create new instances of mutable types as default values for fields:

@dataclass class D: x: list = field(default_factory=list) assert D().x is not D().x

Module level helper functions

Discussion

python-ideas discussion

This discussion started on python-ideas [9] and was moved to a GitHub repo [10] for further discussion. As part of this discussion, we made the decision to use PEP 526 syntax to drive the discovery of fields.

Support for automatically setting __slots__?

At least for the initial release, __slots__ will not be supported. __slots__ needs to be added at class creation time. The Data Class decorator is called after the class is created, so in order to add __slots__ the decorator would have to create a new class, set __slots__, and return it. Because this behavior is somewhat surprising, the initial version of Data Classes will not support automatically setting __slots__. There are a number of workarounds:

For more discussion, see [11].

Why not just use namedtuple?

Why not just use typing.NamedTuple?

For classes with statically defined fields, it does support similar syntax to Data Classes, using type annotations. This produces a namedtuple, so it shares namedtuples benefits and some of its downsides. Data Classes, unlike typing.NamedTuple, support combining fields via inheritance.

Why not just use attrs?

For more discussion, see [12].

post-init parameters

In an earlier version of this PEP before InitVar was added, the post-init function __post_init__ never took any parameters.

The normal way of doing parameterized initialization (and not just with Data Classes) is to provide an alternate classmethod constructor. For example:

@dataclass class C: x: int @classmethod def from_file(cls, filename): with open(filename) as fl: file_value = int(fl.read()) return C(file_value) c = C.from_file('file.txt')

Because the __post_init__ function is the last thing called in the generated __init__, having a classmethod constructor (which can also execute code immediately after constructing the object) is functionally equivalent to being able to pass parameters to a __post_init__ function.

With InitVars, __post_init__ functions can now take parameters. They are passed first to __init__ which passes them to __post_init__ where user code can use them as needed.

The only real difference between alternate classmethod constructors and InitVar pseudo-fields is in regards to required non-field parameters during object creation. With InitVars, using __init__ and the module-level replace() function InitVars must always be specified. Consider the case where a context object is needed to create an instance, but isn’t stored as a field. With alternate classmethod constructors the context parameter is always optional, because you could still create the object by going through __init__ (unless you suppress its creation). Which approach is more appropriate will be application-specific, but both approaches are supported.

Another reason for using InitVar fields is that the class author can control the order of __init__ parameters. This is especially important with regular fields and InitVar fields that have default values, as all fields with defaults must come after all fields without defaults. A previous design had all init-only fields coming after regular fields. This meant that if any field had a default value, then all init-only fields would have to have defaults values, too.

asdict and astuple function names

The names of the module-level helper functions asdict() and astuple() are arguably not PEP 8 compliant, and should be as_dict() and as_tuple(), respectively. However, after discussion [13] it was decided to keep consistency with namedtuple._asdict() and attr.asdict().

Rejected ideas

Copying init=False fields after new object creation in replace()

Fields that are init=False are by definition not passed to __init__, but instead are initialized with a default value, or by calling a default factory function in __init__, or by code in __post_init__.

A previous version of this PEP specified that init=False fields would be copied from the source object to the newly created object after __init__ returned, but that was deemed to be inconsistent with using __init__ and __post_init__ to initialize the new object. For example, consider this case:

@dataclass class Square: length: float area: float = field(init=False, default=0.0) def __post_init__(self): self.area = self.length * self.length s1 = Square(1.0) s2 = replace(s1, length=2.0)

If init=False fields were copied from the source to the destination object after __post_init__ is run, then s2 would end up begin Square(length=2.0, area=1.0), instead of the correct Square(length=2.0, area=4.0).

Automatically support mutable default values

One proposal was to automatically copy defaults, so that if a literal list [] was a default value, each instance would get a new list. There were undesirable side effects of this decision, so the final decision is to disallow the 3 known built-in mutable types: list, dict, and set. For a complete discussion of this and other options, see [14].

Examples

Custom __init__ method

Sometimes the generated __init__ method does not suffice. For example, suppose you wanted to have an object to store *args and **kwargs:

@dataclass(init=False) class ArgHolder: args: List[Any] kwargs: Mapping[Any, Any] def __init__(self, *args, **kwargs): self.args = args self.kwargs = kwargs a = ArgHolder(1, 2, three=3)

A complicated example

This code exists in a closed source project:

class Application: def __init__(self, name, requirements, constraints=None, path='', executable_links=None, executables_dir=()): self.name = name self.requirements = requirements self.constraints = {} if constraints is None else constraints self.path = path self.executable_links = [] if executable_links is None else executable_links self.executables_dir = executables_dir self.additional_items = [] def __repr__(self): return f'Application({self.name!r},{self.requirements!r},{self.constraints!r},{self.path!r},{self.executable_links!r},{self.executables_dir!r},{self.additional_items!r})'

This can be replaced by:

@dataclass class Application: name: str requirements: List[Requirement] constraints: Dict[str, str] = field(default_factory=dict) path: str = '' executable_links: List[str] = field(default_factory=list) executable_dir: Tuple[str] = () additional_items: List[str] = field(init=False, default_factory=list)

The Data Class version is more declarative, has less code, supports typing, and includes the other generated functions.

Acknowledgements

The following people provided invaluable input during the development of this PEP and code: Ivan Levkivskyi, Guido van Rossum, Hynek Schlawack, Raymond Hettinger, and Lisa Roach. I thank them for their time and expertise.

A special mention must be made about the attrs project. It was a true inspiration for this PEP, and I respect the design decisions they made.

References

[1] attrs project on github (https://github.com/python-attrs/attrs) [2] George Sakkis’ recordType recipe (http://code.activestate.com/recipes/576555-records/) [3] DictDotLookup recipe (http://code.activestate.com/recipes/576586-dot-style-nested-lookups-over-dictionary-based-dat/) [4] attrdict package (https://pypi.python.org/pypi/attrdict) [5] StackOverflow question about data container classes (https://stackoverflow.com/questions/3357581/using-python-class-as-a-data-container) [6] David Beazley metaclass talk featuring data classes (https://www.youtube.com/watch?v=sPiWg5jSoZI) [7] Python documentation for __hash__ (https://docs.python.org/3/reference/datamodel.html#object.__hash__) [8] ClassVar discussion in PEP 526 [9] Start of python-ideas discussion (https://mail.python.org/pipermail/python-ideas/2017-May/045618.html) [10] GitHub repo where discussions and initial development took place (https://github.com/ericvsmith/dataclasses) [11] Support __slots__? (https://github.com/ericvsmith/dataclasses/issues/28) [12] why not just attrs? (https://github.com/ericvsmith/dataclasses/issues/19) [13] PEP 8 names for asdict and astuple (https://github.com/ericvsmith/dataclasses/issues/110) [14] Copying mutable defaults (https://github.com/ericvsmith/dataclasses/issues/3)

Copyright

This document has been placed in the public domain.

Contents


Page Source (GitHub)

Source: https://github.com/python/peps/blob/main/peps/pep-0557.rst

Last modified: 2025-02-01 08:55:40 UTC