def device_profile(device_id):
# 如果 devices_id == '0',则是新增客户
if device_id == '0':
device = Device()
else:
device = Device.query.filter_by(id=device_id).first()
if device is None:
device = Device()
can_id = 1
if getattr(device, 'garbage_can_obj'):
can_id = device.garbage_can_obj.id
form = DeviceProfileForm(request.form, garbage_can=can_id)
if current_user:
form.garbage_can.choices = get_garbage_can_choice()
if request.method == 'POST' and form.validate_on_submit():
form.save_form(device)
flash(u'保存成功', category='success')
return redirect(url_for('devices.devices_list'))
return render_template('devices/device_profile.html',
form=form,
device=device)
