Changeset 3126

Timestamp:

07/10/08 14:55:01 (5 years ago)

Author:

nopper

Message:

Making more exotic blah

Location:

branch/UmitPlugins/source-plugins

Files:

• context-menu/sources/main.py (modified) (2 diffs)
• flow-analyzer/sources/main.py (modified) (4 diffs)

branch/UmitPlugins/source-plugins/context-menu/sources/main.py

@@ -25 +25 @@
 from umitPlugin.Engine import Plugin
 from higwidgets.higanimates import HIGAnimatedBar
+from umitGUI.ScanNotebook import ScanNotebookPage, ScanResult
+
+class MyWidget(HIGAnimatedBar):
+    pass
 
 class MenuPlugin(Plugin):
@@ -32 +36 @@
         self.id = Core().connect('ScanHostsView-created', self.__on_created_hosts)
 
+        for page in Core().get_main_scan_notebook():
+            if isinstance(page, ScanNotebookPage):
+                self.__on_created_hosts( \
+                    Core(), page.scan_result.scan_host_view)
+
     def stop(self):
         Core().disconnect(self.id)
 
+        for page in Core().get_main_scan_notebook():
+            if not isinstance(page, ScanNotebookPage):
+                continue
+
+            for child in page.scan_result.scan_host_view:
+                if isinstance(child, MyWidget):
+                    page.scan_result.scan_host_view.remove(child)
+                    child.hide()
+                    child.destroy()
+
     def __on_created_hosts(self, core, view):
-        widget = HIGAnimatedBar('<tt>I\'m here spying your hosts</tt>')
+        widget = MyWidget('<tt>I\'m here spying your hosts</tt>')
         widget.show()
         view.pack_start(widget, False, False)

branch/UmitPlugins/source-plugins/flow-analyzer/sources/main.py

@@ -25 +25 @@
 from umitPlugin.Engine import Plugin
 from higwidgets.higanimates import HIGAnimatedBar
+from umitGUI.ScanNotebook import ScanNotebookPage
 
 class FlowContainer(gtk.VBox):
@@ -44 +45 @@
         paned.pack1(scroll(self.text1))
         paned.pack2(scroll(self.text2))
+
+        message = "Not yet avaiable. Try to start a scan!"
+
+        self.text1.get_buffer().set_text(message)
+        self.text2.get_buffer().set_text(message)
         
         self.pack_start(paned)
@@ -50 +56 @@
     def update(self, page):
         # Simple set the xml and normal output to textviews
-        f = open(page.command_execution.get_xml_output_file(), "r")
-        txt = "".join(f.readlines())
-        
-        self.text1.get_buffer().set_text(txt)
-        
-        f = open(page.command_execution.get_output_file(), "r")
-        txt = "".join(f.readlines())
-        
-        self.text2.get_buffer().set_text(txt)
-        
+
+        if hasattr(page, 'command_execution') and page.command_execution:
+            f = open(page.command_execution.get_xml_output_file(), "r")
+            txt = "".join(f.readlines())
+            
+            self.text1.get_buffer().set_text(txt)
+            
+            f = open(page.command_execution.get_output_file(), "r")
+            txt = "".join(f.readlines())
+            
+            self.text2.get_buffer().set_text(txt)
 
 class FlowPlugin(Plugin):
@@ -65 +72 @@
         self.id = Core().connect('ScanNotebookPage-created', self.__on_created)
 
+        for page in Core().get_main_scan_notebook():
+            if isinstance(page, ScanNotebookPage):
+                self.__on_created(Core(), page)
+                page.emit('scan-finished')
+
     def stop(self):
         Core().disconnect(self.id)
+
+        for page in Core().get_main_scan_notebook():
+            if not isinstance(page, ScanNotebookPage):
+                continue
+
+            for child in page.scan_result.scan_result_notebook:
+                if not isinstance(child, FlowContainer):
+                    continue
+
+                idx = page.scan_result.scan_result_notebook.page_num(child)
+                page.scan_result.scan_result_notebook.remove_page(idx)
+
+                child.hide()
+                child.destroy()
 
     def __on_created(self, core, page):