diff --git a/zabbix/zabbix_api.py b/zabbix/zabbix_api.py index 87ccae9..a4ed071 100644 --- a/zabbix/zabbix_api.py +++ b/zabbix/zabbix_api.py @@ -137,6 +137,7 @@ def __init__(self, server='http://localhost/zabbix', user=httpuser, passwd=httpp self.server = server self.url = server + '/api_jsonrpc.php' self.proto = self.server.split("://")[0] + self.user_agent #self.proto=proto self.httpuser = user self.httppasswd = passwd @@ -154,6 +155,12 @@ def set_log_level(self, level): self.debug(logging.INFO, "Set logging level to %d" % level) self.logger.setLevel(level) + def set_user_agent(self, user_agent): + self.user_agent = user_agent + + def get_user_agent(self): + return self.user_agent + def recent_query(self): """ return recent query @@ -218,8 +225,7 @@ def test_login(self): def do_request(self, json_obj): headers = {'Content-Type': 'application/json-rpc', - 'User-Agent': 'python/zabbix_api'} - + 'User-Agent': 'python/zabbix_api/' + self.user_agent} if self.httpuser: self.debug(logging.INFO, "HTTP Auth enabled") auth = 'Basic ' + string.strip(base64.encodestring(self.httpuser + ':' + self.httppasswd))