diff --git a/populate b/populate deleted file mode 100755 index 52c25be..0000000 --- a/populate +++ /dev/null @@ -1,38 +0,0 @@ -#!/usr/bin/env python3 -import psycopg -import yaml -import sys - -with open('config.yml', 'r', encoding='UTF-8') as data: - config = yaml.safe_load(data.read()) -with open(config['response_tokens'], 'r', encoding='UTF-8') as data: - valid_tokens = {item for item in yaml.safe_load(data.read())} -#with open(sys.argv[1], 'r', encoding='UTF-8') as data: -# phrases = yaml.safe_load(data.read()) -with open(sys.argv[1], 'r', encoding='UTF-8') as data: - names = yaml.safe_load(data.read()) - -with psycopg.connect(config['db_spec']) as conn: - query_phrases = 'INSERT INTO responses (tokens, response) VALUES (%s, %s)' - query_names = 'INSERT INTO names (tg_id, name) VALUES (%s, %s)' - with conn.cursor() as cur: - #for regexref, responses in phrases.items(): - # tokens = set(regexref.split(',')) - # if tokens != tokens & valid_tokens: - # print(f'{str(tokens)} failed to add!') - # continue - # token_string = ','.join(sorted(tokens)) - # for response in responses: - # cur.execute('SAVEPOINT sp1') - # try: - # cur.execute(query_phrases, (token_string, response)) - # except psycopg.errors.UniqueViolation as err: - # cur.execute('ROLLBACK TO SAVEPOINT sp1') - # # print(err) - # continue - # cur.execute('RELEASE SAVEPOINT sp1') - #conn.commit() - for tg_id, usernames in names.items(): - for username in usernames: - cur.execute(query_names, (tg_id, username)) - conn.commit()