week5/b.py

   1 '''
   2 AY250 HW5 B
   3 Build sqlite3 table of states and names
   4 Author: Bryan Steinbach
   5 '''
   6
   7 import os
   8 import sqlite3
   9
  10 rawfn = 'candidate_names.txt'
  11 tablefn = 'poll'
  12
  13 if not os.path.exists(rawfn):
  14         print "Error, missing file %s"%rawfn
  15         exit(1)
  16
  17 conn = sqlite3.connect('poll')
  18
  19 c = conn.cursor()
  20
  21 c.execute(''' create table names
  22                 (state text, gop text, dem text, ind text, incpar text)''')
  23
  24 with open(rawfn) as f:
  25         f.readline()            # Skip first line
  26         for line in f.readlines():
  27                 words = line.strip().split(',')
  28                 state,dem,gop,ind,incpar = words
  29                 dem,gop,ind,incpar = [x.strip() for x in (dem,gop,ind,incpar)]
  30                 c.execute(''' insert into names
  31                                 values (?,?,?,?,?) ''',(state,gop,dem,ind,incpar))
  32
  33 conn.commit()
  34
  35 # sanity check
  36 c.execute('select * from names order by gop')
  37 for i in range(10):
  38         print c.next()
  39
  40 c.close()