def export(calc_id, datastore_key, format='csv', export_dir='.'): """ Export an output from the datastore. """ logging.basicConfig(level=logging.INFO) dstore = datastore.DataStore(calc_id) dstore.export_dir = export_dir with performance.PerformanceMonitor('export', measuremem=True) as mon: for fmt in format.split(','): fnames = export_((datastore_key, fmt), dstore) nbytes = sum(os.path.getsize(f) for f in fnames) print('Exported %s in %s' % (general.humansize(nbytes), fnames)) if mon.duration > 1: print(mon)
def export(datastore_key, export_dir='.', calc_id=-1, exports='csv'): """ Export an output from the datastore. """ logging.basicConfig(level=logging.INFO) dstore = datastore.read(calc_id) parent_id = dstore['oqparam'].hazard_calculation_id if parent_id: dstore.set_parent(datastore.read(parent_id)) dstore.export_dir = export_dir with performance.Monitor('export', measuremem=True) as mon: for fmt in exports.split(','): fnames = export_((datastore_key, fmt), dstore) nbytes = sum(os.path.getsize(f) for f in fnames) print('Exported %s in %s' % (general.humansize(nbytes), fnames)) if mon.duration > 1: print(mon)