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)
