In [ ]:

    

import psycopg2
import csv


    

In [ ]:

    

rdr = csv.DictReader(open('./Batch_3328128_batch_results_final.csv'))
items = [x for x in rdr if x['AssignmentStatus'] == 'Approved']


    

In [ ]:

    

conn = psycopg2.connect('service=firecares')

def sanitize_handle(s):
    return s.strip('@')


    

In [ ]:

    

with conn.cursor() as cur:
    for i in items:
        cur.execute('select id from firestation_firedepartment where name = %(name)s and state = %(state)s',
                    dict(name=i['Input.name'], state=i['Input.state']))
        id = cur.fetchone()[0]
        cur.execute('update firestation_firedepartment set twitter_handle = %(handle)s where id=%(id)s',
                    dict(handle=sanitize_handle(i['Answer.twitter_handle']), id=id))

conn.commit()


    

In [ ]: