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)