Python 3.7: Initialize objects with dataclasses module?

Question

Here is the my code in python 3.6

class A(object)

    def __init__(self, a: str):
        self._int_a: int = int(a)  # desired composition

    def get_int_a(self) -> int:
        return self._int_a

I want to rewrite this code in python 3.7, how can i initialize self._int_a: int = int(a) with dataclasses module?

I know that i can do something like that but i can't get how to initialize _a: int = int(a) or similar to that.

from dataclasses import dataclass


@dataclass
class A(object):
    _a: int = int(a)  # i want to get initialized `int` object here

def get_int_a(self) -> int:
    return self._a

Thanks in advance for your ideas and suggestions.

Patrick Haugh · Accepted Answer · 2018-07-12 20:33:09Z

21

Do away with getters and setters entirely and just use attribute access. We can define an init only field that accepts a string and then convert that string to an integer field in our __post_init__ call.

from dataclasses import dataclass, InitVar, field

@dataclass
class A:
    temp: InitVar[str]
    a: int = field(init=False)
    def __post_init__(self, temp):
        self.a = int(temp)

x = A("1")
print(type(x.a), x.a)
# <class 'int'> 1

edited Jul 12, 2018 at 20:33

answered Jul 12, 2018 at 19:32

Patrick Haugh

61.3k13 gold badges94 silver badges101 bronze badges

Sign up to request clarification or add additional context in comments.

8 Comments

Cedric H. Over a year ago

What I can't understand with the mechanism is that in the end, temp remains as a field for the final object... The dataclass mechanism should remove it completely from the fields (as is done when called the module-level fields method, but not done in __dataclass_fields__).

Patrick Haugh Over a year ago

I'm not sure I understand what you're saying. x.temp should lead to an AttributeError

Cedric H. Over a year ago

Indeed, I was confused, it leads to an AttributeError as expected.

Cedric H. Over a year ago

OK, now I see what was the issue: imagine the same example but now give a default value to temp in the class definition. Now you don't have an AttributeError, which is weird.

Jules Gagnon-Marchand Over a year ago

ok but this breaks keyword init

|

Collectives™ on Stack Overflow

Python 3.7: Initialize objects with dataclasses module?

1 Answer 1

8 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

8 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related