From d2340a2320d51a2aab1601bb9048559ac11b1ca5 Mon Sep 17 00:00:00 2001 From: Andinus Date: Mon, 6 Apr 2020 21:53:36 +0530 Subject: Add grus-add script --- scripts/grus-add | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 54 insertions(+) create mode 100755 scripts/grus-add (limited to 'scripts/grus-add') diff --git a/scripts/grus-add b/scripts/grus-add new file mode 100755 index 0000000..ec1eac8 --- /dev/null +++ b/scripts/grus-add @@ -0,0 +1,54 @@ +#!/usr/bin/env python3 + +import sys +import sqlite3 +import argparse + +parser = argparse.ArgumentParser(description='grus-add') +parser.add_argument('-f', '--file', type=str, required=False, + help='file containing list of strings') +parser.add_argument('-w', '--word', type=str, required=False, + help='file containing list of strings') +parser.add_argument('-d', '--db', type=str, required=True, + help='database file') +args = parser.parse_args() + +if __name__ == '__main__': + if args.file == None and args.word == None: + print("-f or -w required") + print("run grus-add --help to print help") + sys.exit(1) + + conn = sqlite3.connect(args.db) + curs = conn.cursor() + + stmt = """CREATE TABLE IF NOT EXISTS words ( + word TEXT PRIMARY KEY NOT NULL, + sorted TEXT NOT NULL);""" + curs.execute(stmt) + conn.commit() + + stmt = """INSERT INTO words(word, sorted) + VALUES(?, ?);""" + + if args.file != None: + rows = [] + with open(args.file) as words: + for word in words: + word = word.strip() + lexical = ''.join(sorted(word)) + rows.append((word, lexical)) + print(len(rows)) + sys.stdout.write('\x1b[1A') + sys.stdout.write('\x1b[2K') + curs.executemany(stmt, rows) + + elif args.word != None: + word = args.word.strip() + lexical = ''.join(sorted(word)) + curs.execute(stmt, (word, lexical)) + + conn.commit() + + curs.close() + conn.close() -- cgit 1.4.1-2-gfad0