Exemple #1
0
    def buildContent(self):
        """Builds the user interface window"""
        self.parseConfig()
        self.grid()
        
        menubar = Tkinter.Menu(self)
        aboutMenu = Tkinter.Menu(menubar, tearoff=0)
        aboutMenu.add_command(label="About WormBait", command=self.showAboutWindow)
        menubar.add_cascade(label="About", menu=aboutMenu)
       
        self.entryList.grid(column=0, row=0, columnspan=2, sticky='NSEW')
        
        self.dbFileEntry = Tkinter.Entry(self, textvariable=self.dbFilePath)
        self.dbFileEntry.grid(column=0, row=1, sticky='EW')

        dbFileBrowseButton = Tkinter.Button(self, text="Browse...", command=self.OnDBBrowseButtonClick)
        dbFileBrowseButton.grid(column=1, row=1)
        
        self.outFileEntry = Tkinter.Entry(self, textvariable=self.outFilePath)
        self.outFileEntry.grid(column=0, row=2, sticky='EW')

        outFileBrowseButton = Tkinter.Button(self, text="Browse...", command=self.OnOutBrowseButtonClick)
        outFileBrowseButton.grid(column=1, row=2)

        self.button = ProcessButton(self)
        self.button.bind("<Return>", self.button.OnClick)
        self.button.grid(column=0,row=3)
        
        self.console = ConsoleBox(self, 30)
        self.console.grid(column=0,row=4, columnspan=2, sticky='NSEW')

        self.grid_columnconfigure(0, weight=1)
        self.grid_rowconfigure(0, weight=1)
        self.grid_rowconfigure(4, weight=1)
        self.resizable(True,False)
        self.update()
        self.config(menu=menubar)     
        self.entryList.focus_set()
Exemple #2
0
 def initialize(self):
     self.dbFilePath = Tkinter.StringVar()
     self.outFilePath = Tkinter.StringVar()
     self.entryList = ConsoleBox(self, 10)
     self.protocol("WM_DELETE_WINDOW", self.saveIniAndDestroy)
     self.buildContent()
Exemple #3
0
class WormbaitWindow(Tkinter.Tk):
    """The top-level window of WormBait"""
    VERSION = "2.0"
    
    def __init__(self, parent):
        Tkinter.Tk.__init__(self, parent)
        self.parent = parent
        self.initialize()

    def initialize(self):
        self.dbFilePath = Tkinter.StringVar()
        self.outFilePath = Tkinter.StringVar()
        self.entryList = ConsoleBox(self, 10)
        self.protocol("WM_DELETE_WINDOW", self.saveIniAndDestroy)
        self.buildContent()

    def buildContent(self):
        """Builds the user interface window"""
        self.parseConfig()
        self.grid()
        
        menubar = Tkinter.Menu(self)
        aboutMenu = Tkinter.Menu(menubar, tearoff=0)
        aboutMenu.add_command(label="About WormBait", command=self.showAboutWindow)
        menubar.add_cascade(label="About", menu=aboutMenu)
       
        self.entryList.grid(column=0, row=0, columnspan=2, sticky='NSEW')
        
        self.dbFileEntry = Tkinter.Entry(self, textvariable=self.dbFilePath)
        self.dbFileEntry.grid(column=0, row=1, sticky='EW')

        dbFileBrowseButton = Tkinter.Button(self, text="Browse...", command=self.OnDBBrowseButtonClick)
        dbFileBrowseButton.grid(column=1, row=1)
        
        self.outFileEntry = Tkinter.Entry(self, textvariable=self.outFilePath)
        self.outFileEntry.grid(column=0, row=2, sticky='EW')

        outFileBrowseButton = Tkinter.Button(self, text="Browse...", command=self.OnOutBrowseButtonClick)
        outFileBrowseButton.grid(column=1, row=2)

        self.button = ProcessButton(self)
        self.button.bind("<Return>", self.button.OnClick)
        self.button.grid(column=0,row=3)
        
        self.console = ConsoleBox(self, 30)
        self.console.grid(column=0,row=4, columnspan=2, sticky='NSEW')

        self.grid_columnconfigure(0, weight=1)
        self.grid_rowconfigure(0, weight=1)
        self.grid_rowconfigure(4, weight=1)
        self.resizable(True,False)
        self.update()
        self.config(menu=menubar)     
        self.entryList.focus_set()

    def saveIniAndDestroy (self):
        """Writes current information to the ini file (wormBait.ini) and closes the user interface"""
        config = SafeConfigParser()
        config.read('wormBait.ini')

        if not config.has_section('wormBait'):
            config.add_section('wormBait')

        dbIds = self.entryList.getValue()
        if '\n' in dbIds:
            dbIds = ','.join(dbIds.strip().split('\n'))
            
        config.set('wormBait', 'dbIds', dbIds)
        config.set('wormBait', 'degFile', self.dbFilePath.get())
        config.set('wormBait', 'outFile', self.outFilePath.get())

        with open('wormBait.ini', 'w') as f:
            config.write(f)

        self.destroy()

    def parseConfig (self):
        """Reads the config file (wormBait.ini) and initializes variables accordingly"""
        config = SafeConfigParser()
        config.read('wormBait.ini')

        configDbIds = ""
        configDegFile = ""
        configOutFile = ""
        
        if config.has_section('wormBait'):
            configDbIds = config.get('wormBait', 'dbIds')
            configDegFile = config.get('wormBait', 'degFile')
            configOutFile = config.get('wormBait', 'outFile')

        if configDbIds:
            self.entryList.writeln(configDbIds)
        else:
            self.entryList.writeln("Enter DB IDs here")

        if configDegFile:
            self.dbFilePath.set(configDegFile)
        else:
            self.dbFilePath.set('Enter path to input database file here')

        if configOutFile:
            self.outFilePath.set(configOutFile)
        else:
            self.outFilePath.set('Enter desired path to output CSV file here')

    def OnDBBrowseButtonClick (self, *ignore):
        """Opens a browser for selecting the database file"""
        dialogReturn = tkFileDialog.askopenfilename(filetypes=[('Comma-separated value', '*.csv')])
        self.dbFilePath.set(dialogReturn)

    def OnOutBrowseButtonClick (self, *ignore):
        """Opens a browser for selecting the CSV output file location"""
        dialogReturn = tkFileDialog.asksaveasfilename(filetypes=[('Comma-separated value', '*.csv')])
        self.outFilePath.set(dialogReturn)

    def showAboutWindow (self):
        """Shows the About window"""
        aboutWindow = AboutWindow(self)