3131LOGGER = logging .getLogger (__package__ )
3232
3333
34- async def async_setup_entry (
35- hass : HomeAssistant , config_entry : ConfigEntry , async_add_entities
36- ) -> None :
34+ async def async_setup_entry (hass : HomeAssistant , config_entry : ConfigEntry , async_add_entities ) -> None :
3735 cu = hass .data [DOMAIN ][config_entry .entry_id ][DATA_JABLOTRON ]
3836 # how to handle split system?
3937 if cu .mode != JA80CentralUnit .SYSTEM_MODE_SPLIT :
@@ -51,9 +49,7 @@ def check_zone_status(zone: JablotronZone, status: str) -> bool:
5149
5250class Jablotron80AlarmControl (JablotronEntity , AlarmControlPanelEntity ):
5351
54- def __init__ (
55- self , cu : JA80CentralUnit , zones : List [JablotronZone ], main_zone : int = 0
56- ) -> None :
52+ def __init__ (self , cu : JA80CentralUnit , zones : List [JablotronZone ], main_zone : int = 0 ) -> None :
5753 self ._object = zones [main_zone ]
5854 self ._main_zone = main_zone
5955 self ._cu = cu
@@ -75,10 +71,7 @@ def _check_code(code: Optional[str]) -> Optional[str]:
7571 @property
7672 def supported_features (self ) -> int :
7773 if self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_UNSPLIT :
78- return (
79- AlarmControlPanelEntityFeature .ARM_AWAY
80- | AlarmControlPanelEntityFeature .TRIGGER
81- )
74+ return AlarmControlPanelEntityFeature .ARM_AWAY | AlarmControlPanelEntityFeature .TRIGGER
8275 elif self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_PARTIAL :
8376 return (
8477 AlarmControlPanelEntityFeature .ARM_AWAY
@@ -92,10 +85,7 @@ def supported_features(self) -> int:
9285 | AlarmControlPanelEntityFeature .ARM_HOME
9386 | AlarmControlPanelEntityFeature .TRIGGER
9487 )
95- return (
96- AlarmControlPanelEntityFeature .ARM_AWAY
97- | AlarmControlPanelEntityFeature .TRIGGER
98- )
88+ return AlarmControlPanelEntityFeature .ARM_AWAY | AlarmControlPanelEntityFeature .TRIGGER
9989
10090 @property
10191 def code_arm_required (self ) -> bool :
@@ -181,28 +171,16 @@ async def async_alarm_arm_custom_bypass(self, code=None) -> None:
181171 raise NotImplementedError ()
182172
183173 def get_active_zone (self ) -> JablotronZone :
184- if (
185- self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_UNSPLIT
186- and len (self ._zones ) == 1
187- ):
174+ if self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_UNSPLIT and len (self ._zones ) == 1 :
188175 return self ._zones [0 ]
189- elif (
190- self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_PARTIAL
191- and len (self ._zones ) == 3
192- ):
176+ elif self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_PARTIAL and len (self ._zones ) == 3 :
193177 zone_home = self ._zones [0 ]
194178 zone_night = self ._zones [1 ]
195179 zone_away = self ._zones [2 ]
196- for zone in [
197- zone
198- for zone in self ._zones
199- if check_zone_status (zone , JablotronZone .STATUS_ALARM )
200- ]:
180+ for zone in [zone for zone in self ._zones if check_zone_status (zone , JablotronZone .STATUS_ALARM )]:
201181 return zone
202182 for zone in [
203- zone
204- for zone in self ._zones
205- if check_zone_status (zone , JablotronZone .STATUS_ENTRY_DELAY )
183+ zone for zone in self ._zones if check_zone_status (zone , JablotronZone .STATUS_ENTRY_DELAY )
206184 ]:
207185 return zone
208186 if check_zone_status (zone_away , JablotronZone .STATUS_ARMED ):
@@ -212,26 +190,18 @@ def get_active_zone(self) -> JablotronZone:
212190 elif check_zone_status (zone_home , JablotronZone .STATUS_ARMED ):
213191 return zone_home
214192 for zone in [
215- zone
216- for zone in self ._zones
217- if check_zone_status (zone , JablotronZone .STATUS_ARMING )
193+ zone for zone in self ._zones if check_zone_status (zone , JablotronZone .STATUS_ARMING )
218194 ]:
219195 return zone
220196 for zone in [
221- zone
222- for zone in self ._zones
223- if check_zone_status (zone , JablotronZone .STATUS_DISARMED )
197+ zone for zone in self ._zones if check_zone_status (zone , JablotronZone .STATUS_DISARMED )
224198 ]:
225199 return zone
226- elif (
227- self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_SPLIT and len (self ._zones ) == 3
228- ):
200+ elif self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_SPLIT and len (self ._zones ) == 3 :
229201 zone_home = self ._zones [self ._main_zone ]
230202 zone_away = self ._zones [2 ]
231203 for zone in [
232- zone
233- for zone in [zone_home , zone_away ]
234- if check_zone_status (zone , JablotronZone .STATUS_ALARM )
204+ zone for zone in [zone_home , zone_away ] if check_zone_status (zone , JablotronZone .STATUS_ALARM )
235205 ]:
236206 return zone
237207 for zone in [
@@ -269,8 +239,7 @@ def alarm_state(self) -> AlarmControlPanelState | None:
269239 return AlarmControlPanelState .TRIGGERED
270240
271241 elif (
272- zone .status == JablotronZone .STATUS_ARMED
273- and self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_UNSPLIT
242+ zone .status == JablotronZone .STATUS_ARMED and self ._cu .mode == JA80CentralUnit .SYSTEM_MODE_UNSPLIT
274243 ):
275244 return AlarmControlPanelState .ARMED_AWAY
276245
0 commit comments