def set_up(): output = io.BytesIO() writer = csv.writer( output ) heading = SimulationState.get_field_order( False ) heading.extend( config.get_derived_gains_dict( sessionId ).keys() ) writer.writerow( heading ) return output.getvalue()
def handle_history( simulations ): output = cStringIO.StringIO() writer = csv.writer( output ) for simulation in simulations: for state in simulation.history: derivedGains = state.get_derived_gains() stateAsList = [ state.__dict__.get( key ) for key in \ SimulationState.get_field_order( False ) ] stateAsList.extend( [ derivedGains.get( key ) for key in \ config.get_derived_gains_dict( sessionId ).keys() ] ) writer.writerow( stateAsList ) return output.getvalue()