ok
Mini Shell
# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
from __future__ import absolute_import
from distutils.util import strtobool
import lvestats.lib.commons.logsetup as logsetup
import os
import configparser
from clcommon import clconfpars
GLOBAL_CONFIG_LOCATION = '/etc/sysconfig/lvestats2'
CL_CONFIG = '/etc/sysconfig/cloudlinux'
CONFIG_PATH_DIR = '/etc/sysconfig/lvestats.config/'
log = logsetup.setup_logging(None)
HIDE_MAX_UID_LVE_PARAMETER = 'hide_lve_more_than_maxuid'
USER_NOTIFICATIONS_OFF_MARKER = 'lvestats_notifications_off'
class ConfigError(Exception):
def __init__(self, location, ex):
self.value = location
self.ex = ex
def log_and_exit(self, log_=None, exit_code=1):
if log_ is None:
log_ = log
log_.error(str(self))
exit(exit_code)
def __str__(self):
if self.ex is IOError:
return "Unable to read config %s (%s)" % (self.value, str(self.ex))
else:
return "Unable to parse config %s (%s)" % (self.value, str(self.ex))
def _load_config_and_process_errors(location):
try:
return clconfpars.load(location)
except IOError as ie:
if ie.errno == 13 and not location.endswith('.readonly'): # permission denied:
return _load_config_and_process_errors(location + '.readonly')
log.error('Config %s not loadable: %s', location, str(ie))
raise ConfigError(location, ie)
except configparser.ParsingError as pe:
log.error(str(pe))
raise ConfigError(location, pe)
def read_plugin_config(config, plugin):
plugin = plugin.strip()
plugin_cfg_path = CONFIG_PATH_DIR + plugin + '.cfg'
if os.path.isfile(plugin_cfg_path):
plugin_config = _load_config_and_process_errors(plugin_cfg_path)
config.update(plugin_config)
return config
def read_config():
"""
reads lvestats config. Throws ConfigError exception -- if any errors
:return: config
"""
return _load_config_and_process_errors(GLOBAL_CONFIG_LOCATION)
def is_normalized_user_cpu():
"""
Check global config parameter for normalize user CPU values
"""
config = configparser.ConfigParser(strict=False)
config.read(CL_CONFIG)
try:
return config.get("license_check", "normalized_user_cpu").upper() == "Y"
except IOError:
# no file
pass
except configparser.NoOptionError:
# no config parameter
pass
except configparser.NoSectionError:
# no section/invalid file (or m.b no file)
pass
return True
def get_max_lve_id() -> int:
"""
The function returns max lve id according to option `use_big_uids`
"""
lvestats_config = read_config()
if bool(strtobool(lvestats_config.get('use_big_uids', 'false'))):
return 2 ** 31 - 1
else:
return 10 ** 9
Zerion Mini Shell 1.0