C:/work/RevealedSystems/EyeSpy-RS/eyespy/src/objects/gui/networkSettings.py

Go to the documentation of this file.

# Copyright (C) 2008-2009  Mark Hanegraaff
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import wx
import sys

from objects.logic.common.exception import exception
from objects.logic.common.logger import logger
from objects.logic.network.pcapyImplementation import pcapyImplementation


## the networkSettings class represents the "Network Settings" window which is
# used to configure which network adapter to monitor
class networkSettings(wx.Dialog):
     def __init__(self, parent, deviceDict):
        wx.Dialog.__init__(self, parent, -1, "Which device would you like to monitor?", 
                                   size=(400,220))
        
        self.chosenDevice = None
        self.deviceDict = deviceDict
        
        self.mainSizer = wx.BoxSizer(wx.VERTICAL)
        
        self.deviceIdxDict = dict()
        self.saveSelection = False
        
        self.createWidgets()
        self.layoutWidgets()
        self.doBinding()

        
        self.SetSizer(self.mainSizer)
     
     
     
     ## Loads the configuration from the optionsConfigurator object and displays it onto the screen
     # @param optConf optionsConfigurator object
     def loadConfig(self, optConf):
         saveSel = optConf.network_getSaveDeviceSelection()
         selectedDevice = optConf.network_getSelectedDevice()
         
         if saveSel == True:
             self.saveSelection = True                     
             if self.deviceIdxDict.has_key(selectedDevice):
                 self.listc.Select(self.deviceIdxDict[selectedDevice])
                 self.cehckbox.SetValue(True)
         else:
             self.saveSelection = False
             
                 
             
                 
     ## Creates all various widgets and displays them onto the screen.
     def createWidgets(self):
        #
        # list box
        #
        self.listc = wx.ListCtrl(self, -1, style=wx.LC_REPORT|wx.LC_SINGLE_SEL,
                               size=(400, 120))
        
        self.listc.InsertColumn(0, "Network Device", width=340)
        self.listc.InsertColumn(1, "Packets",width=50)
        
        
        for name, desc in self.deviceDict.iteritems():        
            index = self.listc.InsertStringItem(sys.maxint, name)        
            self.listc.SetStringItem(index, 1, "-")
            self.deviceIdxDict[name] = index
            

            
        if len(self.deviceDict) > 0:
            self.listc.Select(0)
    
        #
        # CheckBox
        #
        self.cehckbox = wx.CheckBox(self, -1, "Remember this choice")
         
        
        #
        # Buttons
        #
        self.button_ok     = wx.Button(self, wx.ID_OK, "Ok")
        self.button_cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")        
         
     
     
     ## Lays out all all widgets onto the screen
     def layoutWidgets(self):
        # layout list box
        ctrlSizer = wx.BoxSizer(wx.HORIZONTAL)
        ctrlSizer.Add(self.listc, 10)
                
        self.mainSizer.Add(ctrlSizer, 0, wx.EXPAND | wx.ALIGN_CENTER_VERTICAL, 10)
        
        emptySizer = wx.BoxSizer(wx.HORIZONTAL)
        emptySizer.Add((15,15), 1)
        
        self.mainSizer.Add(emptySizer, 0,wx.EXPAND | wx.ALIGN_CENTER_VERTICAL,  10)
        
        #layout checkbox
        chkSizer = wx.BoxSizer(wx.HORIZONTAL)
        chkSizer.Add(self.cehckbox, 1)
         
        self.mainSizer.Add(chkSizer, 0, wx.LEFT, 10)
        
        emptySizer = wx.BoxSizer(wx.HORIZONTAL)
        emptySizer.Add((15,15), 1)
        
        self.mainSizer.Add(emptySizer, 0,wx.EXPAND | wx.ALIGN_CENTER_VERTICAL,  10)
        
        # buttons
        btnSizer = wx.BoxSizer(wx.HORIZONTAL)
        
        btnSizer.Add(self.button_ok)
        btnSizer.Add((10,10), 1)
        btnSizer.Add(self.button_cancel)

        self.mainSizer.Add(btnSizer, 0,   wx.ALIGN_RIGHT | wx.RIGHT,10)
                
    
     ## Binds all the various buttons and controls to their handler methods
     def doBinding(self):        
        
        self.Bind(wx.EVT_BUTTON, self.OnOK, self.button_ok)
        self.Bind(wx.EVT_BUTTON, self.OnCancel, self.button_cancel)
        
        self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected, self.listc)
        
        self.Bind(wx.EVT_CHECKBOX, self.OnSaveSelection, self.cehckbox)
        
     """
        Event Handlers
     """
     
     ## "OK" Button handler
     # @param event Standard wx.Button event
     def OnOK(self, event):         
        if self.chosenDevice == None:
            dlg = wx.MessageDialog(None, "No Item was selected", 'Error', wx.ICON_ERROR)
            dlg.ShowModal()
            dlg.Destroy()
            return
        
        logger.log(__name__ + ":Item selected:" + self.chosenDevice )
        return self.EndModal(wx.ID_OK)
    
     ## "Cancel" Button handler
     # @param event Standard wx.Button event
     def OnCancel(self, event):            
        return self.EndModal(wx.ID_CANCEL)



     ## Cleanly closes the window
     # @param event Standard wx.Button event
     def OnCloseWindow(self, event):        
        self.Destroy()
        
        
     ## Handler which is called when an item (a network device) is selected
     # @param evt Standard wx.Button event
     def OnItemSelected(self, evt):
        item = evt.GetItem()
        self.chosenDevice = item.GetText()
    
     ## Handler which is called "Remember this choice" CheckBox is clicked
     # @param evt Standard wx.Button event
     def OnSaveSelection(self, evt):
        self.saveSelection = self.cehckbox.GetValue() 
        
        
        
         
         

        
            
        

---