Skip to content

Commit a3461dd

Browse files
authored
fix: add datetime_time back (#68)
1 parent 90f905d commit a3461dd

File tree

2 files changed

+12
-14
lines changed

2 files changed

+12
-14
lines changed

roborock/containers.py

Lines changed: 11 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,6 @@
4545
SENSOR_DIRTY_REPLACE_TIME,
4646
SIDE_BRUSH_REPLACE_TIME,
4747
)
48-
from .util import parse_time_to_datetime
4948

5049
_LOGGER = logging.getLogger(__name__)
5150

@@ -103,20 +102,19 @@ class RoborockBaseTimer(RoborockBase):
103102
end_hour: Optional[int] = None
104103
end_minute: Optional[int] = None
105104
enabled: Optional[int] = None
106-
start_time: Optional[datetime.datetime] = None
107-
end_time: Optional[datetime.datetime] = None
105+
start_time: Optional[datetime.time] = None
106+
end_time: Optional[datetime.time] = None
108107

109108
def __post_init__(self) -> None:
110-
self.start_time, self.end_time = (
111-
parse_time_to_datetime(
112-
datetime.time(hour=self.start_hour, minute=self.start_minute),
113-
datetime.time(hour=self.end_hour, minute=self.end_minute),
114-
)
115-
if self.start_hour is not None
116-
and self.start_minute is not None
117-
and self.end_hour is not None
118-
and self.end_minute is not None
119-
else (None, None)
109+
self.start_time = (
110+
datetime.time(hour=self.start_hour, minute=self.start_minute)
111+
if self.start_hour is not None and self.start_minute is not None
112+
else None
113+
)
114+
self.end_time = (
115+
datetime.time(hour=self.end_hour, minute=self.end_minute)
116+
if self.end_hour is not None and self.end_minute is not None
117+
else None
120118
)
121119

122120

tests/test_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import pytest
44

5-
from roborock import parse_time_to_datetime
5+
from roborock.util import parse_time_to_datetime
66

77

88
@pytest.mark.skip

0 commit comments

Comments
 (0)