class CLI():
'''command line interface for the wordlist'''
def __init__(self, filename, stoplists=None,
freqsort=False, endsort=False):
'''command line interface constructor'''
self.stoplists = list()
if stoplists:
for stopfile in stoplists:
with open(stopfile, 'r') as inputfile:
stoplist = unicode(inputfile.read(), 'utf-8')
stoplist = RE_WORD.findall(stoplist.lower())
self.stoplists.extend(stoplist)
with open(filename, 'r') as inputfile:
self.wordlist = Wordlist(unicode(inputfile.read(), 'utf-8'),
self.stoplists)
self.endsort = endsort
self.freqsort = freqsort
def get_sorted(self):
'''return sorted wordlist'''
if self.endsort:
return self.wordlist.items_by_wordend()
if self.freqsort:
return self.wordlist.items_by_frequency()
return sorted(self.wordlist.items_by_wordbeginning())
def print_table(self):
'''print wordlist as a table'''
values = self.get_sorted()
llen = max([len(s) for s, i in values])
rlen = max([len(str(i)) for s, i in values])
align = '<'
if self.endsort:
align = '>'
print '+-{0}-+-{1}-+'.format(llen * '-', rlen * '-').encode('utf-8')
for word, freq in values:
print u'| {0:{align}{llen}} | {1:>{rlen}} |'.format(word, freq,
align=align,
llen=llen,
rlen=rlen
).encode('utf-8')
print '+-{0}-+-{1}-+'.format(llen * '-', rlen * '-').encode('utf-8')
def print_tabdelimited(self):
'''print wordlist as tab-delimited text'''
values = self.get_sorted()
for word, freq in values:
print u'{0}\t{1}'.format(word, freq).encode('utf-8')
class CLI():
'''command line interface for the wordlist'''
def __init__(self,
filename,
stoplists=None,
freqsort=False,
endsort=False):
'''command line interface constructor'''
self.stoplists = list()
if stoplists:
for stopfile in stoplists:
with open(stopfile, 'r') as inputfile:
stoplist = unicode(inputfile.read(), 'utf-8')
stoplist = RE_WORD.findall(stoplist.lower())
self.stoplists.extend(stoplist)
with open(filename, 'r') as inputfile:
self.wordlist = Wordlist(unicode(inputfile.read(), 'utf-8'),
self.stoplists)
self.endsort = endsort
self.freqsort = freqsort
def get_sorted(self):
'''return sorted wordlist'''
if self.endsort:
return self.wordlist.items_by_wordend()
if self.freqsort:
return self.wordlist.items_by_frequency()
return sorted(self.wordlist.items_by_wordbeginning())
def print_table(self):
'''print wordlist as a table'''
values = self.get_sorted()
llen = max([len(s) for s, i in values])
rlen = max([len(str(i)) for s, i in values])
align = '<'
if self.endsort:
align = '>'
print '+-{0}-+-{1}-+'.format(llen * '-', rlen * '-').encode('utf-8')
for word, freq in values:
print u'| {0:{align}{llen}} | {1:>{rlen}} |'.format(
word, freq, align=align, llen=llen, rlen=rlen).encode('utf-8')
print '+-{0}-+-{1}-+'.format(llen * '-', rlen * '-').encode('utf-8')
def print_tabdelimited(self):
'''print wordlist as tab-delimited text'''
values = self.get_sorted()
for word, freq in values:
print u'{0}\t{1}'.format(word, freq).encode('utf-8')
def on_stoplist(self, event):
'''open 'edit stoplist' dialogue'''
dialog = StoplistDlg(self.wordlist.stoplist, self)
response = dialog.ShowModal()
if response == wx.ID_OK:
self.wordlist = Wordlist(self.wordlist.text, dialog.stoplist)
self.table.update(self.wordlist)
dialog.Destroy()
def __init__(self,
filename,
stoplists=None,
freqsort=False,
endsort=False):
'''command line interface constructor'''
self.stoplists = list()
if stoplists:
for stopfile in stoplists:
with open(stopfile, 'r') as inputfile:
stoplist = unicode(inputfile.read(), 'utf-8')
stoplist = RE_WORD.findall(stoplist.lower())
self.stoplists.extend(stoplist)
with open(filename, 'r') as inputfile:
self.wordlist = Wordlist(unicode(inputfile.read(), 'utf-8'),
self.stoplists)
self.endsort = endsort
self.freqsort = freqsort
def __init__(self, filename, stoplists=None,
freqsort=False, endsort=False):
'''command line interface constructor'''
self.stoplists = list()
if stoplists:
for stopfile in stoplists:
with open(stopfile, 'r') as inputfile:
stoplist = unicode(inputfile.read(), 'utf-8')
stoplist = RE_WORD.findall(stoplist.lower())
self.stoplists.extend(stoplist)
with open(filename, 'r') as inputfile:
self.wordlist = Wordlist(unicode(inputfile.read(), 'utf-8'),
self.stoplists)
self.endsort = endsort
self.freqsort = freqsort
def load_wordlist(self, filename):
'''load wordlist from file'''
filename = os.path.abspath(filename)
self.SetStatusText('Reading data from file...')
try:
with open(filename, 'r') as inputfile:
text = unicode(inputfile.read(), 'utf-8')
except Exception as error:
wx.MessageBox(str(error), '', wx.OK | wx.ICON_ERROR)
self.SetStatusText('Could not read data.')
else:
self.wordlist = Wordlist(text)
self.SetStatusText('Data read.')
self.SetTitle(os.path.basename(filename))
self.dirname = os.path.dirname(filename)
self.filename = os.path.basename(filename)
self.enable_controls()
