'updateExisting': True
}
}
if os.path.isdir(path):
print 'direcotry detected'
#path = path/*.xml
files = glob.glob(path+'/*.xml')
for file in files:
print(file)
with open(file, 'r') as f:
template = f.read()
try:
rrr=zapi.confimport('xml', template, rules)
print(rrr)
except ZabbixAPIException as e:
print(e)
print('')
elif os.path.isfile(path):
print 'xml input detected'
files = glob.glob(path)
for file in files:
with open(file, 'r') as f:
template = f.read()
try:
zapi.confimport('xml', template, rules)
except ZabbixAPIException as e:
print(e)
else:
class Configurator:
def __init__(self):
self.url = os.environ["ZBX_SERVER_URL"]
self.default_admin_username = "******"
self.default_admin_password = "******"
self.admin_password = os.environ[
"ZBX_ADMIN_PASSWORD"] if "ZBX_ADMIN_PASSWORD" in os.environ else self.default_admin_password
self.guest_username = "******"
self.disable_guest = bool(os.environ["ZBX_DISABLE_GUEST"].lower(
) == "true") if "ZBX_DISABLE_GUEST" in os.environ else false
self.uid = ""
self.hostname = "Zabbix server"
self.agent_dns_name = os.environ["ZBX_AGENT_HOSTNAME"]
self.agent_ip_address = socket.gethostbyname(self.agent_dns_name)
self.default_agent_port = 10050
# If server is unreachable than try to reconnect self.attempts_max_count times after wait timeout
self.connect_attempt_wait_timeout = 5
self.connect_attempts_max_count = 10
# SMTP settings
self.smtp_server = os.environ["SMTP_SERVER"]
self.smtp_email = os.environ["SMTP_EMAIL"]
if check_email(self.smtp_email):
self.smtp_helo = os.environ["SMTP_EMAIL"].split("@")[1]
logger.debug("SMTP_HELO value is: %s." % (self.smtp_helo))
self.admin_email_address = os.environ["ADMIN_EMAIL_ADDRESS"]
self.default_notify_period = "1-7,00:00-24:00"
self.default_severity = int("111100", 2)
self.default_admin_group = "Zabbix administrators"
self.default_user_group = "Operation managers"
self.default_report_action = "Report problems to " + self.default_admin_group
self.configuration_folder = os.environ[
"CONFIGURATION_FOLDER"] if "CONFIGURATION_FOLDER" in os.environ else ""
# Auto registration
self.host_metadata = "Linux " + os.environ[
"DEFAULT_HOST_SECRET"] if "DEFAULT_HOST_SECRET" in os.environ and os.environ[
"DEFAULT_HOST_SECRET"].strip() != "" else ""
# Web scenario list
self.url_list = json.loads(
os.environ["URL_LIST"]) if "URL_LIST" in os.environ and os.environ[
"URL_LIST"].strip() != "" else []
#
self.zapi = ZabbixAPI(self.url)
logger.info("Waiting while Zabbix server will be reachable.")
for attempt in range(0, self.connect_attempts_max_count):
try:
if check_http_code(self.url):
break
except:
logger.debug(
"Waiting for while Zabbix will be reachable %d/%d with %d sec interval."
% (attempt + 1, self.connect_attempts_max_count,
self.connect_attempt_wait_timeout))
time.sleep(self.connect_attempt_wait_timeout)
logger.info("Connecting to Zabbix database.")
try:
self.db = MySQLdb.connect(host=os.environ["DB_SERVER_HOST"],
user=os.environ["MYSQL_USER"],
passwd=os.environ["MYSQL_PASSWORD"],
db=os.environ["MYSQL_DATABASE"])
except:
error("Could not connect to Zabbix database")
self.default_authentication_type = 0
self.authentication_type = self.get_configuration(
)["authentication_type"]
self.configuration = json.loads(
os.environ["ZBX_CONFIG"]
) if "ZBX_CONFIG" in os.environ and os.environ["ZBX_CONFIG"].strip(
) != "" else []
self.admin_users = json.loads(
os.environ["ZBX_ADMIN_USERS"]
) if "ZBX_ADMIN_USERS" in os.environ and os.environ[
"ZBX_ADMIN_USERS"].strip() != "" else []
def login(self):
logger.debug("Login into Zabbix server (%s)." % (self.url))
login_error = False
for attempt in range(0, self.connect_attempts_max_count):
if login_error:
logger.info("Login attemtp %d/%d with %d sec inteval." %
(attempt + 1, self.connect_attempts_max_count,
self.connect_attempt_wait_timeout))
time.sleep(self.connect_attempt_wait_timeout)
try:
self.zapi.login(self.default_admin_username,
self.default_admin_password)
login_error = False
break
except:
try:
self.zapi.login(self.default_admin_username,
self.admin_password)
self.default_admin_password = self.admin_password
login_error = False
break
except:
login_error = True
if login_error:
error("Can not login into Zabbix server after %d attemtps." %
(attempt))
self.uid = self.zapi.user.get(
filter={"alias": self.default_admin_username})[0]["userid"]
def logout(self):
logger.debug("Logout from Zabbix server.")
self.zapi.logout
def relogin(self):
self.logout()
self.login()
def change_default_password(self):
if self.default_admin_password != self.admin_password:
logger.info("Changing the default admin password.")
self.zapi.user.update(userid=self.uid, passwd=self.admin_password)
return 1
else:
logger.info("Skip updating the admin password.")
return 0
def create_group(self, group_name):
group = self.zapi.usergroup.get(filter={"name": group_name})
if len(group) > 0:
logger.debug("Group with name %s is already exist." % (group_name))
return group[0]["usrgrpid"]
else:
logger.debug("Creating group with name: %s." % (group_name))
return self.zapi.usergroup.create({
"name": group_name,
"gui_access": 0,
"users_status": 0
})["usrgrpids"][0]
def add_user_to_group(self, username, group_name):
uid = self.zapi.user.get(filter={"alias": username})[0]["userid"]
logger.debug("Check user is already in group %s." % (group_name))
try:
group = self.zapi.usergroup.get(filter={"name": group_name},
selectUsers=group_name)[0]
except:
error("Can not retrieve information from group: %s." %
(group_name))
users_ids = group["users"]
ids = []
if len(users_ids) > 0:
for element in users_ids:
ids.append(element["userid"])
if not uid in ids:
logger.debug("Adding user %s into group %s." %
(username, group_name))
logger.debug("Current list of users in group: " +
str(ids if len(ids) > 0 else "list is empty") + ".")
ids.append(uid)
self.zapi.usergroup.update(usrgrpid=group["usrgrpid"], userids=ids)
else:
logger.debug("User %s is already included into group %s." %
(username, group_name))
return 0
return 1
def disable_user(self, username):
group_name = "Disabled"
if self.add_user_to_group(username, group_name):
logger.info("Disabled user: %s." % (username))
else:
logger.info("User %s is already disabled." % (username))
def get_host_info(self, hostname="", host_id=""):
logger.debug("Retrieving information about host %s." % (hostname))
rule = {"host": hostname} if hostname != "" else {"hostid": host_id}
return self.zapi.host.get(filter=rule)
def enable_host(self, host_id):
host = self.get_host_info(host_id=host_id)
if host[0]["status"] != "0":
logger.debug("Enabling host with id %s." % (host_id))
self.zapi.host.update(hostid=host_id, status=0)
return 1
logger.debug("Host with id %s is already enabled." % (host_id))
return 0
def update_host_addr(self, host_id, dns="", ip="", use_ip=1):
interface = self.zapi.hostinterface.get(hostids=[host_id])[0]
interface_id = interface["interfaceid"]
if (interface["ip"] != ip) or (interface["dns"] != dns):
logger.debug("Setting new addresses for host with id %s." %
(host_id))
self.zapi.hostinterface.update(interfaceid=interface_id,
ip=ip,
dns=dns,
port=self.default_agent_port,
useip=use_ip)
return 1
else:
return 0
def update_mediatype(self, name, data=dict()):
try:
media = self.zapi.mediatype.get(filter={"description": name})[0]
data["mediatypeid"] = media["mediatypeid"]
logger.debug("Updating media with next data:")
logger.debug(data)
self.zapi.mediatype.update(data)
except:
error("Could not update mediatype with name: %s." % (name))
return 1
def update_user_email_settings(self, username, email):
uid = self.zapi.user.get(filter={"alias": username})[0]["userid"]
medias = []
if email.strip() != "":
logger.debug("Adding new email setting for user %s (uid: %s)." %
(username, uid))
delimiter = ","
for email_address in email.split(delimiter):
if check_email(email_address):
logger.debug("Adding email address: %s" % (email_address))
medias.append({
"mediatypeid": 1,
"sendto": email_address,
"active": 0,
"severity": self.default_severity,
"period": self.default_notify_period
})
logger.debug("Cleanup current media for user.")
self.zapi.user.updatemedia(users=[{"userid": uid}], medias=[])
self.zapi.user.updatemedia(users=[{"userid": uid}], medias=medias)
return 1
return 0
def enable_action(self, name):
try:
logger.debug("Trying to get actionid with name: %s." % (name))
action = self.zapi.action.get(filter={"name": name})[0]
action_id = action["actionid"]
logger.debug("ActionID is: %s" % (action_id))
except:
logger.debug("Could not retrive action_id.")
return 0
if int(action["status"]) != 0:
logger.debug("Action with id %s is disabled, activating it." %
(action_id))
if self.zapi.action.update(actionid=action_id, status=0):
logger.debug("Action was successfuly activated.")
elif int(action["status"]) == 0:
logger.debug("Action is already activated. Skipped.")
return 0
return 1
def add_action(self, data):
# Check if exist current action or no
action = self.zapi.action.get(filter={"name": data["name"]})
if len(action) > 0:
logger.debug("Action is updated with next data:")
# Could not be updated with below element in dict
del data["eventsource"]
data["actionid"] = int(action[0]["actionid"])
logger.debug(data)
return self.zapi.action.update(data)
else:
logger.debug("Action is created.")
return self.zapi.action.create(data)
def add_auto_discovery_action(self, metadata=""):
if metadata != "":
logger.debug("Generating conditions from metadate elements.")
conditions = [{
"conditiontype": 24,
"operator": 2,
"value": data
} for data in metadata.split(" ")]
logger.debug(
"Adding auto registration action for hosts with metadata: %s."
% (metadata if metadata != "" else "not defined"))
data = {
"name":
"Auto registration rules for Linux servers",
"eventsource":
2,
"status":
0,
"operations": [{
"operationtype": 2
}, {
"operationtype":
4,
"opgroup": [
self.zapi.hostgroup.get(
filter={"name": "Linux servers"})[0]["groupid"]
]
}, {
"operationtype":
6,
"optemplate": [{
"templateid":
self.zapi.template.get(
filter={"name": "Template OS Linux"})[0]
["templateid"]
}]
}, {
"operationtype":
0,
"opmessage_grp": [{
"usrgrpid":
self.zapi.usergroup.get(
filter={"name": self.default_admin_group})[0]
["usrgrpid"]
}],
"opmessage": {
"mediatypeid": 0,
"default_msg": 1
}
}],
"def_shortdata":
"Auto registration: {HOST.HOST}",
"def_longdata":
'''
Host name: {HOST.HOST}
Host IP: {HOST.IP}
Agent port: {HOST.PORT}''',
"filter": {
"evaltype": 1,
"conditions": conditions
}
}
return self.add_action(data)
else:
logger.debug(
"Host metadata empty, such action is impossible to add because of security reason."
)
return 0
def add_web_scenario(self, host_id, url_list):
logger.debug("Processing adding urls for monitoring.")
if len(url_list) > 0:
host_name = self.get_host_info(host_id=host_id)[0]["name"]
for item in url_list:
logger.debug(item)
http_test = self.zapi.httptest.get(
filter={"name": item["name"]},
hostids=host_id,
selectSteps="extend")
template = {
"hostid":
host_id,
"name":
item["name"],
"delay":
60,
"retries":
3,
"steps": [{
"no": 1,
"name": "Step #1",
"url": item["url"],
"status_codes": "200",
"follow_redirects": 1,
"retrieve_mode": 0
}]
}
if len(http_test) > 0:
if http_test[0]["name"] == template["name"] and http_test[
0]["steps"][0]["url"] == template["steps"][0][
"url"]:
logger.debug(
"No changed were detected. Skipped updating the web scenario"
)
else:
template["httptestid"] = http_test[0]["httptestid"]
del template["hostid"]
self.zapi.httptest.update(template)
else:
self.zapi.httptest.create(template)
trigger = {
"description":
"Health status of %s" % (item["name"]),
"expression":
"{" + host_name + ":web.test.fail[" + item["name"] +
"].last(0)} <> 0",
"priority":
item["priority"] if "priority" in item else 1,
"url":
item["url"]
}
current_trigger = self.zapi.trigger.get(
filter={"description": trigger["description"]})
if len(current_trigger) > 0:
trigger["triggerid"] = current_trigger[0]["triggerid"]
try:
self.update_trigger(trigger)
except:
error("Can not update the trigger with id %s." %
(trigger["triggerid"]))
else:
try:
self.create_trigger(trigger)
except:
error("Can not create the trigger.")
return 1
else:
return 0
def create_item(self, data):
item = self.zapi.item.get(filter={"name": data["name"]})
if len(item) == 0:
logger.debug("Creating item")
return self.zapi.item.create(data)
else:
data["itemid"] = item[0]["itemid"]
return self.update_item(data)
def update_item(self, data):
logger.debug("Updating item")
return self.zapi.item.update(data)
def create_trigger(self, data):
data["type"] = 0
trigger = self.zapi.trigger.get(
filter={"description": data["description"]})
if len(trigger) == 0:
logger.debug("Creating trigger.")
logger.debug(data)
return self.zapi.trigger.create(data)
else:
data["triggerid"] = trigger[0]["triggerid"]
return self.update_trigger(data)
def update_trigger(self, data):
logger.debug("Updating trigger.")
logger.debug(data)
return self.zapi.trigger.update(data)
def get_configuration(self):
cur = self.db.cursor()
cur.execute("SELECT * FROM config")
field = [i[0] for i in cur.description]
result = dict()
for i, row in enumerate(cur.fetchone()):
result[field[i]] = row
cur.close()
return result
def update_configuration(self, config=dict()):
current = self.get_configuration()
logger.debug("Comparasing desired config with current.")
if len(config) > 0:
for key, value in config.items():
if current[key] == value:
del config[key]
if len(config) > 0:
logger.debug("Updating configuration.")
logger.debug(config)
query = "UPDATE config SET "
i = 0
length = len(config)
for key, value in config.iteritems():
i += 1
if isinstance(value, int):
query += "%s=%d" % (key, value)
else:
query += "%s='%s'" % (key, value)
if i < length:
query += ","
logger.debug(query)
cur = self.db.cursor()
cur.execute(query)
self.db.commit()
cur.close()
return 1
return 0
def add_user(self, user=dict(), groups=list(), user_type=1):
# Check if such user is already exist in database
u = self.zapi.user.get(filter={"alias": user["name"]})
if len(u) == 0:
logger.debug("Such user does not exist. Adding.")
self.zapi.user.create({
"alias": user["name"],
"passwd": user["password"],
"usrgrps": groups,
"type": user_type
})
return 1
return 0
def import_configuration(self):
configuration_format = "xml"
files_list = []
rules = {
"applications": {
"createMissing": True
},
"discoveryRules": {
"createMissing": True,
"updateExisting": True
},
"graphs": {
"createMissing": True,
"updateExisting": True
},
"groups": {
"createMissing": True
},
"hosts": {
"createMissing": True,
"updateExisting": True
},
"images": {
"createMissing": True,
"updateExisting": True
},
"items": {
"createMissing": True,
"updateExisting": True
},
"maps": {
"createMissing": True,
"updateExisting": True
},
"screens": {
"createMissing": True,
"updateExisting": True
},
"templateLinkage": {
"createMissing": True
},
"templates": {
"createMissing": True,
"updateExisting": True
},
"templateScreens": {
"createMissing": True,
"updateExisting": True
},
"triggers": {
"createMissing": True,
"updateExisting": True
},
}
logger.debug("Finding configuration in folder: %s." %
(self.configuration_folder))
if os.access(self.configuration_folder, os.R_OK):
templates_count = 0
for filename in os.listdir(self.configuration_folder):
if filename.endswith(configuration_format):
files_list.append(filename)
with open(self.configuration_folder + "/" + filename) as f:
source = f.read()
try:
self.zapi.confimport(configuration_format, source,
rules)
except ZabbixAPIException as e:
error(e)
logger.debug("Configuration %s was imported/updated." %
(filename))
templates_count += 1
if templates_count == 0:
logger.debug("No configuration was found.")
return 0
logger.debug("Was found next list of configratuin templates:")
logger.debug(files_list)
else:
error("Could not access to configuration folder.")
return 1
def add_ssl_certificates_checking(self, host_id, url_list):
distinct_list = list()
for el in url_list:
if el["url"].split("://")[0] == "https":
server_name = el["url"].split("/")[2]
if not server_name in distinct_list:
distinct_list.append(server_name)
logger.debug("Was found next list of vhosts with https mode.")
logger.debug(distinct_list)
host_info = self.get_host_info(host_id=host_id)[0]
interface_id = self.zapi.hostinterface.get(
filter={"hostid": host_id})[0]["interfaceid"]
for vhost in distinct_list:
logger.debug("Creating certificate check item for vhost: %s." %
(vhost))
item = {
"name": "Certificate expiration timestamp of %s" % (vhost),
"hostid": host_id,
"interfaceid": interface_id,
"key_": "certificate.endtimestamp[%s,443]" % (vhost),
"delay": "1h",
"type": 0,
"value_type": 3,
"history": "30d",
"trends": "60d"
}
self.create_item(item)
item = {
"name": "Certificate expiration date of %s" % (vhost),
"hostid": host_id,
"interfaceid": interface_id,
"key_": "certificate.enddate[%s,443]" % (vhost),
"delay": "1h",
"type": 0,
"value_type": 4,
"history": "1d",
"trends": "1d"
}
self.create_item(item)
logger.debug("Creating/updating warning and alert trigger.")
for el in [{
"days": 30,
"severity": 1
}, {
"days": 7,
"severity": 4
}]:
trigger = {
"description":
"SSL ceritificate of %s will expire in %d days" %
(vhost, el["days"]),
"expression":
"{%s:certificate.endtimestamp[%s,443].last()}-{%s:certificate.endtimestamp[%s,443].now()}<%d"
% (host_info["host"], vhost, host_info["host"], vhost,
el["days"] * 24 * 60),
"priority":
el["severity"],
"url":
"https://" + vhost
}
self.create_trigger(trigger)
return 1
def main(self):
if self.authentication_type != self.default_authentication_type:
logger.debug(
"Changing authentication_type to default to use api with basic credentials."
)
self.update_configuration(
config={
"authentication_type": self.default_authentication_type
})
self.login()
if self.change_default_password():
self.relogin()
if self.disable_guest:
self.disable_user(self.guest_username)
host_id = self.get_host_info(hostname=self.hostname)[0]["hostid"]
logger.debug("%s has such id: %s." % (self.hostname, host_id))
logger.info("Associate local agent with %s." %
(self.hostname) if self.update_host_addr(
host_id, self.agent_dns_name, self.agent_ip_address, 0
) else "Skipped address updating %s." % (self.hostname))
logger.info("Enabled %s host." % (self.hostname) if self.
enable_host(host_id) else "Skipped enabling %s." %
(self.hostname))
logger.info(
"Configured default email media type." if self.
update_mediatype(name="Email",
data={
"smtp_server": self.smtp_server,
"smtp_email": self.smtp_email,
"smtp_helo": self.smtp_helo
}) else "Skipped configuring default media type.")
logger.info(
"Updated %s user email settings." %
(self.default_admin_username) if self.update_user_email_settings(
username=self.default_admin_username,
email=self.admin_email_address
) else "Skipped updating email settings")
logger.info("Enabled default notify action." if self.
enable_action(self.default_report_action) else
"Skipped activating the default notify action.")
logger.info("Added/Updated auto discovery action." if self.
add_auto_discovery_action(self.host_metadata) else
"Skipped adding auto discovery action.")
logger.info("Initialization checking web urls." if self.
add_web_scenario(host_id=host_id, url_list=self.url_list)
else "Skipped initialization of web urls.")
logger.info("Created SSL certificates check trigggers." if self.
add_ssl_certificates_checking(
host_id=host_id, url_list=self.url_list
) else "Skipped adding SSL check triggers.")
logger.info("Adding zabbix configuration templates." if self.
configuration_folder != "" and self.import_configuration()
else "No configuration templates folder was identified.")
logger.info("Creating default user group %s." %
(self.default_user_group))
group_id = self.create_group(self.default_user_group)
if len(self.admin_users) > 0:
for user in self.admin_users:
logger.info("Adding user %s as administrator." %
(user["name"]))
if not self.add_user(
user=user, groups=[{
"usrgrpid": group_id
}], user_type=3):
logger.info("Skipped.")
logger.info("Updating Zabbix configuration" if self.
update_configuration(self.configuration
) else "Nothing to update. Skipped.")
if self.authentication_type != self.default_authentication_type:
logger.debug("Returned authentication_type to initial state.")
self.update_configuration(
config={"authentication_type": self.authentication_type})
return self.logout()
zapi.confimport(format='xml',
source=templatexml,
rules={
'applications': {
'createMissing': 'true'
},
'discoveryRules': {
'createMissing': 'true'
},
'graphs': {
'createMissing': 'true'
},
'groups': {
'createMissing': 'true'
},
'hosts': {
'createMissing': 'true'
},
'images': {
'createMissing': 'true'
},
'items': {
'createMissing': 'true'
},
'maps': {
'createMissing': 'true'
},
'screens': {
'createMissing': 'true'
},
'templates': {
'createMissing': 'true'
},
'templateLinkage': {
'createMissing': 'true'
},
'templateScreens': {
'createMissing': 'true'
},
'triggers': {
'createMissing': 'true'
}
})
'templateLinkage': {
'createMissing': True,
},
'templates': {
'createMissing': True,
'updateExisting': True
},
'templateScreens': {
'createMissing': True,
'updateExisting': True
},
'triggers': {
'createMissing': True,
'updateExisting': True
},
}
print("Importing Zabbix config from %s" % config_path)
with open(config_path, 'r') as f:
config = f.read()
try:
# https://github.com/lukecyca/pyzabbix/issues/62
import_result = zapi.confimport("xml", config, import_rules)
print(import_result)
except ZabbixAPIException as e:
print e
for h in zapi.host.get(output="extend"):
print(h['name'])
},
'valueMaps': {
'createMissing': True,
'updateExisting': True
},
}
if os.path.isdir(path):
#path = path/*.xml
files = glob.glob(path+'/*.xml')
for file in files:
print(file)
with open(file, 'r') as f:
template = f.read()
try:
zapi.confimport('xml', template, rules)
except ZabbixAPIException as e:
print(e)
print('')
elif os.path.isfile(path):
files = glob.glob(path)
for file in files:
with open(file, 'r') as f:
template = f.read()
try:
zapi.confimport('xml', template, rules)
except ZabbixAPIException as e:
print(e)
else:
print('I need a xml file')
"createMissing": 'true'
},
"templateScreens": {
"createMissing": 'true',
"updateExisting": 'true'
}
}
with open(nodefile, 'r') as f:
nodes = f.readlines()
rootzh = ZabbixAPI(rootUrl)
rootzh.login(user=rootUser, password=rootPwd)
templates = rootzh.template.get(output=['templateid'])
tlist = []
for t in templates:
tlist.append(t['templateid'])
exports = rootzh.configuration.export(format="json",
options={"templates": tlist})
for node in nodes:
(name, url, usr, pwd) = node.rstrip('\n').split(";")
nzh = ZabbixAPI(rootUrl)
nzh.login(user=rootUser, password=rootPwd)
result = nzh.confimport(format="json", rules=importRules, source=exports)
print(result)
nzh.user.logout()
print(url + " " + usr)
except OSError as e:
print e
sys.exit(1)
for i in templateFiles:
try:
f = open("/tmp/zabbix_templates/"+i,'r')
except IOError as e:
print e
sys.exit(1)
templatexml = f.read()
f.close()
try:
zapi.confimport(format='xml',source=templatexml,rules={'applications':{'createMissing':'true'}, 'discoveryRules':{'createMissing':'true'}, 'graphs':{'createMissing':'true'},'groups':{'createMissing':'true'}, 'hosts':{'createMissing':'true'}, 'images':{'createMissing':'true'}, 'items':{'createMissing':'true'},'maps':{'createMissing':'true'}, 'screens':{'createMissing':'true'}, 'templates':{'createMissing':'true'}, 'templateLinkage':{'createMissing':'true'}, 'templateScreens':{'createMissing':'true'}, 'triggers':{'createMissing':'true'} })
except ZabbixAPIException as e:
print(e)
sys.exit(1)
# Get host group IDs
hostgroups = zapi.hostgroup.get(output='extend')
# build hostgroups dictionary
for t in hostgroups:
gd[t['name']] = t['groupid']
# get template IDs
templates = zapi.template.get(output='extend')
# Login to the Zabbix API
zapi.login("admin", "zabbix")
rules = {
"applications": {"createMissing": "true", "updateExisting": "true"},
"discoveryRules": {"createMissing": "true", "updateExisting": "true"},
"graphs": {"createMissing": "true", "updateExisting": "true"},
"groups": {"createMissing": "true"},
"hosts": {"createMissing": "true", "updateExisting": "true"},
"images": {"createMissing": "true", "updateExisting": "true"},
"items": {"createMissing": "true", "updateExisting": "true"},
"maps": {"createMissing": "true", "updateExisting": "true"},
"screens": {"createMissing": "true", "updateExisting": "true"},
"templateLinkage": {"createMissing": "true", "updateExisting": "true"},
"templates": {"createMissing": "true", "updateExisting": "true"},
"templateScreens": {"createMissing": "true", "updateExisting": "true"},
"triggers": {"createMissing": "true", "updateExisting": "true"},
}
path = "./templates/*.xml"
files = glob.glob(path)
for file in files:
with open(file, "r") as f:
template = f.read()
try:
zapi.confimport("xml", template, rules)
except ZabbixAPIException as e:
print(e)
class Configurator:
def __init__(self):
self.grafana_password = os.environ["GF_SECURITY_ADMIN_PASSWORD"]
self.grafana_datas_yaml = os.environ["GRA_DSOURCE_YAML"]
self.grafana_dashb_yaml = os.environ["GRA_DBOARD_YAML"]
self.grafana_hostname = os.environ["GRA_HOST"]
self.grafana_pt_dashboards = os.environ["GRA_PATH_TO_DASHBOARDS"]
self.url = os.environ["ZBX_SERVER_URL"]
self.server_host = os.environ["ZBX_SERVER_HOST"]
self.default_admin_username = "******"
self.default_admin_password = "******"
self.admin_password = os.environ[
"ZBX_ADMIN_PASSWORD"] if "ZBX_ADMIN_PASSWORD" in os.environ else self.default_admin_password
self.guest_username = "******"
self.disable_guest = bool(os.environ["ZBX_DISABLE_GUEST"].lower(
) == "true") if "ZBX_DISABLE_GUEST" in os.environ else false
self.uid = ""
self.hostname = "Zabbix server"
self.agent_dns_name = os.environ["ZBX_AGENT_HOSTNAME"]
self.agent_ip_address = socket.gethostbyname(self.agent_dns_name)
self.default_agent_port = 10050
self.default_server_port = 10051
# If server is unreachable than try to reconnect self.attempts_max_count times after wait timeout
self.connect_attempt_wait_timeout = 5
self.connect_attempts_max_count = 30
# SMTP settings
self.smtp_server = os.environ["SMTP_SERVER"]
self.smtp_email = os.environ["SMTP_EMAIL"]
if check_email(self.smtp_email):
self.smtp_helo = os.environ["SMTP_EMAIL"].split("@")[1]
logger.debug("SMTP_HELO value is: %s." % (self.smtp_helo))
self.admin_email_address = os.environ["ADMIN_EMAIL_ADDRESS"]
self.default_notify_period = "1-7,00:00-24:00"
self.default_severity = int("111100", 2)
self.default_admin_group = "Zabbix administrators"
self.default_user_group = "Operation managers"
self.default_report_action = "Report problems to " + self.default_admin_group
self.configuration_folder = os.environ[
"CONFIGURATION_FOLDER"] if "CONFIGURATION_FOLDER" in os.environ else ""
self.zabbix_config_folder = os.environ[
"ZBX_CONFIG_FOLDER"] if "ZBX_CONFIG_FOLDER" in os.environ else "/etc/zabbix"
# This custom config for zabbix agent will be used for auto discovery purposes, for automatic cleaning data, if it is required
# Custom config structure: examples/custom.json.example
self.zabbix_custom_config = self.zabbix_config_folder + "/" + os.environ[
"ZBX_CUSTOM_CONFIG"] if "ZBX_CUSTOM_CONFIG" in os.environ else "custom.json"
self.custom_config_json = dict()
# Auto registration
self.host_metadata = "Linux " + os.environ[
"DEFAULT_HOST_SECRET"] if "DEFAULT_HOST_SECRET" in os.environ and os.environ[
"DEFAULT_HOST_SECRET"].strip() != "" else ""
# Web scenario list
self.url_list = json.loads(
os.environ["URL_LIST"]) if "URL_LIST" in os.environ and os.environ[
"URL_LIST"].strip() != "" else []
#
self.zapi = ZabbixAPI(self.url)
logger.info("Waiting while Zabbix server will be reachable.")
for attempt in range(0, self.connect_attempts_max_count):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
result = sock.connect_ex(
(self.server_host, self.default_server_port))
if result == 0:
logger.debug("Server is reachable.")
break
else:
logger.debug(
"Waiting for while Zabbix will be reachable %d/%d with %d sec interval."
% (attempt + 1, self.connect_attempts_max_count,
self.connect_attempt_wait_timeout))
time.sleep(self.connect_attempt_wait_timeout)
logger.info("Connecting to Zabbix database.")
try:
self.db = MySQLdb.connect(host=os.environ["DB_SERVER_HOST"],
user=os.environ["MYSQL_USER"],
passwd=os.environ["MYSQL_PASSWORD"],
db=os.environ["MYSQL_DATABASE"])
except:
error("Could not connect to Zabbix database")
self.default_authentication_type = 0
self.authentication_type = self.get_configuration(
)["authentication_type"]
self.configuration = json.loads(
os.environ["ZBX_CONFIG"]
) if "ZBX_CONFIG" in os.environ and os.environ["ZBX_CONFIG"].strip(
) != "" else []
self.admin_users = json.loads(
os.environ["ZBX_ADMIN_USERS"]
) if "ZBX_ADMIN_USERS" in os.environ and os.environ[
"ZBX_ADMIN_USERS"].strip() != "" else []
self.additional_templates = [
x.strip()
for x in os.environ["ZBX_ADDITIONAL_TEMPLATES"].split(",")
] if "ZBX_ADDITIONAL_TEMPLATES" in os.environ else []
def grafana_plugin_on(self):
logger.debug("Grafana Plugin Curl Start")
plugURL = (
"{0}:3000/api/plugins/alexanderzobnin-zabbix-app/settings?enabled=true"
.format(self.grafana_hostname))
c = pycurl.Curl()
c.setopt(c.USERPWD, "%s:%s" % ("admin", self.grafana_password))
c.setopt(c.URL, plugURL)
c.setopt(c.POSTFIELDS, '{ ' ' }')
c.setopt(c.VERBOSE, True)
c.perform()
def grafana_dashboard_starred(self):
logger.debug("Grafana Dashboard Starred")
path, dirs, files = os.walk(self.grafana_pt_dashboards).next()
file_count = len(files)
print(file_count)
dashboard_id = 1
#Condition for find more then one dashboard.json file (need for add to favorite)
while dashboard_id <= file_count:
#print (dashboard_id)
plugURL = ("{0}:3000/api/user/stars/dashboard/{1}".format(
self.grafana_hostname, dashboard_id))
c = pycurl.Curl()
c.setopt(c.USERPWD, "%s:%s" % ("admin", self.grafana_password))
c.setopt(c.URL, plugURL)
c.setopt(c.POSTFIELDS, '{ ' ' }')
c.setopt(c.VERBOSE, True)
c.perform()
dashboard_id = dashboard_id + 1 #So simply :)
def grafana_configurator(self):
#Datasource.yaml Template generation
with open(self.configuration_folder + "/datasource.yaml.tmpl",
'r') as myfile:
data = myfile.read().format(self.url, self.admin_password)
ds_file = open(self.grafana_datas_yaml, "w")
ds_file.write(data)
ds_file.close()
#Dashboard.yaml Template generation
with open(self.configuration_folder + "/dashboard.yaml.tmpl",
'r') as myfile:
data = myfile.read()
db_file = open(self.grafana_dashb_yaml, "w")
db_file.write(data)
db_file.close()
#Grafana Port Checker (need for check, if server start?)
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
timeconn = 1
timeconn_timeout = 10
while True:
time.sleep(1)
if timeconn == timeconn_timeout:
break
result = sock.connect_ex((self.grafana_hostname, 3000))
if result == 0:
logger.debug("Grafana Server is Started")
self.grafana_plugin_on() #Enable Plugin
self.grafana_dashboard_starred(
) #Add all dashboards to favorite
break
else:
logger.debug("Grafana Server still not start, Try...")
timeconn = timeconn + 1
sock.close()
def login(self):
logger.debug("Login into Zabbix server (%s)." % (self.url))
login_error = False
for attempt in range(0, self.connect_attempts_max_count):
if login_error:
logger.info("Login attemtp %d/%d with %d sec inteval." %
(attempt + 1, self.connect_attempts_max_count,
self.connect_attempt_wait_timeout))
time.sleep(self.connect_attempt_wait_timeout)
try:
self.zapi.login(self.default_admin_username,
self.default_admin_password)
login_error = False
break
except:
try:
self.zapi.login(self.default_admin_username,
self.admin_password)
self.default_admin_password = self.admin_password
login_error = False
break
except:
login_error = True
if login_error:
error("Can not login into Zabbix server after %d attemtps." %
(attempt))
self.uid = self.zapi.user.get(
filter={"alias": self.default_admin_username})[0]["userid"]
def logout(self):
logger.debug("Logout from Zabbix server.")
self.zapi.logout
def relogin(self):
self.logout()
self.login()
def change_default_password(self):
if self.default_admin_password != self.admin_password:
logger.info("Changing the default admin password.")
self.zapi.user.update(userid=self.uid, passwd=self.admin_password)
return 1
else:
logger.info("Skip updating the admin password.")
return 0
def create_group(self, group_name):
group = self.zapi.usergroup.get(filter={"name": group_name})
if len(group) > 0:
logger.debug("Group with name %s is already exist." % (group_name))
return group[0]["usrgrpid"]
else:
logger.debug("Creating group with name: %s." % (group_name))
return self.zapi.usergroup.create({
"name": group_name,
"gui_access": 0,
"users_status": 0
})["usrgrpids"][0]
def add_user_to_group(self, username, group_name):
uid = self.zapi.user.get(filter={"alias": username})[0]["userid"]
logger.debug("Check user is already in group %s." % (group_name))
try:
group = self.zapi.usergroup.get(filter={"name": group_name},
selectUsers=group_name)[0]
except:
error("Can not retrieve information from group: %s." %
(group_name))
users_ids = group["users"]
ids = []
if len(users_ids) > 0:
for element in users_ids:
ids.append(element["userid"])
if not uid in ids:
logger.debug("Adding user %s into group %s." %
(username, group_name))
logger.debug("Current list of users in group: " +
str(ids if len(ids) > 0 else "list is empty") + ".")
ids.append(uid)
self.zapi.usergroup.update(usrgrpid=group["usrgrpid"], userids=ids)
else:
logger.debug("User %s is already included into group %s." %
(username, group_name))
return 0
return 1
def disable_user(self, username):
group_name = "Disabled"
if self.add_user_to_group(username, group_name):
logger.info("Disabled user: %s." % (username))
else:
logger.info("User %s is already disabled." % (username))
def get_host_info(self, hostname="", host_id=""):
logger.debug("Retrieving information about host %s." % (hostname))
rule = {"host": hostname} if hostname != "" else {"hostid": host_id}
return self.zapi.host.get(filter=rule, selectParentTemplates=[])
def enable_host(self, host_id):
host = self.get_host_info(host_id=host_id)
if host[0]["status"] != "0":
logger.debug("Enabling host with id %s." % (host_id))
self.zapi.host.update(hostid=host_id, status=0)
return 1
logger.debug("Host with id %s is already enabled." % (host_id))
return 0
def update_host_addr(self, host_id, dns="", ip="", use_ip=1):
interface = self.zapi.hostinterface.get(hostids=[host_id])[0]
interface_id = interface["interfaceid"]
if (interface["ip"] != ip) or (interface["dns"] != dns):
logger.debug("Setting new addresses for host with id %s." %
(host_id))
self.zapi.hostinterface.update(interfaceid=interface_id,
ip=ip,
dns=dns,
port=self.default_agent_port,
useip=use_ip)
return 1
else:
return 0
def update_mediatype(self, name, data=dict()):
try:
media = self.zapi.mediatype.get(filter={"description": name})[0]
data["mediatypeid"] = media["mediatypeid"]
logger.debug("Updating media with next data:")
logger.debug(data)
self.zapi.mediatype.update(data)
except:
error("Could not update mediatype with name: %s." % (name))
return 1
def update_user_email_settings(self, username, email):
uid = self.zapi.user.get(filter={"alias": username})[0]["userid"]
medias = []
if email.strip() != "":
logger.debug("Adding new email setting for user %s (uid: %s)." %
(username, uid))
delimiter = ","
for email_address in email.split(delimiter):
if check_email(email_address):
logger.debug("Adding email address: %s" % (email_address))
medias.append({
"mediatypeid": 1,
"sendto": email_address,
"active": 0,
"severity": self.default_severity,
"period": self.default_notify_period
})
logger.debug("Cleanup current media for user.")
self.zapi.user.updatemedia(users=[{"userid": uid}], medias=[])
self.zapi.user.updatemedia(users=[{"userid": uid}], medias=medias)
return 1
return 0
def enable_action(self, name):
try:
logger.debug("Trying to get actionid with name: %s." % (name))
action = self.zapi.action.get(filter={"name": name})[0]
action_id = action["actionid"]
logger.debug("ActionID is: %s" % (action_id))
except:
logger.debug("Could not retrive action_id.")
return 0
if int(action["status"]) != 0:
logger.debug("Action with id %s is disabled, activating it." %
(action_id))
if self.zapi.action.update(actionid=action_id, status=0):
logger.debug("Action was successfuly activated.")
elif int(action["status"]) == 0:
logger.debug("Action is already activated. Skipped.")
return 0
return 1
def add_action(self, data):
# Check if exist current action or no
action = self.zapi.action.get(filter={"name": data["name"]})
if len(action) > 0:
logger.debug("Action is updated with next data:")
# Could not be updated with below element in dict
del data["eventsource"]
data["actionid"] = int(action[0]["actionid"])
logger.debug(data)
return self.zapi.action.update(data)
else:
logger.debug("Action is created.")
return self.zapi.action.create(data)
def add_auto_discovery_action(self, metadata=""):
if metadata != "":
logger.debug("Generating conditions from metadate elements.")
conditions = [{
"conditiontype": 24,
"operator": 2,
"value": data
} for data in metadata.split(" ")]
logger.debug(
"Adding auto registration action for hosts with metadata: %s."
% (metadata if metadata != "" else "not defined"))
data = {
"name":
"Auto registration rules for Linux servers",
"eventsource":
2,
"status":
0,
"operations": [{
"operationtype": 2
}, {
"operationtype":
4,
"opgroup": [
self.zapi.hostgroup.get(
filter={"name": "Linux servers"})[0]["groupid"]
]
}, {
"operationtype":
6,
"optemplate": [{
"templateid":
self.zapi.template.get(
filter={"name": "Template OS Linux"})[0]
["templateid"]
}]
}, {
"operationtype":
0,
"opmessage_grp": [{
"usrgrpid":
self.zapi.usergroup.get(
filter={"name": self.default_admin_group})[0]
["usrgrpid"]
}],
"opmessage": {
"mediatypeid": 0,
"default_msg": 1
}
}],
"def_shortdata":
"Auto registration: {HOST.HOST}",
"def_longdata":
'''
Host name: {HOST.HOST}
Host IP: {HOST.IP}
Agent port: {HOST.PORT}''',
"filter": {
"evaltype": 1,
"conditions": conditions
}
}
return self.add_action(data)
else:
logger.debug(
"Host metadata empty, such action is impossible to add because of security reason."
)
return 0
def add_web_scenario(self, host_id, url_list):
logger.debug("Processing adding urls for monitoring.")
self.custom_config_json["web"] = list()
if len(url_list) > 0:
host_name = self.get_host_info(host_id=host_id)[0]["name"]
for item in url_list:
logger.debug(item)
priority = item["priority"] if "priority" in item else 1
http_test = self.zapi.httptest.get(
filter={"name": item["name"]},
hostids=host_id,
selectSteps="extend")
self.custom_config_json["web"].append({
"name": item["name"],
"url": item["url"],
"priority": priority
})
template = {
"hostid":
host_id,
"name":
item["name"],
"delay":
60,
"retries":
3,
"steps": [{
"no": 1,
"name": "Step #1",
"url": item["url"],
"status_codes": "200",
"follow_redirects": 1,
"retrieve_mode": 0
}]
}
if len(http_test) > 0:
if http_test[0]["name"] == template["name"] and http_test[
0]["steps"][0]["url"] == template["steps"][0][
"url"]:
logger.debug(
"No changed were detected. Skipped updating the web scenario."
)
else:
template["httptestid"] = http_test[0]["httptestid"]
del template["hostid"]
self.zapi.httptest.update(template)
else:
self.zapi.httptest.create(template)
trigger = {
"description":
"Health status of %s" % (item["name"]),
"expression":
"{" + host_name + ":web.test.fail[" + item["name"] +
"].last(0)} <> 0",
"priority":
priority,
"url":
item["url"]
}
current_trigger = self.zapi.trigger.get(
filter={"description": trigger["description"]})
if len(current_trigger) > 0:
trigger["triggerid"] = current_trigger[0]["triggerid"]
try:
self.update_trigger(trigger)
except:
error("Can not update the trigger with id %s." %
(trigger["triggerid"]))
else:
try:
self.create_trigger(trigger)
except:
error("Can not create the trigger.")
return 1
else:
return 0
def cleanup_undefined_web_scenario(self, host_id, url_list):
http_test = self.zapi.httptest.get(hostids=host_id)
cleanup = False
for item in http_test:
key = next((index for (index, d) in enumerate(url_list)
if d["name"] == item["name"]), None)
if (key is None):
logger.debug("Removing web check for item with name: %s." %
(item["name"]))
self.zapi.httptest.delete(int(item["httptestid"]))
cleanup = True
if (cleanup):
return 1
return 0
def create_item(self, data):
item = self.zapi.item.get(filter={"name": data["name"]})
if len(item) == 0:
logger.debug("Creating item")
return self.zapi.item.create(data)
else:
data["itemid"] = item[0]["itemid"]
return self.update_item(data)
def update_item(self, data):
logger.debug("Updating item")
return self.zapi.item.update(data)
def create_trigger(self, data):
data["type"] = 0
trigger = self.zapi.trigger.get(
filter={"description": data["description"]})
if len(trigger) == 0:
logger.debug("Creating trigger.")
logger.debug(data)
return self.zapi.trigger.create(data)
else:
data["triggerid"] = trigger[0]["triggerid"]
return self.update_trigger(data)
def update_trigger(self, data):
logger.debug("Updating trigger.")
logger.debug(data)
return self.zapi.trigger.update(data)
def get_configuration(self):
cur = self.db.cursor()
cur.execute("SELECT * FROM config")
field = [i[0] for i in cur.description]
result = dict()
for i, row in enumerate(cur.fetchone()):
result[field[i]] = row
cur.close()
return result
def update_configuration(self, config=dict()):
current = self.get_configuration()
logger.debug("Comparasing desired config with current.")
if len(config) > 0:
for key, value in config.items():
if current[key] == value:
del config[key]
if len(config) > 0:
logger.debug("Updating configuration.")
logger.debug(config)
query = "UPDATE config SET "
i = 0
length = len(config)
for key, value in config.iteritems():
i += 1
if isinstance(value, int):
query += "%s=%d" % (key, value)
else:
query += "%s='%s'" % (key, value)
if i < length:
query += ","
logger.debug(query)
cur = self.db.cursor()
cur.execute(query)
self.db.commit()
cur.close()
return 1
return 0
def add_user(self, user=dict(), groups=list(), user_type=1):
# Check if such user is already exist in database
u = self.zapi.user.get(filter={"alias": user["name"]})
if len(u) == 0:
logger.debug("Such user does not exist. Adding.")
self.zapi.user.create({
"alias": user["name"],
"passwd": user["password"],
"usrgrps": groups,
"type": user_type
})
return 1
return 0
def import_configuration(self):
configuration_format = "xml"
files_list = []
rules = {
"applications": {
"createMissing": True
},
"discoveryRules": {
"createMissing": True,
"updateExisting": True
},
"graphs": {
"createMissing": True,
"updateExisting": True
},
"groups": {
"createMissing": True
},
"hosts": {
"createMissing": True,
"updateExisting": True
},
"images": {
"createMissing": True,
"updateExisting": True
},
"items": {
"createMissing": True,
"updateExisting": True
},
"maps": {
"createMissing": True,
"updateExisting": True
},
"screens": {
"createMissing": True,
"updateExisting": True
},
"templateLinkage": {
"createMissing": True
},
"templates": {
"createMissing": True,
"updateExisting": True
},
"templateScreens": {
"createMissing": True,
"updateExisting": True
},
"triggers": {
"createMissing": True,
"updateExisting": True
},
}
logger.debug("Finding configuration in folder: %s." %
(self.configuration_folder))
if os.access(self.configuration_folder, os.R_OK):
templates_count = 0
for filename in os.listdir(self.configuration_folder):
if filename.endswith(configuration_format):
files_list.append(filename)
with open(self.configuration_folder + "/" + filename) as f:
source = f.read()
try:
self.zapi.confimport(configuration_format, source,
rules)
except ZabbixAPIException as e:
error(e)
logger.debug("Configuration %s was imported/updated." %
(filename))
templates_count += 1
if templates_count == 0:
logger.debug("No configuration was found.")
return 0
logger.debug("Was found next list of configratuin templates:")
logger.debug(files_list)
else:
error("Could not access to configuration folder.")
return 1
def save_json_config(self, source_json_object=dict(), target_file=""):
logger.debug("Saving json object into %s" % target_file)
if (len(source_json_object) == 0):
logger.debug("No elements were found in json object.")
return 0
logger.debug(source_json_object)
try:
with open(target_file, "w") as target:
json.dump(source_json_object, target, indent=4, sort_keys=True)
logger.debug("Custom config file was created.")
except:
logger.error("JSON object can not be saved into file.")
return 1
def get_template_info(self, name="", id=""):
logger.debug(
"Retrieving information about template {:s}.".format(name))
rule = {"host": name} if name != "" else {"id": id}
return self.zapi.template.get(filter=rule)
def assign_template(self, host_id, template_name):
template_id = self.get_template_info(
name=template_name)[0]["templateid"]
logger.debug("Template id: {:s}".format(template_id))
logger.debug("Assigning template {:s} with host id {:s}".format(
template_name, host_id))
current_templates = self.get_host_info(
host_id=host_id)[0]["parentTemplates"]
data = {"hostid": host_id, "templates": current_templates}
for item in current_templates:
if (item["templateid"] == template_id):
data = {}
break
if ("templates" in data):
data["templates"].append({"templateid": template_id})
if (len(data) > 0):
try:
self.zapi.host.update(data)
except:
return 0
else:
logger.debug(
"Template {:s} was already assigned before. Skipped.".format(
template_name))
return 1
def main(self):
self.grafana_configurator()
if self.authentication_type != self.default_authentication_type:
logger.debug(
"Changing authentication_type to default to use api with basic credentials."
)
self.update_configuration(
config={
"authentication_type": self.default_authentication_type
})
self.login()
if self.change_default_password():
self.relogin()
if self.disable_guest:
self.disable_user(self.guest_username)
host_id = self.get_host_info(hostname=self.hostname)[0]["hostid"]
logger.debug("%s has such id: %s." % (self.hostname, host_id))
logger.info("Associate local agent with %s." %
(self.hostname) if self.update_host_addr(
host_id, self.agent_dns_name, self.agent_ip_address, 0
) else "Skipped address updating %s." % (self.hostname))
logger.info("Enabled %s host." % (self.hostname) if self.
enable_host(host_id) else "Skipped enabling %s." %
(self.hostname))
logger.info(
"Configured default email media type." if self.
update_mediatype(name="Email",
data={
"smtp_server": self.smtp_server,
"smtp_email": self.smtp_email,
"smtp_helo": self.smtp_helo
}) else "Skipped configuring default media type.")
logger.info(
"Updated %s user email settings." %
(self.default_admin_username) if self.update_user_email_settings(
username=self.default_admin_username,
email=self.admin_email_address
) else "Skipped updating email settings")
logger.info("Enabled default notify action." if self.
enable_action(self.default_report_action) else
"Skipped activating the default notify action.")
logger.info("Added/Updated auto discovery action." if self.
add_auto_discovery_action(self.host_metadata) else
"Skipped adding auto discovery action.")
logger.info("Initialization checking web urls." if self.
add_web_scenario(host_id=host_id, url_list=self.url_list)
else "Skipped initialization of web urls.")
logger.info("Cleanup undefined web urls." if self.
cleanup_undefined_web_scenario(
host_id=host_id, url_list=self.url_list
) else "Skipped cleanup of web urls. Nothing was found.")
logger.info("Adding zabbix configuration templates." if self.
configuration_folder != "" and self.import_configuration()
else "No configuration templates folder was identified.")
logger.info("Creating default user group %s." %
(self.default_user_group))
group_id = self.create_group(self.default_user_group)
if len(self.admin_users) > 0:
for user in self.admin_users:
logger.info("Adding user %s as administrator." %
(user["name"]))
if not self.add_user(
user=user, groups=[{
"usrgrpid": group_id
}], user_type=3):
logger.info("Skipped.")
logger.info("Updating Zabbix configuration" if self.
update_configuration(self.configuration
) else "Nothing to update. Skipped.")
if self.authentication_type != self.default_authentication_type:
logger.debug("Returned authentication_type to initial state.")
self.update_configuration(
config={"authentication_type": self.authentication_type})
# This step must be the last, when custom_config_json is already prepared for saving
logger.info("Creating custom config." if self.save_json_config(
self.custom_config_json, self.zabbix_custom_config
) else "Skipped. Nothing to be saved.")
for template_name in self.additional_templates:
logger.info(
"Assigning templates with %s default host." %
self.hostname if self.assign_template(host_id, template_name)
else "Cannot assign template with host %s." % self.hostname)
return self.logout()
},
'valueMaps': {
'createMissing': True,
'updateExisting': True
},
}
if os.path.isdir(path):
#path = path/*.xml
files = glob.glob(path+'/*.xml')
for file in files:
print(file)
with open(file, 'r') as f:
template = f.read()
try:
zapi.confimport('xml', template, rules)
except ZabbixAPIException as e:
print(e)
print('')
elif os.path.isfile(path):
files = glob.glob(path)
for file in files:
with open(file, 'r') as f:
template = f.read()
try:
if start >= 0 and end >= 0:
for i in range(start, end+1):
currTemplate = template.replace("{{N}}", str(i))
print("#############", i, "#############")
zapi.confimport('xml', currTemplate, rules)
else:
