#!/usr/bin/env python

import json
import sys
import csv
from pymongo import MongoClient

if len(sys.argv) < 2:
  print 'usage: python read.py filepath'
  sys.exit(1)

reader = csv.DictReader(open(sys.argv[1]), delimiter=',', quotechar='"')
conn = MongoClient()
db = conn.asterank
coll = db.exo
coll.drop()
coll.ensure_index('kepoi_name', unique=True)  # kepid isn't actually unique...
# TODO ensure koi_sma descending index

c = 0
for row in reader:
  #row['pl_fulldes'] = '%s%s' % (row['pl_hostname'], row['pl_letter'])
  for key, val in row.iteritems():
    try:
      val = float(val)
    except ValueError:
      pass
    row[key] = val
  coll.insert(row, continue_on_error=True)
  c += 1


# put in db

print 'Added', c, 'candidate exoplanets'
print 'Done.'

#!/usr/bin/env python

import json
import sys
import csv
from pymongo import MongoClient

if len(sys.argv) < 2:
  print 'usage: python read.py filepath'
  sys.exit(1)

reader = csv.DictReader(open(sys.argv[1]), delimiter=',', quotechar='"')
conn = MongoClient()
db = conn.asterank
coll = db.exo
coll.drop()
coll.ensure_index('kepoi_name', unique=True)  # kepid isn't actually unique...

c = 0
for row in reader:
  for key, val in row.iteritems():
    try:
      val = float(val)
    except ValueError:
      pass
    row[key] = val
  coll.insert(row, continue_on_error=True)
  c += 1

print 'Added', c, 'candidate exoplanets'
print 'Done.'

Clean things up a little
Clean things up a little