valitydev/atomic-threat-coverage
14
0
Fork 0
mirror of https://github.com/valitydev/atomic-threat-coverage.git synced 2024-11-06 17:45:23 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Visualisation validate method update

Browse Source
This commit is contained in:
mrblacyk 2019-03-14 13:04:50 -07:00
parent 84e3685e2a
commit f7846631b5
1 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
scripts/KibanaStuff/visualisation.py
View File

@ -69,6 +69,12 @@ class KibanaVisualizationDoc(base.BaseKibanaDoc):
}
)
def validate(self):
if self._meta_data_set and super().validate():
return True
else:
return False
def add_metric(self, metric):
if not issubclass(metric.__class__, BaseMetric):
raise Exception("Are you trying to add non-metric?")
@ -96,18 +102,26 @@ class KibanaVisualizationDoc(base.BaseKibanaDoc):
raise Exception("Data validation failed")
def set_index_search(self, index_name):
if self.search_id_of_title_by_type(
search_type="index-pattern", search_title=index_name
):
if self.check_kibana_vars():
if self.search_id_of_title_by_type(
search_type="index-pattern", search_title=index_name
):
self.visualization.kibanaSavedObjectMeta["searchSourceJSON"] =\
("{\"index\":\"%s\",\"query\":{\"query_string\":{" +
"\"analyze_wildcard\":true,\"query\":\"*\"}}," +
"\"filter\":[]}") % index_name
self._meta_data_set = True
else:
raise Exception(
"Did not find such index name." +
" Didn't you forget an asterisk?"
)
else:
self.visualization.kibanaSavedObjectMeta["searchSourceJSON"] = \
("{\"index\":\"%s\",\"query\":{\"query_string\":{" +
"\"analyze_wildcard\":true,\"query\":\"*\"}},\"filter\":[]" +
"}") % index_name
"\"analyze_wildcard\":true,\"query\":\"*\"}}," +
"\"filter\":[]}") % index_name
self._meta_data_set = True
else:
raise Exception(
"Did not find such index name. Didn't you forget an asterisk?"
)
def set_saved_search(self, saved_search_name=None, saved_search_id=None):
"""Provide ID if you know it and don't want to engage kibana"""