def __init__(self, parent, guesser=None, itemClass=None):
self.status = StatusBar(parent)
self.status.pack(side=BOTTOM, fill=X)
Frame.__init__(self, parent)
self.pack(side=TOP, fill=BOTH)
self.itemsPerPage = 20
self.rows = []
for i in range(self.itemsPerPage):
self.rows.append(ItemRow())
self.items = []
self.files = []
self.cursor = 0
self.dirty = False
if guesser is None:
from reverend.thomas import Bayes
self.guesser = Bayes()
else:
self.guesser = guesser
if itemClass is None:
self.itemClass = TextItem
else:
self.itemClass = itemClass
for row in self.rows:
row.summary.set('foo')
self.initViews()
class Trainer(Frame):
def __init__(self, parent, guesser=None, itemClass=None):
self.status = StatusBar(parent)
self.status.pack(side=BOTTOM, fill=X)
Frame.__init__(self, parent)
self.pack(side=TOP, fill=BOTH)
self.itemsPerPage = 20
self.rows = []
for i in range(self.itemsPerPage):
self.rows.append(ItemRow())
self.items = []
self.files = []
self.cursor = 0
self.dirty = False
if guesser is None:
from reverend.thomas import Bayes
self.guesser = Bayes()
else:
self.guesser = guesser
if itemClass is None:
self.itemClass = TextItem
else:
self.itemClass = itemClass
for row in self.rows:
row.summary.set('foo')
self.initViews()
def initViews(self):
self.nb = Notebook(self)
## frame1 = Frame(self.nb())
## self.poolView = PoolView(frame1, guesser=self.guesser, app=self)
## self.poolView.pack(side=TOP)
frame2 = Frame(self.nb())
self.poolView = PoolView(frame2, guesser=self.guesser, app=self)
self.poolView.pack(side=TOP)
self.listView = Canvas(frame2, relief=GROOVE)
self.listView.pack(padx=3)
bn = Button(self.listView, text="Load training", command=self.loadCorpus)
bn.pack(side=RIGHT, anchor=NE, fill=X)
self.columnHeadings()
self.addNextPrev()
frame3 = Frame(self.nb())
self.testView = TestView(frame3, guesser=self.guesser, app=self)
self.testView.pack()
frame4 = Frame(self.nb())
bp = Button(frame4, text="Quit", command=self.quitNow)
bp.pack(side=BOTTOM)
#self.nb.add_screen(frame1, 'Reverend')
self.nb.add_screen(frame2, 'Training')
self.nb.add_screen(frame3, 'Testing')
self.nb.add_screen(frame4, 'Quit')
def addNextPrev(self):
npFrame = Frame(self.listView)
npFrame.pack(side=BOTTOM, fill=X)
bn = Button(npFrame, text="Prev Page", command=self.prevPage)
bn.grid(row=0, column=0)
bn = Button(npFrame, text="Next Page", command=self.nextPage)
bn.grid(row=0, column=1)
def loadCorpus(self):
path = tkFileDialog.askdirectory()
if not path:
return
self.loadFileList(path)
self.displayItems()
self.displayRows()
def bulkTest(self):
dirs = []
for pool in self.guesser.poolNames():
path = tkFileDialog.askdirectory()
dirs.append((pool, path))
for pool, path in dirs:
print pool, path
def displayList(self):
for item in self.items:
self.itemRow(item)
def displayRows(self):
for row in self.rows:
self.displayRow(row)
def loadFileList(self, path):
listing = os.listdir(path)
self.files = [os.path.join(path, file) for file in listing]
self.cursor = 0
def prevPage(self):
self.cursor = max(0, self.cursor - self.itemsPerPage)
self.displayItems()
def nextPage(self):
self.cursor = min(len(self.files), self.cursor + self.itemsPerPage)
self.displayItems()
def displayItems(self):
theseFiles = self.files[self.cursor:self.cursor + self.itemsPerPage]
items = []
for file, row in zip(theseFiles, self.rows):
fp = open(file, 'rb')
try:
item = self.itemClass.fromFile(fp)
finally:
fp.close()
if item is None:
continue
items.append(item)
guesses = self.guesser.guess(item)
summary = item.summary()
cols = item.columnDefs()
s = ''
for c, ignore in cols:
s += summary[c] + ' '
row.initialize(item, s, guesses, self.guesser.poolNames())
self.items = items
def quitNow(self):
if self.dirty:
if tkMessageBox.askyesno("You have unsaved changes!", "Quit without saving?"):
self.quit()
self.quit()
def columnHeadings(self):
# FIXME - Something better for columns and rows in general
line = Frame(self.listView, relief=RAISED, borderwidth=1)
line.pack(side=TOP, padx=2, pady=1)
colHeadings = self.itemClass.columnDefs()
currCol = 0
for cHdr, width in colHeadings:
l = Label(line, text=cHdr, width=width, bg='lightblue')
l.grid(row=0, column=currCol)
currCol += 1
line = Frame(self)
line.pack(fill=X)
def training(self, row):
sel = row.selection.get()
self.guesser.train(sel, row.original)
row.current = sel
self.guessAll()
def guessAll(self):
self.poolView.refresh()
pools = self.guesser.poolNames()
for row in self.rows:
row.setGuess(self.guesser.guess(row.original), pools)
def displayRow(self, row, bgc=None):
# UGH - REWRITE!
line = Frame(self.listView, bg=bgc)
line.pack(pady=1)
row.line = line
self.insertRadios(row)
Label(line, text=row.summary.get(), textvariable=row.summary, width=60, bg=bgc,
anchor=W).grid(row=0, column=2)
#Label(line, text=row.guess, width=7, bg=bgc, anchor=W).grid(row=0, column=1)
colourStripe = Label(line, text=' ', width=1, bg=bgc, anchor=W, relief=GROOVE)
colourStripe.grid(row=0, column=1)
line.colourStripe = colourStripe
pools = self.guesser.poolNames()
row.refreshColour(pools)
def poolAdded(self):
if not self.items:
return
pools = self.guesser.poolNames()
for row in self.rows:
for r in row.radios:
r.destroy()
self.insertRadios(row)
row.refreshColour(pools)
self.dirty = True
def insertRadios(self, row):
radioFrame = Frame(row.line)
radioFrame.grid(row=0, column=0)
currCol = 0
radios = []
v = row.selection
ci = 0
colours = row.defaultColours()
pools = self.guesser.poolNames()
for pool in pools:
rb = Radiobutton(radioFrame, text=pool, variable=v, value=pool, command=Command(self.training, row), bg=None)
rb.grid(row=0, column=currCol)
radios.append(rb)
currCol += 1
ci += 1
row.radios = radios
