Merge pull request #224 from bogdant36/CTX-5889-fix-v1

CTX-5921: Added new condition in onBeforeCommandExecute() callback wh…

coretex/cli/main.py

@@ -61,7 +61,7 @@ def update() -> None:
 
 
 @click.group(cls = ClickExceptionInterceptor)
-@utils.onBeforeCommandExecute(utils.checkLibVersion)
+@utils.onBeforeCommandExecute(utils.checkLibVersion, excludeSubcommands = ["update"])
 def cli() -> None:
     pass
 

coretex/cli/modules/utils.py

@@ -150,13 +150,24 @@ def isGPUAvailable() -> bool:
         return False
 
 
-def onBeforeCommandExecute(fun: Callable[..., Any], excludeOptions: Optional[List[str]] = None) -> Any:
+def onBeforeCommandExecute(
+    fun: Callable[..., Any],
+    excludeOptions: Optional[List[str]] = None,
+    excludeSubcommands: Optional[List[str]] = None
+) -> Any:
+
     if excludeOptions is None:
         excludeOptions = []
 
+    if excludeSubcommands is None:
+        excludeSubcommands = []
+
     def decorator(f: Callable[..., Any]) -> Callable[..., Any]:
         @wraps(f)
         def wrapper(*args: Any, **kwargs: Any) -> Any:
+            if click.get_current_context().invoked_subcommand in excludeSubcommands:
+                return f(*args, **kwargs)
+
             for key, value in click.get_current_context().params.items():
                 if key in excludeOptions and value:
                     return f(*args, **kwargs)