sbs-table-dev/submit_rows_sbs.py

import argparse
import csv
import fileinput
from utils import *

class Reader:
    def __init__(self, files, encoding):
        self.reader = csv.reader(fileinput.FileInput(files, openhook=fileinput.hook_encoded(encoding)),
                                 delimiter='\t', quoting=csv.QUOTE_NONE)
    def __iter__(self):
        return self
    def __next__(self):
        type, text, braille, _ = next(self.reader)
        if type == "*":
            return {'text': text,
                    'braille': braille}
        else:
            return next(self)

def main():
    parser = argparse.ArgumentParser(description='Submit entries to dictionary.')
    parser.add_argument('FILE', nargs='*', default=['-'], help="TSV file with entries")
    parser.add_argument('-d', '--dictionary', required=True, dest="DICTIONARY",
                        help="dictionary file")
    parser.add_argument('-a', '--all', type=bool, default=False, dest="SUBMIT_ALL",
                        help="submit all entries. if false, will only update the braille of existing words")
    args = parser.parse_args()
    conn, c = open_dictionary(args.DICTIONARY)
    reader = Reader(args.FILE, "utf-8")
    for row in reader:
        c.execute("UPDATE dictionary SET braille=:braille WHERE text=:text", row)
        if args.SUBMIT_ALL:
            c.execute("SELECT changes()")
            if not c.fetchone()[0]:
                c.execute("INSERT INTO dictionary VALUES (:text,:braille,0)", row)
    conn.commit()
    conn.close()

if __name__ == "__main__": main()