Add query mode to flow graph reports.

1 files changed, 12 insertions, 0 deletions

diff --git a/python/interaction.py b/python/interaction.py
index 13cf93c6..7cc4442b 100644
--- a/python/interaction.py
+++ b/python/interaction.py
@@ -984,6 +984,18 @@ class ReportCollection(object):
 		else:
 			raise TypeError("expected report object")
 
+	def update(self, i, report):
+		# if isinstance(report, PlainTextReport):
+		# 	core.BNUpdatePlainTextReportToCollection(self.handle, i, report.contents)
+		# elif isinstance(report, MarkdownReport):
+		# 	core.BNUpdateMarkdownReportToCollection(self.handle, i, report.contents, report.plaintext)
+		# elif isinstance(report, HTMLReport):
+		# 	core.BNUpdateHTMLReportToCollection(self.handle, i, report.contents, report.plaintext)
+		if isinstance(report, FlowGraphReport):
+			core.BNUpdateReportFlowGraph(self.handle, i, report.graph.handle)
+		else:
+			raise TypeError("expected report object")
+
 
 def markdown_to_html(contents):
 	"""