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 initialize(self):
self.dbFilePath = Tkinter.StringVar()
self.outFilePath = Tkinter.StringVar()
self.entryList = ConsoleBox(self, 10)
self.protocol("WM_DELETE_WINDOW", self.saveIniAndDestroy)
self.buildContent()
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)
