def _validate_kwargs(
*,
plugin_name: CheckPluginName,
subscribed_sections: List[ParsedSectionName],
service_name: str,
requires_item: bool,
discovery_function: Callable,
discovery_default_parameters: Optional[Dict],
discovery_ruleset_name: Optional[str],
discovery_ruleset_type: RuleSetType,
check_function: Callable,
check_default_parameters: Optional[Dict],
check_ruleset_name: Optional[str],
cluster_check_function: Optional[Callable],
) -> None:
_validate_service_name(plugin_name, service_name)
# validate discovery arguments
validate_default_parameters(
"discovery",
discovery_ruleset_name,
discovery_default_parameters,
)
validate_ruleset_type(discovery_ruleset_type)
validate_function_arguments(
type_label="discovery",
function=discovery_function,
has_item=False,
default_params=discovery_default_parameters,
sections=subscribed_sections,
)
# validate check arguments
validate_default_parameters(
"check",
check_ruleset_name,
check_default_parameters,
)
validate_function_arguments(
type_label="check",
function=check_function,
has_item=requires_item,
default_params=check_default_parameters,
sections=subscribed_sections,
)
if cluster_check_function is None:
return
validate_function_arguments(
type_label="cluster_check",
function=cluster_check_function,
has_item=requires_item,
default_params=check_default_parameters,
sections=subscribed_sections,
)
def create_inventory_plugin(
*,
name: str,
sections: Optional[List[str]] = None,
inventory_function: Callable,
inventory_default_parameters: Optional[Dict] = None,
inventory_ruleset_name: Optional[str] = None,
module: Optional[str] = None,
) -> InventoryPlugin:
"""Return an InventoryPlugin object after validating and converting the arguments one by one
For a detailed description of the parameters please refer to the exposed function in the
'register' namespace of the API.
"""
plugin_name = InventoryPluginName(name)
subscribed_sections = create_subscribed_sections(sections, plugin_name)
validate_function_arguments(
type_label="inventory",
function=inventory_function,
has_item=False,
default_params=inventory_default_parameters,
sections=subscribed_sections,
)
# validate check arguments
validate_default_parameters(
"inventory",
inventory_ruleset_name,
inventory_default_parameters,
)
return InventoryPlugin(
name=plugin_name,
sections=subscribed_sections,
inventory_function=_filter_inventory(inventory_function),
inventory_default_parameters=inventory_default_parameters or {},
inventory_ruleset_name=(RuleSetName(inventory_ruleset_name)
if inventory_ruleset_name else None),
module=module,
)
def _validate_host_label_kwargs(
*,
host_label_function: HostLabelFunction,
host_label_default_parameters: Optional[Dict[str, Any]],
host_label_ruleset_name: Optional[str],
host_label_ruleset_type: RuleSetType,
) -> None:
validate_ruleset_type(host_label_ruleset_type)
validate_default_parameters(
"host_label",
host_label_ruleset_name,
host_label_default_parameters,
)
validate_function_arguments(
type_label="host_label",
function=host_label_function,
has_item=False,
default_params=host_label_default_parameters,
sections=[ParsedSectionName("__always_just_one_section__")],
)
def create_check_plugin(
*,
name: str,
sections: Optional[List[str]] = None,
service_name: str,
discovery_function: Callable,
discovery_default_parameters: Optional[Dict] = None,
discovery_ruleset_name: Optional[str] = None,
discovery_ruleset_type: DiscoveryRuleSetType = "merged",
check_function: Callable,
check_default_parameters: Optional[Dict] = None,
check_ruleset_name: Optional[str] = None,
cluster_check_function: Optional[Callable] = None,
module: Optional[str] = None,
validate_item: bool = True,
) -> CheckPlugin:
"""Return an CheckPlugin object after validating and converting the arguments one by one
For a detailed description of the parameters please refer to the exposed function in the
'register' namespace of the API.
"""
plugin_name = CheckPluginName(name)
subscribed_sections = create_subscribed_sections(sections, plugin_name)
_validate_service_name(plugin_name, service_name)
requires_item = _requires_item(service_name)
# validate discovery arguments
validate_default_parameters(
"discovery",
discovery_ruleset_name,
discovery_default_parameters,
)
_validate_discovery_ruleset(
discovery_ruleset_name,
discovery_default_parameters,
)
_validate_discovery_ruleset_type(discovery_ruleset_type, )
validate_function_arguments(
type_label="discovery",
function=discovery_function,
has_item=False,
has_params=discovery_ruleset_name is not None,
sections=subscribed_sections,
)
disco_func = _filter_discovery(discovery_function, requires_item,
validate_item)
disco_params = discovery_default_parameters or {}
disco_ruleset_name = RuleSetName(
discovery_ruleset_name) if discovery_ruleset_name else None
# validate check arguments
validate_default_parameters(
"check",
check_ruleset_name,
check_default_parameters,
)
_validate_check_ruleset(
check_ruleset_name,
check_default_parameters,
)
validate_function_arguments(
type_label="check",
function=check_function,
has_item=requires_item,
has_params=check_default_parameters is not None,
sections=subscribed_sections,
)
if cluster_check_function is None:
cluster_check_function = unfit_for_clustering_wrapper(check_function)
else:
validate_function_arguments(
type_label="cluster_check",
function=cluster_check_function,
has_item=requires_item,
has_params=check_ruleset_name is not None,
sections=subscribed_sections,
)
cluster_check_function = _filter_check(cluster_check_function)
return CheckPlugin(
name=plugin_name,
sections=subscribed_sections,
service_name=service_name,
discovery_function=disco_func,
discovery_default_parameters=disco_params,
discovery_ruleset_name=disco_ruleset_name,
discovery_ruleset_type=discovery_ruleset_type,
check_function=_filter_check(check_function),
check_default_parameters=check_default_parameters or {},
check_ruleset_name=RuleSetName(check_ruleset_name)
if check_ruleset_name else None,
cluster_check_function=cluster_check_function,
module=module,
)