|
45 | 45 | SENSOR_DIRTY_REPLACE_TIME, |
46 | 46 | SIDE_BRUSH_REPLACE_TIME, |
47 | 47 | ) |
48 | | -from .util import parse_time_to_datetime |
49 | 48 |
|
50 | 49 | _LOGGER = logging.getLogger(__name__) |
51 | 50 |
|
@@ -103,20 +102,19 @@ class RoborockBaseTimer(RoborockBase): |
103 | 102 | end_hour: Optional[int] = None |
104 | 103 | end_minute: Optional[int] = None |
105 | 104 | 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 |
108 | 107 |
|
109 | 108 | 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 |
120 | 118 | ) |
121 | 119 |
|
122 | 120 |
|
|
0 commit comments