def rack_edit(id): rack = Rack.query.filter_by(id=id).first() form = RackEditForm(request.form, rack) form.location_id.choices = [(i.id, i.short_name) for i in Location.query.all()] form.location_id.choices.insert(0, (0, u'- 指定机房 -')) form.area_id.choices = [(i.id, i.name) for i in Area.query.all()] form.area_id.choices.insert(0, (0, u'- 指定区域 -')) form.vlan_id.choices = [(i.id, i.name) for i in Vlan.query.all()] form.vlan_id.choices.insert(0, (0, u'- 指定vlan -')) if form.validate_on_submit(): form.populate_obj(rack) vlan_list = [str(i) for i in form.vlan_id.data] rack.vlan_id = ','.join(vlan_list) db.session.commit() return redirect(url_for("datacenter.rack_list")) return render_template('datacenter/rack_edit.html', form=form)
def rack_add(): form = RackEditForm(request.form) form.location_id.choices = [(i.id, i.short_name) for i in Location.query.all()] form.location_id.choices.insert(0, (0, u'- 指定机房 -')) form.area_id.choices = [(i.id, i.name) for i in Area.query.all()] form.area_id.choices.insert(0, (0, u'- 指定区域 -')) form.vlan_id.choices = [(i.id, i.name) for i in Vlan.query.all()] form.vlan_id.choices.insert(0, (0, u'- 指定vlan -')) if form.validate_on_submit(): rack = Rack() form.populate_obj(rack) vlan_list = [str(i) for i in form.vlan_id.data] rack.vlan_id= ','.join(vlan_list) db.session.add(rack) db.session.commit() write_unit_table(form.units.data) flash(u'成功增加一条机柜记录!') return redirect(url_for("datacenter.rack_list")) return render_template('datacenter/rack_edit.html', form=form)