У меня есть класс с необязательным атрибутом. Когда я пытаюсь создать экземпляр класса без указания необязательного атрибута, я получаю сообщение об ошибке TypeError: __init__() missing 1 required positional argument: 'time_range'
.
import attr
from typeguard import typechecked
from typing import Optional
@attr.s(auto_attribs=True)
class TimeRange:
start_time: Optional[str] = attr.ib()
end_time: Optional[str] = attr.ib()
@typechecked
def __init__(
self,
start_time: Optional[str] = None,
end_time: Optional[str] = None,
):
self.start_time = start_time
self.end_time = end_time
@attr.s(auto_attribs=True)
class Date:
date: int
time_range: Optional[TimeRange] = attr.ib()
@typechecked
def __init__(
self,
date: int,
time_range: Optional[TimeRange] = None,
):
self.date = date
self.time_range = time_range
# This throws an error
new_date = Date(date=731)
# This also throws an error
new_date = Date(731)
Как я могу создать экземпляр объекта без указания необязательного аргумента?
attr.s
создает функцию __init__
, которая перезаписывает вашу. Чтобы исправить это, просто предоставьте __init__=False
:
import attr
from typeguard import typechecked
from typing import Optional
# note the init=False
@attr.s(auto_attribs=True, init=False)
class TimeRange:
start_time: Optional[str] = attr.ib()
end_time: Optional[str] = attr.ib()
@typechecked
def __init__(
self,
start_time: Optional[str] = None,
end_time: Optional[str] = None,
):
self.start_time = start_time
self.end_time = end_time
@attr.s(auto_attribs=True, init=False)
class Date:
date: int
time_range: Optional[TimeRange] = attr.ib()
@typechecked
def __init__(
self,
date: int,
time_range: Optional[TimeRange] = None,
):
self.date = date
self.time_range = time_range
new_date = Date(date=731)
new_date = Date(731)
Еще проще: вам не нужно ни писать пользовательскую функцию инициализации, ни блокировать attr
, чтобы сгенерировать ее.
Вам просто нужно добавить значение по умолчанию в attr.ib
вызове каждого необязательного свойства (как вы сделали в своей пользовательской функции инициализации).
import attr
from typing import Optional
@attr.s(auto_attribs=True)
class TimeRange:
start_time: Optional[str] = attr.ib(default=None)
end_time: Optional[str] = attr.ib(default=None)
@attr.s(auto_attribs=True)
class Date:
date: int
time_range: Optional[TimeRange] = attr.ib(default=None)
# No more errors
new_date = Date(date=731)
new_date = Date(731)