Skip to content

Commit bdc9431

Browse files
andrew-klineAndrew Kline
andauthored
Add run_all_insights to analyze message function (#36)
* feat: add run_all_insights to analyze message function * chore: version bump --------- Co-authored-by: Andrew Kline <andrew.kline@getcruise.com>
1 parent b1b6de8 commit bdc9431

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def read(fname):
3434

3535
setup(
3636
name="sublime-cli",
37-
version="0.0.30",
37+
version="0.0.31",
3838
description="Abstraction to interact with the Sublime API.",
3939
url="https://sublimesecurity.com/",
4040
author="Sublime Security",

src/sublime/api.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def create_message(self, raw_message, mailbox_email_address=None, message_type=N
197197
response, _ = self._request(endpoint, request_type='POST', json=body)
198198
return response
199199

200-
def analyze_message(self, raw_message, rules, queries, run_all_detection_rules=False, run_active_detection_rules=False):
200+
def analyze_message(self, raw_message, rules, queries, run_all_detection_rules=False, run_active_detection_rules=False, run_all_insights=False):
201201
"""Analyze a Message Data Model against a list of rules or queries.
202202
203203
:param raw_message: Base64 encoded raw message
@@ -211,6 +211,8 @@ def analyze_message(self, raw_message, rules, queries, run_all_detection_rules=F
211211
:type run_all_detection_rules: bool
212212
:param run_active_detection_rules: whether to run active detection rules against the given message
213213
:type run_active_detection_rules: bool
214+
:param run_all_insights: whether to run all insight queries against the given message
215+
:type run_all_insights: bool
214216
215217
"""
216218

@@ -222,6 +224,7 @@ def analyze_message(self, raw_message, rules, queries, run_all_detection_rules=F
222224
"queries": queries,
223225
"run_all_detection_rules": run_all_detection_rules,
224226
"run_active_detection_rules": run_active_detection_rules,
227+
"run_all_insights": run_all_insights,
225228
}
226229

227230
endpoint = self._EP_MESSAGES_ANALYZE

0 commit comments

Comments
 (0)