#!/usr/bin/env python

import rethinkdb as r
import optparse

if __name__ == "__main__":
    parser = optparse.OptionParser()
    parser.add_option("-P", "--port", dest="port", type="int",
                      help="rethinkdb port", default=30815)
    (options, args) = parser.parse_args()
    conn = r.connect('localhost', options.port, db='materialscommons')

    drafts = list(r.table('drafts').run(conn, time_format='raw'))
    for draft in drafts:
        if 'process' not in draft:
            continue
        input_conditions = draft['process']['input_conditions']
        if 'Specimen Prep' in input_conditions:
            sp = input_conditions['Specimen Prep']
            for prop in sp['default_properties']:
                if prop['attribute'] == 'preparation':
                    prop['value_choice'] = ["Electropolish", "FIB Liftout",
                                            "Other"]
                    prop['unit_choice'] = []
                    prop['value'] = prop['unit']
                    prop['unit'] = ""
                    r.table('drafts').get(draft['id']).update(draft).run(conn)

Add script to fix Emmanuelle's messed up drafts.
Add script to fix Emmanuelle's messed up drafts.
