Delete populate
This commit is contained in:
parent
33ed7cbf96
commit
81d3ea3292
1 changed files with 0 additions and 38 deletions
38
populate
38
populate
|
@ -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()
|
Loading…
Reference in a new issue