Display a data waiting message only if any tracked account is present

There was an error when attempting to enter operations without tracked accounts, the message Waiting for data... was displayed, which was incorrect.

clive/__private/ui/clive_screen.py

@@ -48,59 +48,36 @@ class CliveScreen(Screen[ScreenResultType], CliveWidget):
         """Provides the ability to control the binding order in the footer."""
         return self._sort_bindings(super().active_bindings)
 
-    @staticmethod
-    def _common_decorator_prevent_decorator(
-        availability_check_func: Callable[[Clive], bool],
-        logger_message: str,
-        notify_message: str,
-    ) -> Callable[[Callable[P, None]], Callable[P, None]]:
-        def decorator(func: Callable[P, None]) -> Callable[P, None]:
-            @wraps(func)
-            def wrapper(*args: P.args, **kwargs: P.kwargs) -> None:
-                app_ = get_clive().app_instance()
-                if not availability_check_func(app_):
-                    logger.debug(f"action {func.__name__} {logger_message}")
-                    app_.notify(notify_message, severity="warning")
-                    return
-                func(*args, **kwargs)
-
-            return wrapper
-
-        return decorator
-
-    @staticmethod
-    def prevent_action_when_no_accounts_node_data(func: Callable[P, None]) -> Callable[P, None]:
-        def check_data_available(app_: Clive) -> bool:
+    @classmethod
+    def prevent_action_when_no_accounts_node_data(cls, func: Callable[P, None]) -> Callable[P, None]:
+        def check_data_available(app: Clive) -> bool:
             return (
-                app_.world.profile.accounts.is_tracked_accounts_node_data_available
-                or app_.world.profile.accounts.is_tracked_accounts_alarms_data_available
+                app.world.profile.accounts.is_tracked_accounts_node_data_available
+                or app.world.profile.accounts.is_tracked_accounts_alarms_data_available
             )
 
-        return CliveScreen._common_decorator_prevent_decorator(
+        return cls._common_prevent_decorator(
             check_data_available,
-            "prevented because no data is available",
             "Waiting for data...",
         )(func)
 
-    @staticmethod
-    def prevent_action_when_no_working_account(func: Callable[P, None]) -> Callable[P, None]:
-        def check_data_available(app_: Clive) -> bool:
-            return app_.world.profile.accounts.has_working_account
+    @classmethod
+    def prevent_action_when_no_working_account(cls, func: Callable[P, None]) -> Callable[P, None]:
+        def check_data_available(app: Clive) -> bool:
+            return app.world.profile.accounts.has_working_account
 
-        return CliveScreen._common_decorator_prevent_decorator(
+        return cls._common_prevent_decorator(
             check_data_available,
-            "prevented because no working account is set",
             "Cannot perform this action without working account",
         )(func)
 
-    @staticmethod
-    def prevent_action_when_no_tracked_accounts(func: Callable[P, None]) -> Callable[P, None]:
-        def check_data_available(app_: Clive) -> bool:
-            return app_.world.profile.accounts.has_tracked_accounts
+    @classmethod
+    def prevent_action_when_no_tracked_accounts(cls, func: Callable[P, None]) -> Callable[P, None]:
+        def check_data_available(app: Clive) -> bool:
+            return app.world.profile.accounts.has_tracked_accounts
 
-        return CliveScreen._common_decorator_prevent_decorator(
+        return cls._common_prevent_decorator(
             check_data_available,
-            "prevented because no tracked accounts are present",
             "Cannot perform this action without any tracked accounts",
         )(func)
 
@@ -127,6 +104,26 @@ class CliveScreen(Screen[ScreenResultType], CliveWidget):
 
         return wrapper
 
+    @classmethod
+    def _common_prevent_decorator(
+        cls,
+        availability_check_func: Callable[[Clive], bool],
+        notify_message: str,
+    ) -> Callable[[Callable[P, None]], Callable[P, None]]:
+        def decorator(func: Callable[P, None]) -> Callable[P, None]:
+            @wraps(func)
+            def wrapper(*args: P.args, **kwargs: P.kwargs) -> None:
+                app_ = get_clive().app_instance()
+                if not availability_check_func(app_):
+                    logger.debug(f"action {func.__name__} {notify_message}")
+                    app_.notify(notify_message, severity="warning")
+                    return
+                func(*args, **kwargs)
+
+            return wrapper
+
+        return decorator
+
     @on(ScreenSuspend)
     def _post_suspended(self) -> None:
         """