improved zabbix inventory. added sys.exit(1) where needed, added some exception handling.

This commit is contained in:
René Moser 2013-08-22 22:55:48 +02:00
parent 500e6fa374
commit 308026f234
2 changed files with 31 additions and 11 deletions

View file

@ -4,8 +4,8 @@
[zabbix]
# Server location
server = http://192.168.0.1/zabbix
server = http://zabbix.example.com/zabbix
# Login
username =
password =
username = admin
password = zabbix

View file

@ -1,4 +1,4 @@
#!/usr/bin/python
#!/usr/bin/env python
# (c) 2013, Greg Buehler
#
@ -20,16 +20,26 @@
######################################################################
"""
Zabbix external inventory script. Returns hosts and hostgroups from Zabbix.
Zabbix Server external inventory script.
========================================
Returns hosts and hostgroups from Zabbix Server.
Configuration is read from `zabbix.ini`.
Tested with Zabbix Server 2.0.6.
"""
import os, sys
import json
import argparse
import ConfigParser
try:
from zabbix_api import ZabbixAPI
except:
print "Error: Zabbix API library must be installed: pip install zabbix-api."
sys.exit(1)
try:
import json
@ -97,8 +107,12 @@ class ZabbixInventory(object):
self.read_cli()
if self.zabbix_server and self.zabbix_username:
try:
api = ZabbixAPI(server=self.zabbix_server)
api.login(user=self.zabbix_username, password=self.zabbix_password)
except BaseException, e:
print "Error: Could not login to Zabbix server. Check your zabbix.ini."
sys.exit(1)
if self.options.host:
data = self.get_host(api, self.options.host)
@ -107,7 +121,13 @@ class ZabbixInventory(object):
elif self.options.list:
data = self.get_list(api)
print json.dumps(data, indent=2)
else:
print "Configuration of server and credentials is required"
print "usage: --list ..OR.. --host <hostname>"
sys.exit(1)
else:
print "Error: Configuration of server and credentials are required. See zabbix.ini."
sys.exit(1)
ZabbixInventory()