Home  >  Article  >  Backend Development  >  Detailed explanation of the dataclass decorator, a new feature of Python 3.7

Detailed explanation of the dataclass decorator, a new feature of Python 3.7

不言
不言Original
2018-04-21 14:44:591749browse

This article mainly introduces you to the relevant information about the dataclass decorator, a new feature of Python 3.7. The article introduces it in great detail through example code. It has certain reference learning value for everyone's study or work. Friends who need it Let’s learn together.

Preface

Python 3.7 will be released this summer, and there will be many new things in Python 3.7:

  • Various character set improvements

  • Delayed evaluation of comments

  • and support for dataclass

One of the most exciting new features is the dataclass decorator.

What is Data Class

Most Python developers have written many classes like the following:

class MyClass:
 def __init__(self, var_a, var_b):
 self.var_a = var_a
 self.var_b = var_b

dataclass can automatically generate methods for simple cases, for example, an __init__ that accepts these parameters and assigns them to itself, the previous small example can be rewritten as:

@dataclass
class MyClass:
 var_a: str
 var_b: str

Let’s take an example to see how to use it

Star Wars API

You can use requests to obtain resources from the Star Wars API:

response = requests.get('https://swapi.co/api/films/1/')
dictionary = response.json()

Let's take a look at the dictionary (simplified) results:

{
 'characters': ['https://swapi.co/api/people/1/',… ],
 'created': '2014-12-10T14:23:31.880000Z',
 'director': 'George Lucas',
 'edited': '2015-04-11T09:46:52.774897Z',
 'episode_id': 4,
 'opening_crawl': 'It is a period of civil war.\r\n … ',
 'planets': ['https://swapi.co/api/planets/2/', … ],
 'producer': 'Gary Kurtz, Rick McCallum',
 'release_date': '1977-05-25',
 'species': ['https://swapi.co/api/species/5/',…],
 'starships': ['https://swapi.co/api/starships/2/',…],
 'title': 'A New Hope',
 'url': 'https://swapi.co/api/films/1/',
 'vehicles': ['https://swapi.co/api/vehicles/4/',…]

Encapsulating API

In order to properly encapsulate an API, we should create a user that can Object used in its application, therefore, in Python 3.6 an object is defined to contain requests responses to /films/endpoint:

class StarWarsMovie:
 def __init__(self,
   title: str,
   episode_id: int,
   opening_crawl: str,
   director: str,
   producer: str,
   release_date: datetime,
   characters: List[str],
   planets: List[str],
   starships: List[str],
   vehicles: List[str],
   species: List[str],
   created: datetime,
   edited: datetime,
   url: str
   ):

 self.title = title
 self.episode_id = episode_id
 self.opening_crawl= opening_crawl
 self.director = director
 self.producer = producer
 self.release_date = release_date
 self.characters = characters
 self.planets = planets
 self.starships = starships
 self.vehicles = vehicles
 self.species = species
 self.created = created
 self.edited = edited
 self.url = url

 if type(self.release_date) is str:
  self.release_date = dateutil.parser.parse(self.release_date)

 if type(self.created) is str:
  self.created = dateutil.parser.parse(self.created)

 if type(self.edited) is str:
  self.edited = dateutil.parser.parse(self.edited)

Careful Reader You may have noticed some duplicate code here.

This is a classic case of using the dataclass decorator, we need to create a class that is primarily used to hold data, with just a little validation, so let's see what we need to modify.

First, the data class automatically generates some dunder methods. If we do not specify any options for the data class decorator, the generated methods are: __init__, __eq__ and __repr__, if you have defined __repr__ But __str__ is not defined, and by default Python (not just data classes) will implement the output __str__ method that returns __repr__. Therefore, the four dunder methods can be implemented by simply changing the code to the following code:

@dataclass
class StarWarsMovie:
 title: str
 episode_id: int
 opening_crawl: str
 director: str
 producer: str
 release_date: datetime
 characters: List[str]
 planets: List[str]
 starships: List[str]
 vehicles: List[str]
 species: List[str]
 created: datetime
 edited: datetime
 url: str

We removed the __init__ method to ensure that the data class decoration The converter can add the corresponding methods it generates. However, we lost some functionality in the process, our Python 3.6 constructor not only defines all the values, but also attempts to parse the date, how can we do this with a data class?

If we want to override __init__, we will lose the advantage of data class, so if we want to handle any additional functionality, we can use the new dunder method: __post_init__, let us see the __post_init__ method for our wrapper class What does it look like:

def __post_init__(self):
 if type(self.release_date) is str:
  self.release_date = dateutil.parser.parse(self.release_date)

 if type(self.created) is str:
  self.created = dateutil.parser.parse(self.created)

 if type(self.edited) is str:
  self.edited = dateutil.parser.parse(self.edited)

That’s it! We can use the data class decorator to implement our class in two-thirds of the code.

More good stuff

The data class can be further customized for the use case by using the decorator's options. The default options are:

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

  • init determines whether to generate the __init__ dunder method

  • repr determines whether to generate the __repr__ dunder method

  • eq does the same for the __eq__ dunder method, which determines the behavior of the equality check (your_class_instance == another_instance)

  • order actually creates There are four dunder methods that determine the behavior of all checks for less than, and/or, and greater than, and if set to true, the list of objects can be sorted.

The last two options determine whether the object can be hashed, which is necessary if you want to use objects of your class as dictionary keys.

For more information, please refer to: PEP 557 -- Data Classes


##

The above is the detailed content of Detailed explanation of the dataclass decorator, a new feature of Python 3.7. For more information, please follow other related articles on the PHP Chinese website!

Statement:
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn