diff options
Diffstat (limited to 'scripts/grus-add')
-rwxr-xr-x | scripts/grus-add | 56 |
1 files changed, 0 insertions, 56 deletions
diff --git a/scripts/grus-add b/scripts/grus-add deleted file mode 100755 index 4c3311a..0000000 --- a/scripts/grus-add +++ /dev/null @@ -1,56 +0,0 @@ -#!/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() - - print("Database initialized.") |