def uninstallQuery (tree, query):
req = buildReq ('uninstallQuery', tree, query)
resp, content = r.get (controller, json.dumps (req, default=json_util.default), "pathdump")
if resp['status'] != '200':
return []
else:
return json.loads (content, object_hook=json_util.object_hook)
def getFlowCollectionDir():
req = {'api': 'getFlowCollDir'}
resp, content = r.get (controller, json.dumps (req, default=json_util.default), "pathdump")
if resp['status'] != '200':
return ''
else:
return json.loads (content, object_hook=json_util.object_hook)[0]
def getAggTree (groupnodes):
req = {'api': 'getAggTree'}
req.update ({'groupnodes': groupnodes})
resp, content = r.get (controller, json.dumps (req, default=json_util.default), "pathdump")
if resp['status'] != '200':
return {}
else:
return json.loads (content, object_hook=json_util.object_hook)[0]
def installQuery (tree, query, interval):
hosts = check_source (tree, query['name'])
if send_source (hosts, tree, query['name']) == False:
return []
req = buildReq ('installQuery', tree, query, None, interval)
resp, content = r.get (controller, json.dumps (req, default=json_util.default), "pathdump")
if resp['status'] != '200':
return []
else:
return json.loads (content, object_hook=json_util.object_hook)
def execQuery (tree, query, aggcode=None):
hosts = check_source (tree, query['name'])
if send_source (hosts, tree, query['name']) == False:
return []
if aggcode:
hosts = check_source (tree, aggcode['name'])
if send_source (hosts, tree, aggcode['name']) == False:
return []
req = buildReq ('execQuery', tree, query, aggcode)
resp, content = r.get (controller, json.dumps (req, default=json_util.default), "pathdump")
if resp['status'] != '200':
return []
else:
return json.loads (content, object_hook=json_util.object_hook)
def registerQuery (filepath):
filename = os.path.basename (filepath)
try:
with open (filepath, 'r') as f:
filedata = f.read()
except EnvironmentError:
return [False]
req = {'api': 'registerQuery'}
req.update ({'name': filename})
req.update ({'data': filedata})
resp, content = r.get (controller, json.dumps (req, default=json_util.default), "pathdump")
if resp['status'] != '200':
return []
else:
return json.loads (content, object_hook=json_util.object_hook)
def httpcmd(server, req, url):
return r.get(server, json.dumps(req, default=json_util.default), url)