Esempio n. 1
0
def main():
    """ The main routine """
    ###
    # Parse arguments
    parser = argparse.ArgumentParser(description='Run etymdendron')
    parser.add_argument('word', help='Word to search for (will be asked for '
        'if not specified)', nargs='?', default=None)
    args = parser.parse_args()

    ###
    # First let's load the XML
    words_tree = loadDB(WORDS_FILE)
    if type(words_tree) is str:
        print(words_tree) # This holds the error message
        sys.exit(1)

    ###
    # Display a header
    print('=========================')
    print('====   Etymdendron   ====')
    print('=========================')
    print('{0} loaded, {1} trees found'.format(WORDS_FILE,
        len(words_tree.getroot())))

    ###
    # Search for a word
    if args.word is None:
        search_word = cli_funcs.get_search_word()
    else:
        search_word = args.word

    num_trees, matched_words = searchDB(words_tree, search_word)

    ###
    # Display the tree if we have matches
    # No match
    if num_trees == 0:
        print('{0} is not found in {1}'.format(search_word, WORDS_FILE))
        sys.exit(0)
    # Multiple matches
    elif num_trees > 1:
        print('{0} is found in {1} trees'.format(search_word, num_trees))
        chosen_root, chosen_word = cli_funcs.choose_word_from_many(
                matched_words)
    # One match
    elif num_trees == 1:
        print('{0} is found in one tree'.format(search_word))
        # Extract the tree and all matched words separately
        # Just pick the first entry (m_w[...][0] are the same in this case)
        chosen_root = matched_words[0][0]
        chosen_word = [match[1] for match in matched_words]

    # And now display the tree
    cli_funcs.display_tree(chosen_root, chosen_word, search_word)

    ###
    # That's all!
    sys.exit(0)
Esempio n. 2
0
 def LoadWordDB(self, filename=WORDS_FILE):
     """ Load the database file """
     self.words_tree = cf.loadDB(filename)
     if type(self.words_tree) is str:
         dlg_err = wx.MessageDialog(self.frame, self.words_tree
             ,'Error', wx.OK|wx.ICON_EXCLAMATION)
         dlg_err.ShowModal()
         dlg_err.Destroy()
         self.words_tree = None
     else:
         print('{0} loaded, {1} trees found'.format(filename,
         len(self.words_tree.getroot())))
Esempio n. 3
0
 def setUp(self):
     self.db = cf.loadDB(global_opts.WORDS_FILE)
Esempio n. 4
0
 def getDB(self):
     """ Helper function to load the DB """
     return cf.loadDB(global_opts.WORDS_FILE)