def add_datastream(id): feed = Feed.query.get(id) if feed == None: flash('Feed not found.') abort(404) form = AddDatastreamForm() form.type.choices = [(-1,'Select...'),(TYPE_ELECTRICITY_INST, 'Electricity Power'),(TYPE_ELECTRICITY, 'Electricity Cumulative'),(TYPE_HEAT, 'Heat'),(TYPE_WATER, 'Water')] if form.validate_on_submit(): newDatastream = Datastream(xively_id=form.xively_id.data,feed=feed,info=form.info.data,unit=form.unit.data,type=form.type.data) db.session.add(newDatastream) db.session.commit() flash('New datastream "'+newDatastream.xively_id+'" was added to '+newDatastream.feed.xively_id+'.') return redirect(url_for('datastream', id=newDatastream.id)) return render_template('add_datastream.html', form = form)
def edit_datastream(id): datastream = Datastream.query.get(id) if datastream == None: flash('Datastream not found.') abort(404) form = AddDatastreamForm() form.type.choices = [(-1,'Select...'),(TYPE_ELECTRICITY_INST, 'Electricity Power'),(TYPE_ELECTRICITY, 'Electricity Cumulative'),(TYPE_HEAT, 'Heat'),(TYPE_WATER, 'Water')] if form.validate_on_submit(): datastream.xively_id = form.xively_id.data datastream.unit = form.unit.data datastream.info = form.info.data datastream.type = form.type.data db.session.add(datastream) db.session.commit() flash('Your changes have been saved.') return redirect(url_for('datastream', id = id)) elif request.method != "POST": form.xively_id.data = datastream.xively_id form.unit.data = datastream.unit form.info.data = datastream.info return render_template('edit_datastream.html', datastream = datastream, form = form)