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…
	
	Add table
		Add a link
		
	
		Reference in a new issue