import csv, os, psycopg2
from psycopg2.extras import DictCursor

TABLE = "community.other_tix_platform_attendees"

def db_connect():
  db_name = os.environ['db_name']
  db_user = os.environ['db_user']
  db_host = os.environ['db_host']
  db_credentials = os.environ['db_credentials']

  conn_string = "dbname='" + str(db_name) + "' user='" + str(db_user) + "' host='" + str(db_host) + "' password='" + str(db_credentials) + "'"

  try:
    conn = psycopg2.connect(str(conn_string))
    conn.autocommit = True
  except:
    print "Unable to connect to the database"

  cur = conn.cursor(cursor_factory=DictCursor)
  return cur

cur = db_connect()

cur.execute('ALTER TABLE %s ADD  UNIQUE (email, swoop_id);'%TABLE)

with open('usethisone.csv') as csv_file:
  csv_reader = csv.reader(csv_file, delimiter=',')
  line_count = 0
  for row in csv_reader:
    if line_count == 0:
      line_count += 1
    else:
      values = (row[4], row[1], ' '.join([row[2],row[3]]), row[3], 'manually added by twood for diego')
      cur.execute("""INSERT INTO community.other_tix_platform_attendees
	          (swoop_id, email, attendee_name, last_name, other_data)
		  ON CONFLICT DO NOTHING""", values)
      line_count +=1
  print('WROTE %i lines'%line_count)