def setup_tables(host, port, user, passwd, db): config = {'host': host, 'port': port, 'user': user, 'passwd': passwd, 'db': db} try: db_connection = io.open_db_connection(config) io.setup_db_tables(db_connection) except Exception, e: print e
def setup_tables(host, port, user, passwd, db): config = { 'host': host, 'port': port, 'user': user, 'passwd': passwd, 'db': db } try: db_connection = io.open_db_connection(config) io.setup_db_tables(db_connection) except Exception, e: print e
def update_db(config, new_version=None, tmp_dir=None): obj_types = { 'vistrail': io.open_bundle_from_db, # 'workflow': io.open_from_db, # 'log': io.open_from_db, # 'registry': io.open_from_db } if new_version is None: new_version = currentVersion if tmp_dir is None: tmp_dir = tempfile.mkdtemp(prefix='vt_db') print 'creating tmpdir:', tmp_dir obj_id_lists = {} for obj_type in obj_types: obj_id_lists[obj_type] = io.get_db_object_list(config, obj_type) # read data out of database filenames = [] thumbnail_dir = os.path.join(tmp_dir, 'thumbs') os.mkdir(thumbnail_dir) db_connection = io.open_db_connection(config) for obj_type, obj_ids in obj_id_lists.iteritems(): for (obj_id, _, _) in obj_ids: old_version = io.get_db_object_version(db_connection, obj_id, 'vistrail') print 'getting', obj_type, 'id', obj_id local_tmp_dir = os.path.join(tmp_dir, str(obj_id)) vt_name = os.path.join(tmp_dir, str(obj_id) + '.vt') filenames.append(vt_name) os.mkdir(local_tmp_dir) res = obj_types[obj_type](obj_type, db_connection, obj_id, thumbnail_dir) io.save_vistrail_bundle_to_zip_xml(res, vt_name, local_tmp_dir) # drop the old database # recreate with the new version of the specs io.setup_db_tables(db_connection, None, old_version) # add the new data back for filename in filenames: (res, _) = io.open_vistrail_bundle_from_zip_xml(filename) io.save_vistrail_bundle_to_db(res, db_connection, 'with_ids') io.close_db_connection(db_connection)
def update_db(config, new_version=None, tmp_dir=None): obj_types = { "vistrail": io.open_bundle_from_db, # 'workflow': io.open_from_db, # 'log': io.open_from_db, # 'registry': io.open_from_db } if new_version is None: new_version = currentVersion if tmp_dir is None: tmp_dir = tempfile.mkdtemp(prefix="vt_db") print "creating tmpdir:", tmp_dir obj_id_lists = {} for obj_type in obj_types: obj_id_lists[obj_type] = io.get_db_object_list(config, obj_type) # read data out of database filenames = [] thumbnail_dir = os.path.join(tmp_dir, "thumbs") os.mkdir(thumbnail_dir) db_connection = io.open_db_connection(config) for obj_type, obj_ids in obj_id_lists.iteritems(): for (obj_id, _, _) in obj_ids: old_version = io.get_db_object_version(db_connection, obj_id, "vistrail") print "getting", obj_type, "id", obj_id local_tmp_dir = os.path.join(tmp_dir, str(obj_id)) vt_name = os.path.join(tmp_dir, str(obj_id) + ".vt") filenames.append(vt_name) os.mkdir(local_tmp_dir) res = obj_types[obj_type](obj_type, db_connection, obj_id, thumbnail_dir) io.save_vistrail_bundle_to_zip_xml(res, vt_name, local_tmp_dir) # drop the old database # recreate with the new version of the specs io.setup_db_tables(db_connection, None, old_version) # add the new data back for filename in filenames: (res, _) = io.open_vistrail_bundle_from_zip_xml(filename) io.save_vistrail_bundle_to_db(res, db_connection, "with_ids") io.close_db_connection(db_connection)