make_cards.py

import argparse
from PyDictionary import PyDictionary
import csv
import random

class Anki(object):

    def __init__(self,  vocab_file, output_file, error_file):
        self.vocab_file = vocab_file
        self.output_file = output_file
        self.error_file = error_file

    def read_vocab_file(self):
        vocab_list = [word.rstrip() for word in  open(self.vocab_file).readlines()]
        return vocab_list

    def define_vocab(self,vocab_list,verbose):
        dictionary = PyDictionary()
        defs_list = list()
        error_list= list()
        for word in vocab_list:
            if dictionary.meaning(word) != "":
                if verbose:
                    print([word, dictionary.meaning(word)])
                defs_list.append(dictionary.meaning(word))
            else:
                if verbose:
                    print("%s has been added to the error list"%word)
                error_list.append(word)
        return  defs_list, error_list

    def make_deck(self,vocab_list,defs_list,verbose):
        with open(self.output_file,"w") as csvfile:
            writer = csv.writer(csvfile, delimiter='\t')
            for word, definition in zip(vocab_list,defs_list):
                writer.writerow([word, definition])
        if verbose:
            print("anki deck has been created.")
        return self.output_file

    def write_errors(self,error_list,verbose):
        with open(self.error_file,"w") as csvfile:
            writer = csv.writer(csvfile, delimiter = '\t')
            for err in error_list:
                writer.writerow(err)
        if verbose:
            print("error log has been written.")
        return self.error_file

    def edit_cards(self):
        word_edits=list()
        words=list()
        defs=list()
        with open(self.output_file,"r") as csvfile:
            reader = csv.reader(csvfile, delimiter='\t')
            for row in reader:
                if len(row) == 2:
                    words.append(row[0])
                    defs.append(row[1])
                else:
                    word_edits.append(row[0])

        for edit in word_edits:
            print("\n")
            defin = input("definition for %s: "%edit)
            words.append(edit)
            defs.append(defin)

        return self

    def sentence_mode(self):
        import tkinter
        from tkinter import simpledialog

        tk_window = tkinter.Tk()
        tk_window.withdraw()
        tk_window.deiconify()


        word_list=list()

        with open(self.output_file,"r") as csvfile:
            reader = csv.reader(csvfile, delimiter='\t')
            for row in reader:
                word_list.append(row)

        test_idxs = list(range(len(word_list)))
        random.shuffle(test_idxs)

        for idx in test_idxs:
            sentence = simpledialog.askstring("Input","Write a sentence for the word: \n \n %s \n %s"%(word_list[idx][0],word_list[idx][1]), parent = tk_window)
        return self

    def sentence_no_gui_mode(self):
        word_list=list()
        with open(self.output_file,"r") as csvfile:
            reader = csv.reader(csvfile, delimiter='\t')
            for row in reader:
                word_list.append(row)

        test_idxs = list(range(len(word_list)))
        random.shuffle(test_idxs)

        for idx in test_idxs:
            print("***********************************************************")
            print("Write a sentence for the word:")
            print(" ")
            print(word_list[idx][0])
            print(word_list[idx][1])
            print(" ")
            test = input("sentence: ")



if __name__ == "__main__":

    parser = argparse.ArgumentParser(description='This script makes anki flash cards from a text file of vocab words.')
    parser.add_argument('vocab_file', metavar='vocab_file', action="store",
                        help='txt file full of vocab words')
    parser.add_argument('-o', '--output_file', metavar='output_file',
                        default='anki_gre_vocab.txt', action="store",
                        help='output anki deck txt file')
    parser.add_argument('-e', '--error_file',metavar='error_file',
                        default='error_words.txt',action="store",
                        help='error file for the words that there were no definitions')
    parser.add_argument('-m','--edit_mode', dest='edit_mode',default=False,
                        action="store_true", help='manually edit mode allows for the editing of an existing deck file to manually edit definitions')
    parser.add_argument('-v','--verbose', dest='verbose',default=False,
                        action="store_true", help='verbose mode')
    parser.add_argument('--sentence', dest='sentence_mode', default = False, action="store_true", help="sentence mode for learning words, dependency: tkinter required.")
    parser.add_argument('--sentence_no_gui', dest='sentence_no_gui_mode', default = False, action = "store_true", help = "no gui sentence mode for learning words")

    args = parser.parse_args()

    if args.edit_mode == True:
        anki = Anki(args.vocab_file, args.output_file,args.error_file)
        defs_list = anki.edit_cards()
    elif args.sentence_mode == True:
        anki = Anki(args.vocab_file, args.output_file,args.error_file)
        sentences = anki.sentence_mode()
    elif args.sentence_no_gui_mode == True:
        anki = Anki(args.vocab_file, args.output_file, args.error_file)
        sentences = anki.sentence_no_gui_mode()
    else:
        anki = Anki(args.vocab_file, args.output_file,args.error_file)
        vocab_list = anki.read_vocab_file()
        defs_list, error_list = anki.define_vocab(vocab_list, args.verbose)
        anki_deck = anki.make_deck(vocab_list,defs_list,args.verbose)
        error_log = anki.write_errors(error_list,args.verbose)