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
|