import sys from functools import wraps from clcommon.const import Feature from clcommon.cpapi import is_panel_feature_supported from clcommon.lib.cledition import ( CLEditionDetectionError, ) __all__ = ( "skip_without_lve", "lve_supported_or_exit", ) _SKIP_MESSAGE = ( "Command is skipped as LVE feature is not supported in current environment\n" "Refer to https://docs.cloudlinux.com/solo/introduction/#cloudlinux-os-editions-comparison for more info" ) def skip_without_lve(): try: # we still have some utils that could be run # under user (e.g cloudlinux-selector) if not is_panel_feature_supported(Feature.LVE): print(_SKIP_MESSAGE) sys.exit(0) except CLEditionDetectionError as e: print(f"Error: {e}") sys.exit(1) def lve_supported_or_exit(f): @wraps(f) def inner(*args, **kwargs): if not is_panel_feature_supported(Feature.LVE): print(_SKIP_MESSAGE) else: return f(*args, **kwargs) return inner