def edit_pc():
if current_user.is_authenticated:
if current_user.username not in ACCESS_ADD_ITEM:
return redirect(url_for('login'))
else:
return redirect(url_for('login'))
form = AddPC()
if request.method == 'POST'and request.form.get("edit"):
pc_id = request.form.get("edit")
form.cpu.data = models.PC.query.filter_by(id=pc_id).first().CPU
form.ram.data = models.PC.query.filter_by(id=pc_id).first().RAM
form.mb.data = models.PC.query.filter_by(id=pc_id).first().MB
form.video.data = models.PC.query.filter_by(id=pc_id).first().video
form.hdd.data = models.PC.query.filter_by(id=pc_id).first().HDD
form.price.data = models.PC.query.filter_by(id=pc_id).first().price
form.comment.data = models.PC.query.filter_by(id=pc_id).first().comment
form.pc_id.data = models.PC.query.filter_by(id=pc_id).first().id
if request.method == 'POST' and form.validate_on_submit():
print('ok')
models.PC.query.filter_by(id=form.pc_id.data).update(dict(CPU=form.cpu.data,
RAM=form.ram.data,
HDD=form.hdd.data,
price=form.price.data,
comment=form.comment.data))
db.session.commit()
flash("Позиция {} успешно обновлена.".format(form.pc_id.data))
return redirect(url_for('manage'))
return render_template('edit_pc.html', form=form, approve=ACCESS_APPROVE, add_item=ACCESS_ADD_ITEM)
def add_pc():
# if current_user.is_authenticated:
# if current_user.username not in ACCESS_ADD_ITEM:
# return redirect(url_for('login'))
# else:
# return redirect(url_for('login'))
form = AddPC()
text = 'системный блок'
if request.method == 'POST' and form.validate_on_submit():
print('ok')
new_pc = models.PC(CPU=form.cpu.data,
RAM=form.ram.data,
HDD=form.hdd.data,
price=form.price.data,
comment=form.comment.data,
assembly_time=datetime.now(),
assembly_by=current_user.username,
status_id='1')
db.session.add(new_pc)
db.session.commit()
flash("Позиция успешно добавлена.")
return redirect(url_for('add_pc'))
return render_template('add_item.html', approve=ACCESS_APPROVE, form=form, add_item=ACCESS_ADD_ITEM, text=text)
