tyler.durden
/
centreon-plugins
mirror of https://github.com/centreon/centreon-plugins.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

+ Enhance custom mode load: if there is only one custommode we load it (simple)

This commit is contained in:
garnier-quentin 2015-09-18 15:44:17 +02:00
parent c080f651b3
commit 8da8362e1a
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
centreon/plugins/script_custom.pm
View File

@ -66,6 +66,15 @@ sub new {
return $self;
}
sub load_custom_mode {
my ($self, %options) = @_;
$self->is_custommode(custommode => $self->{custommode_name});
centreon::plugins::misc::mymodule_load(output => $self->{output}, module => $self->{custom_modes}{$self->{custommode_name}},
error_msg => "Cannot load module --custommode.");
$self->{custommode_current} = $self->{custom_modes}{$self->{custommode_name}}->new(options => $self->{options}, output => $self->{output}, mode => $self->{custommode_name});
}
sub init {
my ($self, %options) = @_;
# $options{version} = string version
@ -92,10 +101,10 @@ sub init {
$self->{options}->add_help(package => 'centreon::plugins::output', sections => 'OUTPUT OPTIONS');
if (defined($self->{custommode_name}) && $self->{custommode_name} ne '') {
$self->is_custommode(custommode => $self->{custommode_name});
centreon::plugins::misc::mymodule_load(output => $self->{output}, module => $self->{custom_modes}{$self->{custommode_name}},
error_msg => "Cannot load module --custommode.");
$self->{custommode_current} = $self->{custom_modes}{$self->{custommode_name}}->new(options => $self->{options}, output => $self->{output}, mode => $self->{custommode_name});
$self->load_custom_mode();
} elsif (scalar(%{$self->{custom_modes}}) == 1) {
$self->{custommode_name} = keys(%{$self->{custom_modes}})[0];
$self->load_custom_mode();
} else {
$self->{output}->add_option_msg(short_msg => "Need to specify '--custommode'.");
$self->{output}->option_exit();