See CHANGELOG.md for history """ __author__ = "Michael Conlon" __copyright__ = "Copyright 2015 (c) Michael Conlon" __license__ = "New BSD License" __version__ = "0.02" from vivopump import read_csv_fp, write_csv_fp, get_vivo_ufid, get_parms import sys parms = get_parms() data_in = read_csv_fp(sys.stdin) print >>sys.stderr, len(data_in) data_out = {} vivo_ufid = get_vivo_ufid(parms) # get dictionary of uri keyed by ufid vivo_ufids = vivo_ufid.keys() print >>sys.stderr, 'VIVO ufid', len(vivo_ufid) source_ufid = [data_in[x]['UFID'] for x in data_in] print >>sys.stderr, 'Source ufid', len(source_ufid) vivo_data = data_in[data_in.keys()[0]] # Grab a row, any row # Process ufid in VIVO and in Source for row, data in data_in.items(): ufid = data['UFID'] if ufid in vivo_ufids: # ufid is in vivo and source data_out[row] = dict(data) data_out[row]['uri'] = vivo_ufid[ufid] data_out[row]['current'] = 'yes'
def test_vivo_get_ufid(self): result = get_vivo_ufid() self.assertTrue(len(result) > 0)