开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 系统/网络安全 > 杀毒 > 查看源码
wxDialogScheduledScan.py
资源名称：10040637740.zip　[点击查看]
上传用户：lswyart
上传日期：2008-06-12
资源大小：3441k
文件大小：16k
源码类别：

杀毒

开发平台：

Visual C++

wxDialogScheduledScan.py：源码内容
  1. #Boa:Dialog:wxDialogScheduledScan
  3. #-----------------------------------------------------------------------------
  4. # Name:        wxDialogScheduledScan.py
  5. # Product:     ClamWin Free Antivirus
  6. #
  7. # Author:      alch [alch at users dot sourceforge dot net]
  8. #
  9. # Created:     2004/18/04
  10. # Copyright:   Copyright alch (c) 2004
  11. # Licence:     
  12. #   This program is free software; you can redistribute it and/or modify
  13. #   it under the terms of the GNU General Public License as published by
  14. #   the Free Software Foundation; either version 2 of the License, or
  15. #   (at your option) any later version.
  16. # 
  17. #   This program is distributed in the hope that it will be useful,
  18. #   but WITHOUT ANY WARRANTY; without even the implied warranty of
  19. #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  20. #   GNU General Public License for more details.
  21. # 
  22. #   You should have received a copy of the GNU General Public License
  23. #   along with this program; if not, write to the Free Software
  24. #   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  26. #-----------------------------------------------------------------------------
  28. from wxPython.wx import *
  29. from wxPython.lib.timectrl import *
  30. import os, sys, time, locale
  31. import Utils
  32. import shelve
  34. # scheduled scan information holder
  35. # it is used for persistent storage
  36. class ScheduledScanInfo(list):
  37.     def __init__(self, frequency='Daily', time='18:30:00', weekDay=3, path='', description='', active=True):
  38.         list.__init__(self, [frequency, time, weekDay, path, description, active])            
  39.         
  40.     def __getFrequency(self): return self[0]
  41.     def __setFrequency(self, value): self[0] = value    
  42.     Frequency = property(__getFrequency, __setFrequency)
  43.     
  44.     def __getTime(self): return self[1]
  45.     def __setTime(self, value): self[1] = value    
  46.     Time = property(__getTime, __setTime)
  47.     
  48.     def __getWeekDay(self): return self[2]
  49.     def __setWeekDay(self, value): self[2] = value    
  50.     WeekDay = property(__getWeekDay, __setWeekDay)
  51.         
  52.     def __getPath(self): return self[3]
  53.     def __setPath(self, value): self[3] = value    
  54.     Path = property(__getPath, __setPath)
  55.     
  56.     def __getDescription(self): return self[4]
  57.     def __setDescription(self, value): self[4] = value    
  58.     Description = property(__getDescription, __setDescription)
  59.     
  60.     def __getActive(self): return self[5]
  61.     def __setActive(self, value): self[5] = value    
  62.     Active = property(__getActive, __setActive)
  63.     
  65. def LoadPersistentScheduledScans(filename):
  66.     try:              
  67.         _shelve = shelve.open(filename)
  68.         # set version of the persistent storage data
  69.         # we may need it in future when upgrading to newer data set
  70.         try:
  71.             version = _shelve['version']
  72.         except KeyError:
  73.             version = 1
  74.         # read our scheduled scans info
  75.         # or create a new empty list    
  76.         try:
  77.             scheduledScans = _shelve['ScheduledScans']                                 
  78.         except KeyError:
  79.             scheduledScans = []        
  80.         if version < 2:
  81.             for i in range(len(scheduledScans)):
  82.                 scheduledScans[i] = ScheduledScanInfo(scheduledScans[i][0],
  83.                                     scheduledScans[i][1], scheduledScans[i][2],
  84.                                     scheduledScans[i][3], 
  85.                                     'Scan ' + scheduledScans[i][3], True)          
  86.             _shelve['version'] = 2        
  87.             _shelve['ScheduledScans'] = scheduledScans
  88.     except Exception, e:       
  89.         scheduledScans = [] 
  90.         print 'Could not open persistent storage for scheduled scans. Error: %s' % str(e)            
  91.     
  92.     return scheduledScans
  94. def SavePersistentScheduledScans(filename, scheduledScans):
  95.     try:
  96.         _shelve = shelve.open(filename)        
  97.         _shelve['ScheduledScans'] = scheduledScans
  98.         _shelve['version'] = 2
  99.     except Exception, e:               
  100.         print 'Could not save scheduled scans to persistent storage. Error: %s' % str(e)                
  101.     
  102. def create(parent, scanInfo):
  103.     return wxDialogScheduledScan(parent, scanInfo)
  105. [wxID_WXDIALOGSCHEDULEDSCAN, wxID_WXDIALOGSCHEDULEDSCANBUTTONBROWSEFOLDER, 
  106.  wxID_WXDIALOGSCHEDULEDSCANBUTTONCANCEL, wxID_WXDIALOGSCHEDULEDSCANBUTTONOK, 
  107.  wxID_WXDIALOGSCHEDULEDSCANCHECKBOXENABLED, 
  108.  wxID_WXDIALOGSCHEDULEDSCANCHOICEDAY, 
  109.  wxID_WXDIALOGSCHEDULEDSCANCHOICEFREQUENCY, 
  110.  wxID_WXDIALOGSCHEDULEDSCANSPINBUTTONTIME, 
  111.  wxID_WXDIALOGSCHEDULEDSCANSTATICBOX1, 
  112.  wxID_WXDIALOGSCHEDULEDSCANSTATICLINETIMECTRL, 
  113.  wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTDAY, 
  114.  wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTDESCRIPTION, 
  115.  wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTFOLDER, 
  116.  wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTFREQUENCY, 
  117.  wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTTIME, 
  118.  wxID_WXDIALOGSCHEDULEDSCANTEXTCTRLDESCRIPTION, 
  119.  wxID_WXDIALOGSCHEDULEDSCANTEXTCTRLFOLDER, 
  120. ] = map(lambda _init_ctrls: wxNewId(), range(17))
  122. class wxDialogScheduledScan(wxDialog):
  123.     def _init_ctrls(self, prnt):
  124.         # generated method, don't edit
  125.         wxDialog.__init__(self, id=wxID_WXDIALOGSCHEDULEDSCAN,
  126.               name='wxDialogScheduledScan', parent=prnt, pos=wxPoint(427, 201),
  127.               size=wxSize(311, 301), style=wxDEFAULT_DIALOG_STYLE,
  128.               title='Scheduled Scan')
  129.         self.SetClientSize(wxSize(303, 274))
  130.         self.SetToolTipString('')
  131.         self.Center(wxBOTH)
  132.         EVT_CHAR_HOOK(self, self.OnCharHook)
  134.         self.staticBox1 = wxStaticBox(id=wxID_WXDIALOGSCHEDULEDSCANSTATICBOX1,
  135.               label='Schedule', name='staticBox1', parent=self, pos=wxPoint(11,
  136.               8), size=wxSize(282, 104), style=0)
  138.         self.staticTextFrequency = wxStaticText(id=wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTFREQUENCY,
  139.               label='Scanning &Frequency:', name='staticTextFrequency',
  140.               parent=self, pos=wxPoint(25, 30), size=wxSize(131, 13), style=0)
  141.         self.staticTextFrequency.SetToolTipString('')
  143.         self.choiceFrequency = wxChoice(choices=['Hourly', 'Daily', 'Workdays',
  144.               'Weekly'], id=wxID_WXDIALOGSCHEDULEDSCANCHOICEFREQUENCY,
  145.               name='choiceFrequency', parent=self, pos=wxPoint(171, 27),
  146.               size=wxSize(107, 21), style=0)
  147.         self.choiceFrequency.SetColumns(2)
  148.         self.choiceFrequency.SetToolTipString('How often the schedule is executed')
  149.         self.choiceFrequency.SetStringSelection('Daily')
  150.         EVT_CHOICE(self.choiceFrequency,
  151.               wxID_WXDIALOGSCHEDULEDSCANCHOICEFREQUENCY,
  152.               self.OnChoiceFrequency)
  154.         self.staticTextTime = wxStaticText(id=wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTTIME,
  155.               label='&Time:', name='staticTextTime', parent=self,
  156.               pos=wxPoint(25, 56), size=wxSize(121, 18), style=0)
  158.         self.staticLineTimeCtrl = wxStaticLine(id=wxID_WXDIALOGSCHEDULEDSCANSTATICLINETIMECTRL,
  159.               name='staticLineTimeCtrl', parent=self, pos=wxPoint(171, 54),
  160.               size=wxSize(90, 22), style=0)
  161.         self.staticLineTimeCtrl.Show(False)
  162.         self.staticLineTimeCtrl.SetToolTipString('When the schedule should be started')
  164.         self.spinButtonTime = wxSpinButton(id=wxID_WXDIALOGSCHEDULEDSCANSPINBUTTONTIME,
  165.               name='spinButtonTime', parent=self, pos=wxPoint(261, 53),
  166.               size=wxSize(16, 22), style=wxSP_ARROW_KEYS | wxSP_VERTICAL)
  167.         self.spinButtonTime.SetToolTipString('')
  169.         self.staticTextDay = wxStaticText(id=wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTDAY,
  170.               label='&Day Of The Week:', name='staticTextDay', parent=self,
  171.               pos=wxPoint(25, 85), size=wxSize(123, 18), style=0)
  172.         self.staticTextDay.SetToolTipString('')
  174.         self.choiceDay = wxChoice(choices=['Monday', 'Tuesday', 'Wednesday',
  175.               'Thursday', 'Friday', 'Saturday', 'Sunday'],
  176.               id=wxID_WXDIALOGSCHEDULEDSCANCHOICEDAY, name='choiceDay',
  177.               parent=self, pos=wxPoint(171, 82), size=wxSize(107, 21), style=0)
  178.         self.choiceDay.SetColumns(2)
  179.         self.choiceDay.SetToolTipString('When schedule frequency is weekly select day of the week')
  180.         self.choiceDay.SetStringSelection('Tuesday')
  182.         self.staticTextFolder = wxStaticText(id=wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTFOLDER,
  183.               label='&Scan Folder:', name='staticTextFolder', parent=self,
  184.               pos=wxPoint(11, 121), size=wxSize(78, 15), style=0)
  185.         self.staticTextFolder.SetToolTipString('')
  187.         self.textCtrlFolder = wxTextCtrl(id=wxID_WXDIALOGSCHEDULEDSCANTEXTCTRLFOLDER,
  188.               name='textCtrlFolder', parent=self, pos=wxPoint(11, 139),
  189.               size=wxSize(260, 20), style=0, value='')
  190.         self.textCtrlFolder.SetToolTipString('Specify a folder to be scanned')
  192.         self.buttonBrowseFolder = wxButton(id=wxID_WXDIALOGSCHEDULEDSCANBUTTONBROWSEFOLDER,
  193.               label='...', name='buttonBrowseFolder', parent=self,
  194.               pos=wxPoint(273, 139), size=wxSize(20, 20), style=0)
  195.         self.buttonBrowseFolder.SetToolTipString('Click to browse for a folder')
  196.         EVT_BUTTON(self.buttonBrowseFolder,
  197.               wxID_WXDIALOGSCHEDULEDSCANBUTTONBROWSEFOLDER,
  198.               self.OnButtonBrowseFolder)
  200.         self.staticTextDescription = wxStaticText(id=wxID_WXDIALOGSCHEDULEDSCANSTATICTEXTDESCRIPTION,
  201.               label='D&escription:', name='staticTextDescription', parent=self,
  202.               pos=wxPoint(11, 164), size=wxSize(68, 16), style=0)
  203.         self.staticTextDescription.SetToolTipString('')
  205.         self.textCtrlDescription = wxTextCtrl(id=wxID_WXDIALOGSCHEDULEDSCANTEXTCTRLDESCRIPTION,
  206.               name='textCtrlDescription', parent=self, pos=wxPoint(11, 182),
  207.               size=wxSize(282, 20), style=0, value='')
  208.         self.textCtrlDescription.SetToolTipString('Specify a friendly description for the scheduled scan')
  210.         self.checkBoxEnabled = wxCheckBox(id=wxID_WXDIALOGSCHEDULEDSCANCHECKBOXENABLED,
  211.               label='&Activate This Schedule', name='checkBoxEnabled',
  212.               parent=self, pos=wxPoint(11, 213), size=wxSize(278, 15), style=0)
  213.         self.checkBoxEnabled.SetValue(False)
  214.         self.checkBoxEnabled.SetToolTipString('Select if you wish to enable this schedule')
  216.         self.buttonOK = wxButton(id=wxID_WXDIALOGSCHEDULEDSCANBUTTONOK,
  217.               label='OK', name='buttonOK', parent=self, pos=wxPoint(73, 242),
  218.               size=wxSize(75, 23), style=0)
  219.         self.buttonOK.SetDefault()
  220.         self.buttonOK.SetToolTipString('Closes the dialog and applies the settings')
  221.         EVT_BUTTON(self.buttonOK, wxID_WXDIALOGSCHEDULEDSCANBUTTONOK, self.OnOK)
  223.         self.buttonCancel = wxButton(id=wxID_WXDIALOGSCHEDULEDSCANBUTTONCANCEL,
  224.               label='Cancel', name='buttonCancel', parent=self, pos=wxPoint(160,
  225.               242), size=wxSize(75, 23), style=0)
  226.         self.buttonCancel.SetToolTipString('Closes the dialog and discards the changes')
  227.         EVT_BUTTON(self.buttonCancel, wxID_WXDIALOGSCHEDULEDSCANBUTTONCANCEL,
  228.               self.OnCancel)
  230.     def __init__(self, parent, scanInfo):   
  231.         self._scanInfo = None     
  232.         self._scanInfo = scanInfo        
  233.         self._init_ctrls(parent)
  234.         locale.setlocale(locale.LC_ALL, 'C')                            
  235.         self.timeCtrl = wxTimeCtrl(parent=self, 
  236.                         pos=self.staticLineTimeCtrl.GetPosition(), 
  237.                         size=self.staticLineTimeCtrl.GetSize(), 
  238.                         fmt24hr=Utils.IsTime24(), 
  239.                         spinButton=self.spinButtonTime,
  240.                         useFixedWidthFont=False, display_seconds=True)
  241.         self.timeCtrl.SetToolTipString(self.staticLineTimeCtrl.GetToolTip().GetTip())
  242.         
  243.         self.choiceFrequency.SetValidator(MyValidator(self._scanInfo, 'Frequency'))        
  244.         self.timeCtrl.SetValidator(MyValidator(self._scanInfo, 'Time'))
  245.         self.choiceDay.SetValidator(MyValidator(self._scanInfo, 'WeekDay'))        
  246.         self.textCtrlDescription.SetValidator(MyValidator(self._scanInfo, 'Description', False))                
  247.         self.textCtrlFolder.SetValidator(MyValidator(self._scanInfo, 'Path', False))                
  248.         self.checkBoxEnabled.SetValidator(MyValidator(self._scanInfo, 'Active'))                
  249.         self.TransferDataToWindow()   
  250.         
  251.         self.choiceDay.Enable(self.choiceFrequency.GetStringSelection() == 'Weekly')        
  252.         
  253.     def _Apply(self):                       
  254.         if not self.Validate():
  255.             return False
  256.         self.TransferDataFromWindow()              
  257.         return True        
  258.     
  259.     def OnChoiceFrequency(self, event):        
  260.         self.choiceDay.Enable(self.choiceFrequency.GetStringSelection() == 'Weekly')
  261.         event.Skip()
  262.   
  263.     def OnOK(self, event):
  264.         if self._Apply():
  265.             self.EndModal(wxID_OK)
  267.     def OnCancel(self, event):
  268.         self.EndModal(wxID_CANCEL)
  270.     def OnCharHook(self, event):        
  271.         if event.GetKeyCode() == WXK_ESCAPE:
  272.             self.EndModal(wxID_CANCEL)
  273.         else:
  274.             event.Skip()  
  276.     def OnButtonBrowseFolder(self, event):
  277.         dlg = wxDirDialog(self)
  278.         try:
  279.             if dlg.ShowModal() == wxID_OK:
  280.                 dir = dlg.GetPath()                            
  281.                 self.textCtrlFolder.Clear()
  282.                 self.textCtrlFolder.WriteText(dir)   
  283.         finally:
  284.             dlg.Destroy()
  286. class MyValidator(wxPyValidator):
  287.     def __init__(self, scanInfo, propName, canEmpty=True):         
  288.         wxPyValidator.__init__(self)         
  289.         self._scanInfo = scanInfo
  290.         self._propName = propName
  291.         self._canEmpty = canEmpty
  292.                   
  293.     def Clone(self):         
  294.         return MyValidator(self._scanInfo, self._propName, self._canEmpty)
  296.     def Validate(self, win):         
  297.         ctrl = self.GetWindow()
  298.         if isinstance(ctrl, (wxChoice, wxCheckBox)) or self._canEmpty:
  299.             return True   
  300.         if isinstance(ctrl, wxSpinCtrl):     
  301.             text = str(ctrl.GetValue())
  302.         else:
  303.             text = ctrl.GetValue()
  304.         invalid = False
  305.         if len(text) == 0:            
  306.             wxMessageBox("Value cannot be empty", "ClamWin", style=wxICON_EXCLAMATION|wxOK)
  307.             invalid = True            
  308.         elif ctrl.GetName() == 'textCtrlFolder' and not os.path.exists(text):
  309.             wxMessageBox("Specified path is invalid. Plese verify your selection.", "ClamWin", style=wxICON_EXCLAMATION|wxOK)
  310.             invalid = True            
  311.         else:
  312.             ctrl.SetBackgroundColour(wxSystemSettings_GetColour(wxSYS_COLOUR_WINDOW))
  313.             ctrl.Refresh()            
  314.         if invalid:       
  315.             ctrl.SetBackgroundColour("yellow")
  316.             ctrl.SetFocus()
  317.             ctrl.Refresh()     
  318.         return not invalid
  320.     def TransferToWindow(self):
  321.         value = getattr(self._scanInfo, self._propName)         
  322.         ctrl = self.GetWindow()             
  323.         if(isinstance(ctrl, wxChoice)):
  324.             if ctrl.GetName() == 'choiceDay':
  325.                 ctrl.SetSelection(value)
  326.             else:
  327.                 ctrl.SetStringSelection(value)
  328.         else:            
  329.             ctrl.SetValue(value)         
  330.         return True
  333.     def TransferFromWindow(self):
  334.         ctrl = self.GetWindow()
  335.         if(isinstance(ctrl, wxChoice)):
  336.             if ctrl.GetName() == 'choiceDay':
  337.                 value = ctrl.GetSelection()
  338.             else:
  339.                 value = ctrl.GetStringSelection()            
  340.         elif isinstance(ctrl, wxCheckBox):
  341.             value = ctrl.GetValue()
  342.         elif isinstance(ctrl, wxTimeCtrl):
  343.             # set C locale, otherwise python and wxpython complain
  344.             locale.setlocale(locale.LC_ALL, 'C')            
  345.             value = ctrl.GetWxDateTime().Format('%H:%M:%S')
  346.         else:
  347.             value = ctrl.GetValue()                 
  348.         setattr(self._scanInfo, self._propName, value)
  349.         return True