1212
1313from ..api import NodeEvent , NodeType , PlugwiseNode , StickEvent
1414from ..connection import StickController
15- from ..constants import ENERGY_NODE_TYPES , UTF8
15+ from ..constants import (
16+ ENERGY_NODE_TYPES ,
17+ NODE_RETRY_DISCOVER_INTERVAL ,
18+ NODE_RETRY_LOAD_INTERVAL ,
19+ UTF8 ,
20+ )
1621from ..exceptions import CacheError , MessageError , NodeError , StickError , StickTimeout
1722from ..helpers .util import validate_mac
1823from ..messages .requests import CircleMeasureIntervalRequest , NodePingRequest
@@ -445,14 +450,11 @@ async def _discover_registered_nodes(self) -> None:
445450 self ._registry_stragglers [address ] = mac
446451 registered_counter += 1
447452 await sleep (0 )
448- if len (self ._registry_stragglers ) > 0 :
449- if (
450- self ._discover_stragglers_task is None
451- or self ._discover_stragglers_task .done ()
452- ):
453- self ._discover_stragglers_task = create_task (
454- self ._discover_stragglers ()
455- )
453+ if len (self ._registry_stragglers ) > 0 and (
454+ self ._discover_stragglers_task is None
455+ or self ._discover_stragglers_task .done ()
456+ ):
457+ self ._discover_stragglers_task = create_task (self ._discover_stragglers ())
456458 _LOGGER .debug (
457459 "Total %s online of %s registered node(s)" ,
458460 str (len (self ._nodes )),
@@ -462,7 +464,7 @@ async def _discover_registered_nodes(self) -> None:
462464 async def _discover_stragglers (self ) -> None :
463465 """Repeat Discovery of Nodes with unknown NodeType."""
464466 while len (self ._registry_stragglers ) > 0 :
465- await sleep (60 )
467+ await sleep (NODE_RETRY_DISCOVER_INTERVAL )
466468 stragglers : dict [int , str ] = {}
467469 for address , mac in self ._registry_stragglers .items ():
468470 if not await self ._discover_node (address , mac , None ):
@@ -486,9 +488,9 @@ async def _load_node(self, mac: str) -> bool:
486488
487489 async def _load_stragglers (self ) -> None :
488490 """Retry failed load operation."""
489- await sleep (60 )
491+ await sleep (NODE_RETRY_LOAD_INTERVAL )
490492 while not self ._load_discovered_nodes ():
491- await sleep (60 )
493+ await sleep (NODE_RETRY_LOAD_INTERVAL )
492494
493495 async def _load_discovered_nodes (self ) -> bool :
494496 """Load all nodes currently discovered."""
0 commit comments