|
57 | 57 | CACHE_RELAY = "relay" |
58 | 58 | CACHE_RELAY_INIT = "relay_init" |
59 | 59 | CACHE_RELAY_LOCK = "relay_lock" |
| 60 | +FEATURES_CIRCLE = ( |
| 61 | + NodeFeature.CIRCLE, |
| 62 | + NodeFeature.RELAY, |
| 63 | + NodeFeature.RELAY_INIT, |
| 64 | + NodeFeature.RELAY_LOCK, |
| 65 | + NodeFeature.ENERGY, |
| 66 | + NodeFeature.POWER, |
| 67 | +) |
60 | 68 |
|
61 | 69 | FuncT = TypeVar("FuncT", bound=Callable[..., Any]) |
62 | 70 | _LOGGER = logging.getLogger(__name__) |
@@ -884,17 +892,7 @@ async def load(self) -> bool: |
884 | 892 | _LOGGER.debug("Loading Circle node %s from cache", self._mac_in_str) |
885 | 893 | if await self._load_from_cache(): |
886 | 894 | self._loaded = True |
887 | | - self._setup_protocol( |
888 | | - CIRCLE_FIRMWARE_SUPPORT, |
889 | | - ( |
890 | | - NodeFeature.CIRCLE, |
891 | | - NodeFeature.RELAY, |
892 | | - NodeFeature.RELAY_INIT, |
893 | | - NodeFeature.RELAY_LOCK, |
894 | | - NodeFeature.ENERGY, |
895 | | - NodeFeature.POWER, |
896 | | - ), |
897 | | - ) |
| 895 | + self._setup_protocol(CIRCLE_FIRMWARE_SUPPORT, FEATURES_CIRCLE) |
898 | 896 | if await self.initialize(): |
899 | 897 | await self._loaded_callback(NodeEvent.LOADED, self.mac) |
900 | 898 | return True |
@@ -926,17 +924,7 @@ async def load(self) -> bool: |
926 | 924 | return False |
927 | 925 |
|
928 | 926 | self._loaded = True |
929 | | - self._setup_protocol( |
930 | | - CIRCLE_FIRMWARE_SUPPORT, |
931 | | - ( |
932 | | - NodeFeature.CIRCLE, |
933 | | - NodeFeature.RELAY, |
934 | | - NodeFeature.RELAY_INIT, |
935 | | - NodeFeature.RELAY_LOCK, |
936 | | - NodeFeature.ENERGY, |
937 | | - NodeFeature.POWER, |
938 | | - ), |
939 | | - ) |
| 927 | + self._setup_protocol(CIRCLE_FIRMWARE_SUPPORT, FEATURES_CIRCLE) |
940 | 928 | if not await self.initialize(): |
941 | 929 | return False |
942 | 930 |
|
|
0 commit comments